LLM Memory & RAG 相关度: 9/10

Self-Evolving LLM Memory Extraction Across Heterogeneous Tasks

Yuqing Yang, Tengxiao Liu, Wang Bill Zhu, Taiwei Shi, Linxin Song, Robin Jia
arXiv: 2604.11610v1 发布: 2026-04-13 更新: 2026-04-13

AI 摘要

论文提出解决LLM在异构任务中记忆提取问题的CluE框架,优于现有自进化框架。

主要贡献

  • 提出了异构记忆提取任务和BEHEMOTH基准
  • 分析了现有自进化prompt优化框架在异构任务中的局限性
  • 提出了基于聚类的自进化策略CluE

方法论

提出CluE框架,通过聚类将训练样本分组,独立分析每个簇,综合跨簇信息更新提取prompt。

原文摘要

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 \textit{heterogeneous memory extraction} task and introduce \textbf{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 \textbf{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.

标签

LLM Memory Extraction Self-Evolving Heterogeneous Tasks Prompt Optimization

arXiv 分类

cs.CL