CoEvo-Mem: Co-Evolving Retrieval Policy and Memory Bank for LLM Agents

AuthorsBowen Ye, Yongchao Xu, Zhijian Li et al.

arXiv 20262026

TL;DR

CoEvo-Mem co-evolves a Prior-Guided Residual Router with a Typed Relational Memory Graph, reaching 68.33% on GPQA Diamond (+7.50 over MemRL and MemQ).

SharePost on XLinkedIn

Read our summary here, or open the publisher PDF on the next tab.

THE PROBLEM

Retrieval and Memory Evolve in Isolation, Breaking the Feedback Loop

CoEvo-Mem targets the issue that existing methods "typically optimize either memory access ... or memory evolution" while ignoring their mutual feedback loop.

When retrieval and memory updates are optimized separately, long-term LLM agents misalign stored utilities and routing, degrading long-horizon reasoning and cross-session performance.

HOW IT WORKS

CoEvo-Mem — Closed-Loop Co-Evolution of Retrieval and Memory

CoEvo-Mem centers on Self-Routed Query Rewriter, Prior-Guided Residual Router, Q-Weighted Hybrid Retriever, and a Typed Relational Memory Graph updated by Trajectory-Conditioned Memory Update.

You can think of CoEvo-Mem like a CPU with a smart cache: the router chooses which memory channels to read, while the graph reorganizes disk blocks based on usage.

This tight loop lets CoEvo-Mem adapt routing and memory utilities over time, enabling behaviors a fixed context window or static RAG pipeline cannot support.

DIAGRAM

Query-Time Retrieval Flow in CoEvo-Mem

This diagram shows how CoEvo-Mem processes a single query through rewriting, routing, hybrid retrieval, and answer generation.

DIAGRAM

Alternating Co-Evolution Training Schedule

This diagram shows how CoEvo-Mem alternates Router and Memory phases over the training set D_tr.

PROCESS

How CoEvo-Mem Handles a Training Interaction

  1. 01

    Self-Routed Query Rewriter

    CoEvo-Mem uses Self-Routed Query Rewriter to generate dense and sparse rewrites plus a routing prior that conditions the Prior-Guided Residual Router.

  2. 02

    Prior-Guided Residual Router

    CoEvo-Mem feeds rewrites into the Prior-Guided Residual Router, which samples or outputs dense–sparse mixture weights guided by task rewards.

  3. 03

    Q-Weighted Hybrid Retriever

    CoEvo-Mem applies the Q-Weighted Hybrid Retriever over the Typed Relational Memory Graph, fusing dense, sparse, and utility ranks into the exposure set A_t.

  4. 04

    Trajectory-Conditioned Memory Update

    CoEvo-Mem runs Trajectory-Conditioned Memory Update to assign TD-style utility residuals and propagate credit across the Typed Relational Memory Graph.

KEY CONTRIBUTIONS

Key Contributions

  • 01

    Coupled Retrieval–Memory Learning

    CoEvo-Mem formalizes retrieval routing and memory evolution as a coupled learning problem, with the Prior-Guided Residual Router and Typed Relational Memory Graph co-adapted via alternating phases.

  • 02

    Prior-Guided Residual Routing

    CoEvo-Mem introduces Self-Routed Query Rewriter plus Prior-Guided Residual Router, learning only a lightweight residual over the LLM prior using Beta-policy reinforcement learning.

  • 03

    Retrieval-Aligned Relational Memory Evolution

    CoEvo-Mem designs Trajectory-Conditioned Memory Update on a Typed Relational Memory Graph, propagating TD-style utility across Dense, Sparse, and Time edges with up to 10.00 point gains in ablations.

RESULTS

By the Numbers

LLAB success rate

76.00%

+1.56 over MemQ

LiveCodeBench score

55.24%

+3.81 over MemQ

MMMU Pro accuracy

55.49%

+1.06 over MemQ

GPQA Diamond accuracy

68.33%

+7.50 over MemRL and MemQ

Across LLAB, LiveCodeBench, MMMU Pro, and GPQA Diamond, CoEvo-Mem consistently leads the benchmark table, showing that co-evolving routing and memory utilities improves diverse long-term agent tasks.

BENCHMARK

By the Numbers

Across LLAB, LiveCodeBench, MMMU Pro, and GPQA Diamond, CoEvo-Mem consistently leads the benchmark table, showing that co-evolving routing and memory utilities improves diverse long-term agent tasks.

BENCHMARK

Performance comparison across benchmarks

Mean accuracy or success rate (%) for CoEvo-Mem and baselines on LLAB.

BENCHMARK

Ablation study of CoEvo-Mem across different benchmarks

Accuracy (%) of CoEvo-Mem variants on GPQA Diamond.

KEY INSIGHT

The Counterintuitive Finding

On GPQA Diamond, CoEvo-Mem reaches 68.33%, while MemRL and MemQ both plateau at 60.83%, a +7.50 percentage point gap.

This is surprising because MemRL and MemQ already learn memory utilities from reward, yet CoEvo-Mem still gains by jointly adapting routing and relational credit propagation.

WHY IT MATTERS

What this unlocks for the field

CoEvo-Mem shows that retrieval policies and memory banks should be trained as a coupled system, not as independent modules.

Builders can now design long-term agents where query routing, memory valuation, and graph structure co-evolve, enabling robust performance across heterogeneous tasks without fine-tuning the base LLM.

~14 min read← Back to papers

Related papers

BenchmarkAgent Memory

Active Context Compression: Autonomous Memory Management in LLM Agents

Nikhil Verma

· 2026

Focus Agent adds start_focus, complete_focus, a persistent Knowledge block, and an optimized Persistent Bash plus String-Replace Editor scaffold to actively compress context during long software-engineering tasks. On five hard SWE-bench Lite instances against a Baseline ReAct agent, Focus Agent achieves 22.7% token reduction (14.9M → 11.5M) while matching 3/5 = 60% task success.

Questions about this paper?

Paper: CoEvo-Mem: Co-Evolving Retrieval Policy and Memory Bank for LLM Agents

Answers use this explainer on Memory Papers.

Checking…