MemCollab: Cross-Model Memory Collaboration via Contrastive Trajectory Distillation
On the map: LLM agents and tool use → Agent memory systems
Abstract (arXiv)
LLM agents increasingly rely on memory mechanisms to reuse knowledge from past problem-solving experiences. However, existing methods typically construct memory for a single agent and reuse it with the same underlying model, tightly coupling stored knowledge to model-specific reasoning styles. In heterogeneous deployments, where agents may be instantiated with backbone models of different sizes, architectures, or specializations, this raises a key question: can a single memory system be shared across agents with different backbone models? We find that naive cross-model memory transfer can degrade performance, because stored memories often entangle task-relevant knowledge with model-specific biases. To address this challenge, we propose MemCollab, a collaborative memory framework that builds shared cross-model memory by contrasting reasoning trajectories generated by different model-based agents on the same task. Through this contrastive process, MemCollab distills abstract reasoning constraints that capture shared task-level invariants while suppressing model-specific artifacts. We further introduce a task-aware retrieval mechanism that conditions memory access on task category, ensuring that only relevant constraints are retrieved at inference time. Experiments on mathematical reasoning and code generation benchmarks show that MemCollab consistently improves both accuracy and inference-time efficiency across diverse agents, including settings with different model families. These results demonstrate that collaboratively constructed cross-model memory can serve as a shared reasoning resource for heterogeneous LLM-based agents.
Key facts. Cited by 7 papers in the map (data through 14 Oct 2026). Placed in Agent memory systems, one of 16 directions in LLM agents and tool use. Builds on 38 of its 41 references that are in the map. First posted on arXiv 24 Mar 2026.
Builds on
38 of 41 references.
Closest ancestors · 7
Same direction, newest first
- 2025ACON: Optimizing Context Compression for Long-horizon LLM Agents1 Oct
- 2025ReasoningBank: Scaling Agent Self-Evolving with Reasoning Memory29 Sep
- 2025Memory-R1: Enhancing Large Language Model Agents to Manage and Utilize Memories via Reinforcement Learning27 Aug
- 2025MEM1: Learning to Synergize Memory and Reasoning for Efficient Long-Horizon Agents18 Jun
- 2025A-MEM: Agentic Memory for LLM Agents17 Feb
- 2025Position: Episodic Memory is the Missing Piece for Long-Term LLM Agents10 Feb
- 2024A Survey on the Memory Mechanism of Large Language Model based Agents21 Apr
Other references · 31
- 2024The Llama 3 Herd of Modelscited by 14,715
- 2021Training Verifiers to Solve Math Word Problemscited by 10,109
- 2020Retrieval-Augmented Generation for Knowledge-Intensive NLP Taskscited by 9,297
- 2021Evaluating Large Language Models Trained on Codecited by 8,320
- 2018Representation Learning with Contrastive Predictive Codingcited by 7,777
- 2021Measuring Mathematical Problem Solving With the MATH Datasetcited by 5,371
- 2023Let's Verify Step by Stepcited by 3,789
- 2021Program Synthesis with Large Language Modelscited by 3,637
- 2023Retrieval-Augmented Generation for Large Language Models: A Surveycited by 2,212
- 2024Qwen2 Technical Reportcited by 2,120
- 2024From Local to Global: A Graph RAG Approach to Query-Focused Summarizationcited by 1,453
- 2023Large Language Models Can Be Easily Distracted by Irrelevant Contextcited by 841
19 more
- 2023MemoryBank: Enhancing Large Language Models with Long-Term Memorycited by 556
- 2019A Theoretical Analysis of Contrastive Unsupervised Representation Learningcited by 530
- 2025LLMs Get Lost In Multi-Turn Conversationcited by 429
- 2024AppWorld: A Controllable World of Apps and People for Benchmarking Interactive Coding Agentscited by 401
- 2022Generate rather than Retrieve: Large Language Models are Strong Context Generatorscited by 298
- 2024G-Retriever: Retrieval-Augmented Generation for Textual Graph Understanding and Question Answeringcited by 223
- 2018An efficient framework for learning sentence representationscited by 198
- 2023Chain-of-Note: Enhancing Robustness in Retrieval-Augmented Language Modelscited by 184
- 2025Dynamic Cheatsheet: Test-Time Learning with Adaptive Memorycited by 163
- 2023Lift Yourself Up: Retrieval-augmented Text Generation with Self Memorycited by 103
- 2024Buffer of Thoughts: Thought-Augmented Reasoning with Large Language Modelscited by 91
- 2023Dense X Retrieval: What Retrieval Granularity Should We Use?cited by 88
- 2025ReasonFlux: Hierarchical LLM Reasoning via Scaling Thought Templatescited by 63
- 2023Learning to Retrieve In-Context Examples for Large Language Modelscited by 60
- 2023TableGPT: Towards Unifying Tables, Nature Language and Commands into One GPTcited by 59
- 2025Distilling LLM Agent into Small Models with Retrieval and Code Toolscited by 47
- 2023Augmented Large Language Models with Parametric Knowledge Guidingcited by 37
- 2021ISEEQ: Information Seeking Question Generation using Dynamic Meta-Information Retrieval and Knowledge Graphscited by 22
- 2023From Classification to Generation: Insights into Crosslingual Retrieval Augmented ICLcited by 17
Built on by
Citing papers per year: 2026: 7.
7 papers.
- 2026SkillGrad: Optimizing Agent Skills Like Gradient Descentcited by 26
- 2026EVOCHAMBER: Test-Time Co-evolution of Multi-Agent System at Individual, Team, and Population Scalescited by 4
- 2026Ace-Skill: Bootstrapping Multimodal Agents with Prioritized and Clustered Evolutioncited by 2
- 2026MeloTune: On-Device Arousal Learning and Peer-to-Peer Mood Coupling for Proactive Music Curationcited by 1
- 2026When Does Memory Help Multi-Trajectory Inference for Tool-Use LLM Agents?cited by 1
- 2026Collaborative Memory for Multi-Agent VLM Systemscited by 0
- 2026MemCo: Memory-Centric Collaboration for Generalizing LLM Agents to Unseen Environmentscited by 0
- 2026MemCo: Memory-Centric Collaboration for Generalizing LLM Agents to Unseen Environments5 Oct 2026
- 2026Collaborative Memory for Multi-Agent VLM Systems15 Sep 2026
- 2026When Does Memory Help Multi-Trajectory Inference for Tool-Use LLM Agents?27 May 2026
- 2026SkillGrad: Optimizing Agent Skills Like Gradient Descent26 May 2026
- 2026EVOCHAMBER: Test-Time Co-evolution of Multi-Agent System at Individual, Team, and Population Scales11 May 2026
- 2026Ace-Skill: Bootstrapping Multimodal Agents with Prioritized and Clustered Evolution9 May 2026
- 2026MeloTune: On-Device Arousal Learning and Peer-to-Peer Mood Coupling for Proactive Music Curation12 Apr 2026
Similar, not cited
Close in wording, no citation either way
- 2026MemWeaver: Weaving Hybrid Memories for Traceable Long-Horizon Agentic Reasoning
- 2026MemAdapter: Fast Alignment across Agent Memory Paradigms via Generative Subgraph Retrieval
- 2026Memory is Reconstructed, Not Retrieved: Graph Memory for LLM Agents
- 2025MemGen: Weaving Generative Latent Memory for Self-Evolving Agents
- 2025Agent KB: Leveraging Cross-Domain Experience for Agentic Problem Solving
- 2026MemMA: Coordinating the Memory Cycle through Multi-Agent Reasoning and In-Situ Self-Evolution
- 2026AMA: Adaptive Memory via Multi-Agent Collaboration
- 2025Enhancing Reasoning with Collaboration and Memory