English

Did You Check the Right Pocket? Cost-Sensitive Store Routing for Memory-Augmented Agents

Artificial Intelligence 2026-03-18 v1 Computation and Language Information Retrieval

Abstract

Memory-augmented agents maintain multiple specialized stores, yet most systems retrieve from all stores for every query, increasing cost and introducing irrelevant context. We formulate memory retrieval as a store-routing problem and evaluate it using coverage, exact match, and token efficiency metrics. On downstream question answering, an oracle router achieves higher accuracy while using substantially fewer context tokens compared to uniform retrieval, demonstrating that selective retrieval improves both efficiency and performance. Our results show that routing decisions are a first-class component of memory-augmented agent design and motivate learned routing mechanisms for scalable multi-store systems. We additionally formalize store selection as a cost-sensitive decision problem that trades answer accuracy against retrieval cost, providing a principled interpretation of routing policies.

Keywords

Cite

@article{arxiv.2603.15658,
  title  = {Did You Check the Right Pocket? Cost-Sensitive Store Routing for Memory-Augmented Agents},
  author = {Madhava Gaikwad},
  journal= {arXiv preprint arXiv:2603.15658},
  year   = {2026}
}

Comments

accepted in ICLR 2026 Workshop on Memory for LLM-Based Agentic Systems