S4R: Selective Sampling, Subspaces, and Sparse Reconstruction for Compressed Long-Context KV Caching

Jialong Han, You Wu, Kewei Tu


Abstract
The growth of context window lengths in Large Language Models (LLMs) significantly enhances their long-context capabilities but incurs prohibitive memory costs due to the Key-Value (KV) cache. Although low-rank compression of KV cache is a promising remedy, existing methods face a dilemma: offline approaches depend on external calibration data, whereas online approaches incur substantial compute for full-prompt decomposition and reconstruction. In this paper, we propose **S4R**, which builds low-rank subspaces from selectively sampled tokens and computes attention over a sparsely reconstructed KV representation. S4R uses prompt-aware initialization to build initial key/value bases from a representative prompt subset, trading off calibration-data dependence against prefilling cost. Because fully reconstructing the cache at every decoding step is prohibitively expensive and hurts throughput, we further adopt sparse reconstruction to retain only informative positions during decoding. Extensive experiments on LongBench and RULER with Llama and Qwen model families show that S4R achieves up to 5x KV compression with near full-cache accuracy, combining the efficiency of fixed compression with the adaptability of prompt-dependent methods. Our code is publicly available at https://github.com/Draconis98/S4R
Anthology ID:
2026.aacl-long.137
Volume:
Proceedings of the 15th International Joint Conference on Natural Language Processing and the 5th Conference of the Asia-Pacific Chapter of the Association for Computational Linguistics (Volume 1: Long Papers)
Month:
November
Year:
2026
Address:
Hengqin, China
Editors:
Yue Zhang, Ruifeng Xu, Ayu Purwarianti, Tong Rong, Jun Suzuki
Venues:
AACL | IJCNLP
SIG:
Publisher:
Association for Computational Linguistics
Note:
Pages:
2997–3017
Language:
URL:
https://aclanthology.org/2026.aacl-long.137/
DOI:
Bibkey:
Cite (ACL):
Jialong Han, You Wu, and Kewei Tu. 2026. S4R: Selective Sampling, Subspaces, and Sparse Reconstruction for Compressed Long-Context KV Caching. In Proceedings of the 15th International Joint Conference on Natural Language Processing and the 5th Conference of the Asia-Pacific Chapter of the Association for Computational Linguistics (Volume 1: Long Papers), pages 2997–3017, Hengqin, China. Association for Computational Linguistics.
Cite (Informal):
S4R: Selective Sampling, Subspaces, and Sparse Reconstruction for Compressed Long-Context KV Caching (Han et al., AACL-IJCNLP 2026)
Copy Citation:
PDF:
https://aclanthology.org/2026.aacl-long.137.pdf