Memory Dial: A Training Framework for Controllable Memorization in Language Models

Xiangbo Zhang, Ali Emami


Abstract
Memorization in language models is widely studied but remains difficult to isolate and control. Understanding when and what models memorize is essential for explaining their predictions, yet existing approaches are post-hoc: they can detect memorization in trained models, but cannot disentangle its effects from architecture, data, or optimization. We introduce **Memory Dial**, a training framework that makes memorization an explicit, controllable variable. Memory Dial interpolates between standard cross-entropy and a temperature-sharpened objective via a single parameter, producing a family of models identical in architecture, data, and optimization, but varying in memorization pressure. Experiments across six architectures and five benchmarks demonstrate that: (1) reliably controls memorization, with seen-example accuracy increasing monotonically while unseen accuracy remains stable; (2) larger models are more responsive to memorization pressure; and (3) frequent sequences are easier to memorize than rare ones. Memory Dial provides a controlled experimental framework for studying how memorization behavior emerges and interacts with generalization in language models.
Anthology ID:
2026.findings-acl.179
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:
3654–3673
Language:
URL:
https://preview.aclanthology.org/ingest-acl/2026.findings-acl.179/
DOI:
Bibkey:
Cite (ACL):
Xiangbo Zhang and Ali Emami. 2026. Memory Dial: A Training Framework for Controllable Memorization in Language Models. In Findings of the Association for Computational Linguistics: ACL 2026, pages 3654–3673, San Diego, California, United States. Association for Computational Linguistics.
Cite (Informal):
Memory Dial: A Training Framework for Controllable Memorization in Language Models (Zhang & Emami, Findings 2026)
Copy Citation:
PDF:
https://preview.aclanthology.org/ingest-acl/2026.findings-acl.179.pdf
Checklist:
 2026.findings-acl.179.checklist.pdf