Self-Evolving LLM Memory Extraction Across Heterogeneous Tasks
On the map: Synthetic data and self-improvement for LLMs → Self-rewarding alignment and self-training
Abstract (arXiv)
As LLM-based assistants become persistent and personalized, they must extract and retain useful information from past conversations as memory. However, the types of information worth remembering vary considerably across tasks. We formalize the textitheterogeneous memory extraction task and introduce BEHEMOTH, a benchmark that repurposes 18 existing datasets spanning personalization, problem-solving, and agentic tasks, using a downstream utility-driven metric for systematic evaluation. Our empirical analysis confirms that no single static extraction prompt dominates across all task categories, and that existing self-evolving prompt optimization frameworks, originally designed for homogeneous distributions, degrade when training tasks are heterogeneous. To address this, we propose CluE, a cluster-based self-evolving strategy that groups training examples into clusters by extraction scenarios, analyzes each cluster independently, and synthesizes cross-cluster insights to update the extraction prompt. Experiments on BEHEMOTH show that CluE generalizes effectively across heterogeneous tasks (+9.04% relative gain), consistently outperforming prior self-evolving frameworks.
Key facts. Cited by 2 papers in the map (data through 7 Oct 2026). Placed in Self-rewarding alignment and self-training, one of 13 directions in Synthetic data and self-improvement for LLMs. Builds on 34 of its 58 references that are in the map. First posted on arXiv 13 Apr 2026.
Builds on
34 of 58 references.
References · 34
- 2023Lost in the Middle: How Language Models Use Long Contextscited by 3,383
- 2020Longformer: The Long-Document Transformercited by 3,362
- 2023GPQA: A Graduate-Level Google-Proof Q&A Benchmarkcited by 2,748
- 2024MMLU-Pro: A More Robust and Challenging Multi-Task Language Understanding Benchmarkcited by 1,758
- 2018FEVER: a large-scale dataset for Fact Extraction and VERificationcited by 1,362
- 2020ALFWorld: Aligning Text and Embodied Environments for Interactive Learningcited by 1,143
- 2022Large Language Models Are Human-Level Prompt Engineerscited by 1,020
- 2025A-MEM: Agentic Memory for LLM Agentscited by 1,000
- 2023Large Language Models as Optimizerscited by 830
- 2023ExpeL: LLM Agents Are Experiential Learnerscited by 814
- 2025Mem0: Building Production-Ready AI Agents with Scalable Long-Term Memorycited by 722
- 2024LongMemEval: Benchmarking Chat Assistants on Long-Term Interactive Memorycited by 656
22 more
- 2024A Survey on the Memory Mechanism of Large Language Model based Agentscited by 622
- 2024BigCodeBench: Benchmarking Code Generation with Diverse Function Calls and Complex Instructionscited by 527
- 2025GEPA: Reflective Prompt Evolution Can Outperform Reinforcement Learningcited by 449
- 2024Justice or Prejudice? Quantifying Biases in LLM-as-a-Judgecited by 354
- 2025Agentic Context Engineering: Evolving Contexts for Self-Improving Language Modelscited by 349
- 2024AgentBoard: An Analytical Evaluation Board of Multi-turn LLM Agentscited by 319
- 2024Humans or LLMs as the Judge? A Study on Judgement Biasescited by 314
- 2025Memory in the Age of AI Agentscited by 285
- 2025ReasoningBank: Scaling Agent Self-Evolving with Reasoning Memorycited by 253
- 2025Dynamic Cheatsheet: Test-Time Learning with Adaptive Memorycited by 146
- 2025Do LLMs Recognize Your Preferences? Evaluating Personalized Preference Following in LLMscited by 142
- 2024StableToolBench: Towards Stable Large-Scale Benchmarking on Tool Learning of Large Language Modelscited by 129
- 2025Evo-Memory: Benchmarking LLM Agent Test-time Learning with Self-Evolving Memorycited by 129
- 2026MemSkill: Learning and Evolving Memory Skills for Self-Evolving Agentscited by 129
- 2024Meta-Prompting: Enhancing Language Models with Task-Agnostic Scaffoldingcited by 115
- 2025G-Memory: Tracing Hierarchical Memory for Multi-Agent Systemscited by 112
- 2025MemGen: Weaving Generative Latent Memory for Self-Evolving Agentscited by 99
- 2025MemBench: Towards More Comprehensive Evaluation on the Memory of LLM-based Agentscited by 82
- 2025PersonaMem-v2: Towards Personalized Intelligence via Learning Implicit User Personas and Agentic Memorycited by 82
- 2025MemEvolve: Meta-Evolution of Agent Memory Systemscited by 81
- 2025A Survey of Personalized Large Language Models: Progress and Future Directionscited by 65
- 2026Learning to Continually Learn via Meta-learning Agentic Memory Designscited by 16
Built on by
Citing papers per year: 2026: 2.
2 papers.
Similar, not cited
Close in wording, no citation either way
- 2026UMEM: Unified Memory Extraction and Management Framework for Generalizable Memory
- 2026AMemGym: Interactive Memory Benchmarking for Assistants in Long-Horizon Conversations
- 2026Towards Autonomous Memory Agents
- 2026Evaluating Memory Structure in LLM Agents
- 2026Generalizable Self-Evolving Memory for Automatic Prompt Optimization
- 2026Learning What to Remember and What to Internalize in LLM Self-Evolution via Adaptive Memory-Parameter Coordination
- 2026ERSkill: Evolving for Skill-Guided Adaptive Memory Retrieval
- 2026GenericAgent: A Token-Efficient Self-Evolving LLM Agent via Contextual Information Density Maximization (V1.0)