Search papers, labs, and topics across Lattice.
The paper formalizes the problem of heterogeneous memory extraction for LLM-based assistants, where the relevant information to remember varies across tasks. They introduce BEHEMOTH, a benchmark repurposing 18 datasets to evaluate memory extraction performance across personalization, problem-solving, and agentic tasks. To address the performance degradation of existing self-evolving prompt optimization frameworks in heterogeneous settings, they propose CluE, a cluster-based self-evolving strategy that groups training examples by extraction scenarios and synthesizes cross-cluster insights to update the extraction prompt.
Existing self-evolving prompt optimization frameworks falter when faced with the diverse memory demands of heterogeneous tasks, but a new clustering-based approach, CluE, restores generalization performance.
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.
