English

MemBoost: A Memory-Boosted Framework for Cost-Aware LLM Inference

Computation and Language 2026-03-30 v1

Abstract

Large Language Models (LLMs) deliver strong performance but incur high inference cost in real-world services, especially under workloads with repeated or near-duplicate queries across users and sessions. In this work, we propose MemBoost, a memory-boosted LLM serving framework that enables a lightweight model to reuse previously generated answers and retrieve relevant supporting information for cheap inference, while selectively escalating difficult or uncertain queries to a stronger model. Unlike standard retrieval-augmented generation, which primarily grounds a single response, MemBoost is designed for interactive settings by supporting answer reuse, continual memory growth, and cost-aware routing. Experiments across multiple models under simulated workloads show that MemBoost substantially reduces expensive large-model invocations and overall inference cost, while maintaining high answer quality comparable to the strong model baseline.

Keywords

Cite

@article{arxiv.2603.26557,
  title  = {MemBoost: A Memory-Boosted Framework for Cost-Aware LLM Inference},
  author = {Joris Köster and Zixuan Liu and Siavash Khajavi and Zizhan Zheng},
  journal= {arXiv preprint arXiv:2603.26557},
  year   = {2026}
}
R2 v1 2026-07-01T11:41:04.194Z