FlashMem: Distilling Intrinsic Latent Memory via Computation Reuse

Yubo Hou, Zhisheng Chen, Tao Wan, Zengchang Qin


Abstract
The stateless architecture of Large Language Models inherently lacks the mechanism to preserve dynamic context, compelling agents to redundantly reprocess history to maintain long-horizon autonomy. While latent memory offers a solution, current approaches are hindered by architectural segregation, relying on auxiliary encoders that decouple memory from the reasoning backbone. We propose FlashMem, a framework that distills intrinsic memory directly from transient reasoning states via computation reuse. Leveraging the property that internal representations uniquely encode input trajectories, FlashMem identifies the last hidden state as a sufficient statistic for the interaction history. This enables a Shared-KV Consolidator to synthesize memory by attending directly to the backbone’s frozen cache, eliminating redundant re-parameterization. Furthermore, a parameter-free Cognitive Monitor leverages attention entropy to adaptively trigger consolidation only when high epistemic uncertainty is detected. Experiments demonstrate that FlashMem matches the performance of heavy baselines while reducing inference latency by 5 times, effectively bridging the gap between efficiency and persistent cognition.
Anthology ID:
2026.findings-acl.230
Volume:
Findings of the Association for Computational Linguistics: ACL 2026
Month:
July
Year:
2026
Address:
San Diego, California, United States
Editors:
Maria Liakata, Viviane P. Moreira, Jiajun Zhang, David Jurgens
Venue:
Findings
SIG:
Publisher:
Association for Computational Linguistics
Note:
Pages:
4687–4705
Language:
URL:
https://preview.aclanthology.org/ingest-acl/2026.findings-acl.230/
DOI:
Bibkey:
Cite (ACL):
Yubo Hou, Zhisheng Chen, Tao Wan, and Zengchang Qin. 2026. FlashMem: Distilling Intrinsic Latent Memory via Computation Reuse. In Findings of the Association for Computational Linguistics: ACL 2026, pages 4687–4705, San Diego, California, United States. Association for Computational Linguistics.
Cite (Informal):
FlashMem: Distilling Intrinsic Latent Memory via Computation Reuse (Hou et al., Findings 2026)
Copy Citation:
PDF:
https://preview.aclanthology.org/ingest-acl/2026.findings-acl.230.pdf
Checklist:
 2026.findings-acl.230.checklist.pdf