BudgetMem: Training-Free Selective Memory for Cost-Efficient Long-Context Processing in Language Models
Not on the map. Why?
Abstract (arXiv)
Processing long documents with large language models (LLMs) is expensive: a single query over a 100K-token document can cost from tens of cents to over a dollar in API fees, depending on the model, and memory grows linearly with context length. We introduce BudgetMem, a training-free architecture that keeps only high-salience content under an explicit memory budget. Unlike token-level neural compressors such as LLMLingua, BudgetMem makes chunk-level keep-or-discard decisions from interpretable features: entity density, TF-IDF importance, position, numerical density, discourse markers, and question presence. Across four benchmarks, BudgetMem matches the uncompressed baseline on template-generated structured documents (F1 = 0.859 vs. 0.855) while discarding 70% of chunks. As a pre-retrieval filter on the same documents it outperforms LLMLingua-2 by a wide margin (0.859 vs. 0.554), because token-level compression destroys the phrasal structure that retrieval depends on; the advantage holds across model scale and family (Llama-3.2-3B, Qwen2.5-7B), under dense retrieval (0.918 vs. 0.530), and in LLMLingua-2's intended direct-feed mode (0.797 vs. 0.720). On real NLP research papers (Qasper), BudgetMem is competitive but not free, trailing the full baseline by 7.4% and LLMLingua-2 by 5.5% relative F1 at a 72% storage reduction. Removing discourse markers or numerical density hurts most, although TF-IDF alone scores as well on template documents. The pipeline needs no trained models or GPU for compression and runs on a 10/month Colab instance.
Key facts. Cited by 1 paper in the map (data through 14 Oct 2026). Builds on 45 of its 53 references that are in the map. First posted on arXiv 7 Nov 2025.
Builds on
45 of 53 references.
References · 45
- 2017Attention Is All You Needcited by 60,361
- 2023GPT-4 Technical Reportcited by 18,396
- 2024The Llama 3 Herd of Modelscited by 14,715
- 2021LoRA: Low-Rank Adaptation of Large Language Modelscited by 14,380
- 2023LLaMA: Open and Efficient Foundation Language Modelscited by 14,132
- 2015Distilling the Knowledge in a Neural Networkcited by 10,324
- 2020Retrieval-Augmented Generation for Knowledge-Intensive NLP Taskscited by 9,297
- 2023Efficient Memory Management for Large Language Model Serving with PagedAttentioncited by 5,944
- 2021RoFormer: Enhanced Transformer with Rotary Position Embeddingcited by 4,703
- 2020Dense Passage Retrieval for Open-Domain Question Answeringcited by 3,824
- 2022FlashAttention: Fast and Memory-Efficient Exact Attention with IO-Awarenesscited by 3,673
- 2020Longformer: The Long-Document Transformercited by 3,413
33 more
- 2017Billion-scale similarity search with GPUscited by 2,497
- 2020REALM: Retrieval-Augmented Language Model Pre-Trainingcited by 2,141
- 2020Transformers are RNNs: Fast Autoregressive Transformers with Linear Attentioncited by 2,022
- 2022GPTQ: Accurate Post-Training Quantization for Generative Pre-trained Transformerscited by 1,919
- 2019Transformer-XL: Attentive Language Models Beyond a Fixed-Length Contextcited by 1,850
- 2020Big Bird: Transformers for Longer Sequencescited by 1,684
- 2020Rethinking Attention with Performerscited by 1,601
- 2018Know What You Don't Know: Unanswerable Questions for SQuADcited by 1,554
- 2019Generating Long Sequences with Sparse Transformerscited by 1,546
- 2021Improving language models by retrieving from trillions of tokenscited by 1,293
- 2020Leveraging Passage Retrieval with Generative Models for Open Domain Question Answeringcited by 1,210
- 2021Train Short, Test Long: Attention with Linear Biases Enables Input Length Extrapolationcited by 1,103
- 2025A-MEM: Agentic Memory for LLM Agentscited by 1,063
- 2022Atlas: Few-shot Learning with Retrieval Augmented Language Modelscited by 952
- 2019Generalization through Memorization: Nearest Neighbor Language Modelscited by 800
- 2023REPLUG: Retrieval-Augmented Black-Box Language Modelscited by 792
- 2023Extending Context Window of Large Language Models via Positional Interpolationcited by 740
- 2019Compressive Transformers for Long-Range Sequence Modellingcited by 726
- 2020Long Range Arena: A Benchmark for Efficient Transformerscited by 700
- 2014Neural Turing Machinescited by 644
- 2017The NarrativeQA Reading Comprehension Challengecited by 597
- 2022Efficiently Scaling Transformer Inferencecited by 539
- 2021A Dataset of Information-Seeking Questions and Answers Anchored in Research Paperscited by 486
- 2023LongLLMLingua: Accelerating and Enhancing LLMs in Long Context Scenarios via Prompt Compressioncited by 453
- 2014Memory Networkscited by 363
- 2023Adapting Language Models to Compress Contextscited by 356
- 2024LLMLingua-2: Data Distillation for Efficient and Faithful Task-Agnostic Prompt Compressioncited by 348
- 2023RECOMP: Improving Retrieval-Augmented LMs with Compression and Selective Augmentationcited by 256
- 2022Memorizing Transformerscited by 226
- 2023Compressing Context to Enhance Inference Efficiency of Large Language Modelscited by 189
- 2025MIRIX: Multi-Agent Memory System for LLM-Based Agentscited by 173
- 2024Larimar: Large Language Models with Episodic Memory Controlcited by 53
- 2021∞-former: Infinite Memory Transformercited by 7
Built on by
Citing papers per year: 2026: 1.
1 paper.