arXiv:2607.06974cs.CLcs.LG2026-07

让大模型在推理时动态积累经验,自动挑选有用记忆提升解题能力。

MILES: Modular Instruction Memory with Learnable Selection for Self-Improving LLM Reasoning

论文配图:MILES: Modular Instruction Memory with Learnable Selection for Self-Improving LLM Reasoning
图 1 · 摘自论文原文
  • 用可学习的模块化记忆单元存储子目标与指令对
  • 通过粗粒度检索+细粒度重排序实现精准记忆选择
  • 适合需要持续优化推理能力的在线应用

大语言模型在测试时可通过额外计算提升推理能力,但多数方法孤立处理每个问题。当问题顺序到来时,累积可复用的经验能进一步提升性能。现有基于记忆的方法要么存储完整解法模板,泛化性差;要么使用启发式步骤级选择,无法优化最终答案正确性。学习选择策略需大规模训练数据和固定动作空间,不适用于测试时内存逐步扩展且监督有限的场景。我们提出MILES(Modular Instruction Memory with LEarnable Selection for self-improving LLM reasoning),一种在真实测试时约束下动态扩展步骤级记忆并优化正确性的框架。MILES维护由子目标嵌入与子指令组成的模块化记忆单元,每个单元配有可学习的选择头。该结构支持粗到细的检索机制:粗粒度层实现记忆扩展,并从高置信样本中收集监督信号训练选择头;细粒度层利用学习到的选择头重排序候选记忆,引导不确定样本的推理。大量实验表明,MILES在性能上一致匹配或超越先前方法,同时取得更优的准确率-效率权衡,展现出有效性、鲁棒性和可迁移性。

原文摘要 · Abstract (English)

Large language models (LLMs) increasingly improve their reasoning at test time via additional computation, yet most existing works treat each problem in isolation. When problems arrive sequentially, accumulating reusable experience across them can further improve performance. Existing memory-based methods either store whole-solution templates that generalize poorly to novel problems or use heuristic step-level selection that is not optimized for final-answer correctness. Learning selection policies requires large-scale training data and fixed action spaces, making such approaches unsuitable for test-time settings where memory expands incrementally and only limited supervision is available. We propose MILES (Modular Instruction Memory with LEarnable Selection for self-improving LLM reasoning), a framework that dynamically expands step-wise memory and applies correctness-optimized memory composition under realistic test-time constraints. MILES maintains modular memory units consisting of asymmetric pairs of sub-goal embeddings and sub-instructions, each associated with a learnable selection head. This memory structure enables a coarse-to-fine retrieval mechanism: The coarse level enables memory expansion and collects supervision for training selection heads from confident samples, while the fine stage applies learned selection heads to rerank coarse-level candidates and guide reasoning for uncertain samples. MILES consistently matches or outperforms prior methods while achieving superior accuracy-efficiency tradeoffs. Extensive experiments demonstrate its effectiveness, robustness, and transferability.

自改进记忆增强推理优化

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。