AURA-KV: Adaptive Subspace Rank Approximation for Efficient Long-Context LLM Inference

Jincheng Xie, Hao Shi, Zhengyi Jiang, Zhongyi Huang, Linqi Song, Hanxu Hou


Abstract
KV caching is essential for efficient autoregressive decoding, but its memory footprint grows linearly with context length and quickly becomes a bottleneck for long-context large language models (LLMs). We study whether feature-dimension compression can serve as an orthogonal second stage for KV-cache reduction, and introduce AURA-KV, an adaptive-rank subspace approximation method for KV caches. AURA-KV is training-free: it decomposes post-RoPE key and value caches into compact bases and coefficients, supports incremental decoding, and allocates rank adaptively across layers under a global memory budget. Because it compresses the feature dimension rather than selecting tokens or lowering precision, it can be combined with token eviction and quantization methods. Across LongBench, RULER, and InfiniteBench, AURA-KV preserves long-context performance on Llama-3-8B, Mistral-7B, and LongChat-7B under moderate compression and degrades gracefully under more aggressive budgets. In latency measurements, AURA-KV improves per-layer decoding speed by up to 2.75 ×, and by up to 3.23 × when combined with token-level cache reduction, with a small one-time prefill overhead.
Anthology ID:
2026.emnlp-main.2250
Volume:
Proceedings of the 2026 Conference on Empirical Methods in Natural Language Processing
Month:
October
Year:
2026
Address:
Budapest, Hungary
Editors:
Hung-Yi Lee, Sunipa Dev, Jessy Li, Desmond Elliott
Venue:
EMNLP
SIG:
SIGDAT
Publisher:
Association for Computational Linguistics
Note:
Pages:
51179–51203
Language:
URL:
https://aclanthology.org/2026.emnlp-main.2250/
DOI:
Bibkey:
Cite (ACL):
Jincheng Xie, Hao Shi, Zhengyi Jiang, Zhongyi Huang, Linqi Song, and Hanxu Hou. 2026. AURA-KV: Adaptive Subspace Rank Approximation for Efficient Long-Context LLM Inference. In Proceedings of the 2026 Conference on Empirical Methods in Natural Language Processing, pages 51179–51203, Budapest, Hungary. Association for Computational Linguistics.
Cite (Informal):
AURA-KV: Adaptive Subspace Rank Approximation for Efficient Long-Context LLM Inference (Xie et al., EMNLP 2026)
Copy Citation:
PDF:
https://aclanthology.org/2026.emnlp-main.2250.pdf