arXiv AI

Where Should a Document Live: Context, Representations, or Parameters?

arXiv Machine Learning
Oct 15

Latent Compaction with Test-Time Training

Latent Compaction is a test‑time training method that learns a compact, document‑specific key–value cache for large‑context language models. It initializes cache entries using reconstruction importance, teacher attention, and key‑space diversity, then fits attention outputs and log‑normalizers to the full cache before jointly optimizing the entire cache via end‑to‑end distillation. Experiments on RULER, LongBench, and Phonebook Lookup show significant accuracy gains over budget‑matched compression baselines, achieving 96.7% accuracy at 32× compression on RULER and transferring well across model scales and families.

By Yingfa Chen, Xingyu Shen, Chaojun Xiao, Xu Han
arXiv Computation and Language
Sep 22

BudgetMem: Training-Free Selective Memory for Cost-Efficient Long-Context Processing in Language Models

arXiv:2511. 04919v3 Announce Type: replace Abstract: 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.

By Chandra Vamsi Krishna Alla, Harish Naidu Gaddam, Manohar Kommi, Sheikh Nazib Ahmed
arXiv Machine Learning
Oct 9

Compile the Table: Query-Calibrated Operator Compression for Tabular In-Context Learning

The paper introduces QCOC, a method for compressing the key‑value cache used in tabular in‑context learning. QCOC clusters in‑context example states into joint‑KV prototypes, calibrates them against query vectors, and preserves multiplicities to maintain accuracy while achieving high compression. Experiments on 64 OpenML‑CC18 datasets and seven long tables show QCOC outperforms other compression and retrieval methods, achieving up to a 10.5× compression ratio and 508× speed‑up over dynamic retrieval baselines while staying within 0.23 percentage points of full‑context accuracy.

By Xu Zhao, Jiaming Zhao, Bin Zhao, Yong Yang
arXiv Machine Learning
5d ago

CacheGraft: Enhancing Retrieval-Augmented Generation via Latent-Space Document Enrichment

CacheGraft is a retrieval‑augmented generation system that enriches a small language model’s latent space by injecting fused key‑value cache representations of corpus passages, encoded offline by a larger model. The approach eliminates the need to run the larger model at inference time, and on five QA benchmarks it improves answer quality in 29 of 30 comparisons, with 17 statistically significant gains. These results demonstrate that latent‑space document enrichment can boost RAG performance, especially when only a small model can be deployed.

By Vihan Lakshman, Peter Baile Chen, Zhuohan Gu, Michael Cafarella
arXiv AI
Jul 8

Benchmarking KV-Cache Optimizations across Task Quality and System Performance for Long-Context Serving

arXiv:2607. 05399v1 Announce Type: cross Abstract: Large language model serving is increasingly limited by KV-cache growth under long-context workloads, yet existing KV-cache compression techniques are difficult to compare because they were evaluated on different models, tasks, budgets, and serving stacks.

By Nikita Agrawal, Ruben Mayer
arXiv Machine Learning
5d ago

LowRAM: Low-Rank Keys and Amortised Values for Composable KV Cache Compression at Test-Time

LowRAM introduces a two‑stage compression scheme for the KV cache used in autoregressive decoding. It first amortises values by predicting them from keys with a lightweight per‑head predictor, storing only residuals for poorly reconstructed tokens, achieving roughly 4× near‑lossless value compression. Then it applies cross‑layer low‑rank key compression, resulting in an asymmetric pipeline that can boost peak decode throughput by up to 5× while maintaining near‑lossless performance, and can be further combined with quantisation and eviction for extreme compression ratios.

By Teresa Delgado, Hugo Hazard, Martin Benfeghoul, Jun Wang, Zafeirios Fountas