Deep Fusion Memory: A Chunk-Level Parametric Memory for Large Language Models
Abstract
Deep Fusion Memory integrates predicted parametric memory into intermediate hidden states via gated cross-attention, reducing retrieval overhead and improving knowledge-intensive tasks without online retrieval.
Large language models need effective ways to acquire and use external knowledge beyond what they retain from pretraining. Continued pretraining risks catastrophic forgetting, while retrieval at inference time can add substantial overhead. External parametric memory avoids online retrieval, but constructing its training data requires extensive offline search, and output-level interpolation provides only shallow fusion. We introduce Deep Fusion Memory (DFM), a parametric memory architecture that reduces this construction workload and incorporates learned memory into intermediate hidden states. A causal memory predictor learns to generate memory from the preceding context, using retrieved chunk embeddings from the corpus as supervision. The predicted memory is shared across tokens within each chunk and fused into their hidden states through gated cross-attention. Memory prediction and fusion are trained end to end with the backbone frozen, and inference requires no explicit retrieval. Compared with existing parametric memory approaches, which often construct token-level memory targets, DFM requires 32times fewer retrieval queries and 2048times fewer retrieved items to construct its training data. Across six QA tasks, DFM performs better over the strongest baseline by 12.71% on Mistral-7B-v0.3 and 11.19% on Qwen3-8B-Base. Beyond QA, domain adaptation yields average gains of 6.51 and 4.38 points over Qwen3-4B-Base and Qwen3-8B-Base, respectively.
Community
We introduce Deep Fusion Memory (DFM), a chunk-level parametric memory architecture that enables LLMs to incorporate external knowledge without retrieval at inference time.
DFM introduces two key ideas:
- Efficient memory construction: A causal memory predictor learns from retrieved chunk embeddings, reducing retrieval queries by 32× and retrieved items by 2048× during training data construction.
- Deep memory fusion: Predicted memory is integrated into intermediate hidden states through gated cross-attention, while keeping the LLM backbone frozen.
Experiments demonstrate consistent improvements across knowledge-intensive QA and domain adaptation tasks.
We hope this work offers a scalable and efficient direction for equipping LLMs with external parametric memory. Feedback and discussions are welcome!
Get this paper in your agent:
hf papers read 2610.25100 Don't have the latest CLI?
curl -LsSf https://hf.co/cli/install.sh | bash Models citing this paper 0
No model linking this paper
Datasets citing this paper 0
No dataset linking this paper
Spaces citing this paper 0
No Space linking this paper