元认知记忆策略优化用于长时域LLM智能体
阅读原文· arxiv.org记忆增强的LLM智能体通过递归摘要交互轨迹应对长时域任务,但现有结果导向强化学习无法定位中间记忆质量下降。本研究提出信念熵(Belief Entropy)作为自监督代理,衡量模型对潜在任务状态的不确定性,并在此基础上提出元认知记忆策略优化(MMPO),通过惩罚高认知不确定性的摘要提供细粒度监督。实验表明,MMPO在多种长时域任务上持续优于现有方法,在扩展到1.75M token上下文时仍保持97.1%性能。
Memory-augmented LLM agents tackle complex long-horizon tasks by recursively summarizing interaction trajectories into compact memory. However, existing approaches typically train these memory policies using outcome-based reinforcement learning, failing to localize where intermediate memory quality degrades. As interactions unfold, ambiguous recursive summaries progressively discard task-relevant information and introduce semantic noise. This exacerbates belief deviation, obscuring the agent's estimate of the latent task state and ultimately derailing long-horizon reasoning. We therefore argue that memory optimization should focus not merely on trajectory-level success, but on the clarity of the belief induced by intermediate summaries. To this end, we introduce Belief Entropy, a self-supervised proxy that probes how uncertain the model remains about the latent task state given its current memory. Based on this proxy, we propose Metacognitive Memory Policy Optimization (MMPO). Instead of relying only on sparse outcome-based signals, MMPO provides fine-grained, memory-specific supervision via explicitly penalizing summaries that induce high epistemic uncertainty. Experiments show that MMPO consistently outperforms existing methods on diverse long-horizon tasks, maintaining 97.1% performance even when scaled to 1.75M-token contexts.