MemCollab: Cross-Agent Memory Collaboration via Contrastive Trajectory Distillation

Yurui Chang, Yiran Wu, Qingyun Wu, Lu Lin
3/24/2026
cs.AIcs.LG

No confident code implementation found for this paper.

Abstract

Large language model (LLM)-based agents rely on memory mechanisms to reuse knowledge from past problem-solving experiences. Existing approaches typically construct memory in a per-agent manner, tightly coupling stored knowledge to a single model's reasoning style. In modern deployments with heterogeneous agents, a natural question arises: can a single memory system be shared across different models? We found that naively transferring memory between agents often degrades performance, as such memory entangles task-relevant knowledge with agent-specific biases. To address this challenge, we propose MemCollab, a collaborative memory framework that constructs agent-agnostic memory by contrasting reasoning trajectories generated by different agents on the same task. This contrastive process distills abstract reasoning constraints that capture shared task-level invariants while suppressing agent-specific artifacts. We further introduce a task-aware retrieval mechanism that conditions memory access on task category, ensuring that only relevant constraints are used at inference time. Experiments on mathematical reasoning and code generation benchmarks demonstrate that MemCollab consistently improves both accuracy and inference-time efficiency across diverse agents, including cross-modal-family settings. Our results show that the collaboratively constructed memory can function as a shared reasoning resource for diverse LLM-based agents.

View on arXivView PDF

Code Implementations

No confident code match yet

We couldn't find an author-owned or strongly-evidenced community implementation for this paper. Any repos shown below are weak matches — verify before relying on them.

No code implementations found yet.

Know of an implementation? Let us know in the comments below!

Similar Papers

Cite this paper

@article{chang2026memcollab,
  title  = {MemCollab: Cross-Agent Memory Collaboration via Contrastive Trajectory Distillation},
  author = {Yurui Chang and Yiran Wu and Qingyun Wu and Lu Lin},
  year   = {2026},
  eprint = {2603.23234},
  archivePrefix = {arXiv},
  url    = {http://arxiv.org/abs/2603.23234v1}
}

Add a badge to your README

Shows whether official code is linked for this paper.

Code of Paper badge
[![Code of Paper](https://codeofpaper.com/badge/2603.23234.svg)](https://codeofpaper.com/paper/2603.23234)

Discussion