dorsal/arxiv
View SchemaFlashMem: Distilling Intrinsic Latent Memory via Computation Reuse
| Authors | Yubo Hou, Zhisheng Chen, Tao Wan, Zengchang Qin |
|---|---|
| Categories | |
| ArXiv ID | 2601.05505vv1 |
| URL | https://arxiv.org/abs/2601.05505 |
| License | http://arxiv.org/licenses/nonexclusive-distrib/1.0/ |
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.
{
"annotation_id": "177bdb54-58b2-4803-aece-9b1c96aafd6f",
"date_created": "2026-02-17T05:53:04.778000Z",
"date_modified": "2026-02-17T05:53:04.778000Z",
"file_hash": "0fc4ac6d2774f7f2d25ccbecbbd0fd6e8c4d4d1b1ea8bb343bd8a4d8bdc15c6a",
"private": false,
"record": {
"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\u0027s 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.",
"arxiv_id": "2601.05505",
"authors": [
"Yubo Hou",
"Zhisheng Chen",
"Tao Wan",
"Zengchang Qin"
],
"categories": [
"cs.CL"
],
"license": "http://arxiv.org/licenses/nonexclusive-distrib/1.0/",
"title": "FlashMem: Distilling Intrinsic Latent Memory via Computation Reuse",
"url": "https://arxiv.org/abs/2601.05505",
"version": "v1"
},
"schema_id": "dorsal/arxiv",
"source": {
"execution_id": "273ea842-2140-46d1-8abf-ec3678ede875",
"id": "arXiv Dataset",
"type": "Model",
"variant": "snapshot-2026-01-17",
"version": "0.1.0"
},
"user_id": 1000002
}