Enabling Confidence-Aware Control of Effective Memorization in Large Language Models

Chenjie Ni, Omid Mersa, Zhepeng Wang, Runxue Bao, Qun Li, Shangqian Gao, Yanfu Zhang


Abstract
Current evaluations of memorization in Large Language Models (LLMs) predominantly rely on overlap ratios, which fails to distinguish between sporadic token matches and reliable, sequential recall. What’s more, these methods become inapplicable in practical scenarios where ground truth suffixes is unavailable. To address these limitations, we propose a paradigm shift from quantifying "rates" to determining the Effective Memorization Length—the distinct boundary where a model transitions from faithful recall to open-ended generation. We introduce a Ensemble Memorization Classifier that leverages internal states from the LLM to estimate this boundary in a reference-free manner. This framework enables Confidence-Aware Generation Control, allowing to extract high-precision suffixes by dynamically truncating sequences precisely at the memorization boundary. Furthermore, we demonstrate that this boundary is optimizable; by integrating our classifier as a reward model within a reinforcement learning framework, we actively extend the average effective length of generated suffixes. Experiments on the GPT-Neo and Pythia families demonstrate that our approach not only accurately filters non-memorization tokens but also effectively "unlocks" deeper chains of memorized knowledge, transforming memorization from an unpredictable risk into a controllable utility.
Anthology ID:
2026.emnlp-main.2252
Volume:
Proceedings of the 2026 Conference on Empirical Methods in Natural Language Processing
Month:
October
Year:
2026
Address:
Budapest, Hungary
Editors:
Hung-Yi Lee, Sunipa Dev, Jessy Li, Desmond Elliott
Venue:
EMNLP
SIG:
SIGDAT
Publisher:
Association for Computational Linguistics
Note:
Pages:
51233–51246
Language:
URL:
https://aclanthology.org/2026.emnlp-main.2252/
DOI:
Bibkey:
Cite (ACL):
Chenjie Ni, Omid Mersa, Zhepeng Wang, Runxue Bao, Qun Li, Shangqian Gao, and Yanfu Zhang. 2026. Enabling Confidence-Aware Control of Effective Memorization in Large Language Models. In Proceedings of the 2026 Conference on Empirical Methods in Natural Language Processing, pages 51233–51246, Budapest, Hungary. Association for Computational Linguistics.
Cite (Informal):
Enabling Confidence-Aware Control of Effective Memorization in Large Language Models (Ni et al., EMNLP 2026)
Copy Citation:
PDF:
https://aclanthology.org/2026.emnlp-main.2252.pdf