mlsanityData to 7 Oct 2026

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

arXiv 2604.05074 ↗First posted 6 Apr 2026cs.CLcited by 1

Not on the map. Why?

Abstract (arXiv)

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 pressure 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 and training setup (within each sweep), differing only in memorization pressure. Experiments across six architectures and five benchmarks demonstrate that: (1) α reliably controls memorization pressure, 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. Additional analyses show that the effect is robust across a range of sharpening temperatures, differs qualitatively from single-temperature cross-entropy, transfers to multilingual settings, and is detectable even on naturally occurring single-occurrence sequences. Memory Dial provides a controlled experimental framework for studying how memorization behavior emerges and interacts with generalization in language models.

Key facts. Cited by 1 paper in the map (data through 7 Oct 2026). Builds on 24 of its 32 references that are in the map. First posted on arXiv 6 Apr 2026.

Builds on

24 of 32 references.

References · 24

  1. 2015Distilling the Knowledge in a Neural Networkcited by 10,076
  2. 2015Rethinking the Inception Architecture for Computer Visioncited by 5,451
  3. 2018Think you have Solved Question Answering? Try ARC, the AI2 Reasoning Challengecited by 4,710
  4. 2022OPT: Open Pre-trained Transformer Language Modelscited by 3,464
  5. 2019The Curious Case of Neural Text Degenerationcited by 2,881
  6. 2019PIQA: Reasoning about Physical Commonsense in Natural Languagecited by 2,818
  7. 2020Extracting Training Data from Large Language Modelscited by 2,144
  8. 2019BoolQ: Exploring the Surprising Difficulty of Natural Yes/No Questionscited by 2,140
  9. 2018Can a Suit of Armor Conduct Electricity? A New Dataset for Open Book Question Answeringcited by 2,114
  10. 2019Language Models as Knowledge Bases?cited by 1,922
  11. 2023Inference-Time Intervention: Eliciting Truthful Answers from a Language Modelcited by 1,259
  12. 2023Steering Llama 2 via Contrastive Activation Additioncited by 1,009
12 more
  1. 2018The Secret Sharer: Evaluating and Testing Unintended Memorization in Neural Networkscited by 847
  2. 2023Progress measures for grokking via mechanistic interpretabilitycited by 822
  3. 2021Deduplicating Training Data Makes Language Models Bettercited by 705
  4. 2024TinyLlama: An Open-Source Small Language Modelcited by 604
  5. 2023Dissecting Recall of Factual Associations in Auto-Regressive Language Modelscited by 512
  6. 2017Regularizing Neural Networks by Penalizing Confident Output Distributionscited by 461
  7. 2018SWAG: A Large-Scale Adversarial Dataset for Grounded Commonsense Inferencecited by 333
  8. 2022Towards Understanding Grokking: An Effective Theory of Representation Learningcited by 229
  9. 2023Proving Test Set Contamination in Black Box Language Modelscited by 227
  10. 2025The Reasoning-Memorization Interplay in Language Models Is Mediated by a Single Directioncited by 21
  11. 2020An Analysis of Dataset Overlap on Winograd-Style Taskscited by 16
  12. 2024LLMs and Memorization: On Quality and Specificity of Copyright Compliancecited by 16

Built on by

Cited by year (2026 to 7 Oct)

Citing papers per year: 2026: 1.

1 paper.