REVIEW 3 major objections 5 minor 52 references
Supervising keep-or-drop decisions with future attention lets language models discard most of the KV cache while keeping nearly full reasoning accuracy.
Reviewed by Pith at T0; open to challenge. T0 means a machine referee read the full paper against a public rubric. the ladder, T0–T4 →
T0 review · grok-4.5
2026-07-11 09:23 UTC pith:ELHFPE4V
load-bearing objection Solid systems paper: future-attention boundary supervision plus delayed scoring actually moves fixed-budget KV eviction on hard math, with matched baselines that make the gains believable. the 3 major comments →
KVpop -- Key-Value Cache Compression with Predictive Online Pruning
The pith
A machine-rendered reading of the paper's core claim, the machinery that carries it, and where it could break.
Core claim
Fixed-budget KV eviction can be cast as a supervised prediction problem at the eviction boundary: a scorer trained against each token’s post-window future-attention mass, optionally delayed until near-future context is available, recovers nearly full dense-attention accuracy under aggressive compression and outperforms both heuristic and prior learned eviction baselines on mathematical reasoning.
What carries the argument
Future-attention target plus boundary-aware retention loss: each token’s keep/drop label is defined by the mean attention mass it receives after leaving the protected window (aggregated across query heads sharing a KV head), recovered efficiently by a transposed-attention pass that reuses sparse log-sum-exp normalizers; the scorer is trained with a pairwise logistic loss only against the teacher’s top-k cutoff token, and may use a delayed memory-based (mLSTM) readout that scores the token only when it becomes eligible.
Load-bearing premise
The paper assumes that how much future attention a token receives after leaving the protected window is a good enough proxy for whether keeping that token actually helps the final answer, even when the normalizers used to measure that attention come from the sparse student rather than full dense attention.
What would settle it
Train the same scorers and budgets but replace the future-attention target with a pure recency or random ranking (or with dense-LSE targets that systematically disagree with the sparse-LSE approximation) and check whether pass@1 on AIME/HMMT collapses relative to the reported KVpop numbers under identical compression ratios.
If this is right
- Aggressive fixed KV budgets (75–88 percent compression) can preserve near-teacher math-reasoning accuracy when eviction is supervised by future attention rather than local proxies.
- Delaying the scoring decision until a token leaves a short protected window lets memory-based scorers use near-future context and improves token accuracy over immediate scoring.
- The same learned policy, trained only on math traces, remains competitive on out-of-domain code generation and STEM reasoning without retraining the eviction rule.
- Homogeneous per-head fixed budgets yield more regular sparse patterns and lower long-generation latency than dynamic gates that concentrate capacity in a few heads.
Where Pith is reading between the lines
- If future-attention mass is a transferable utility signal, the same boundary-supervised scorer could be distilled once and reused across related model families without re-collecting dense teacher maps.
- The method suggests that irreversible eviction can approach sparse-retrieval quality on reasoning workloads provided the keep decision is postponed until a short future window is observed.
- Hybrid designs that give a few critical layers denser budgets while keeping most heads under KVpop-style fixed top-k may further close residual quality gaps without returning to linear memory growth.
Editorial analysis
A structured set of objections, weighed in public.
Referee Report
Summary. The paper proposes KVpop, a post-training retrofit for fixed-budget KV-cache eviction in pretrained Transformers. Each head keeps sink tokens, a protected recent window, and a learned long-range top-k set. Scorers are supervised at the eviction boundary by a future-attention mass target (Eqs. 2–3, 5–7), computed via a training-only transposed-attention pass that reuses sparse LSE normalizers and avoids materializing dense S×S maps (Eqs. 8–9, App. D). A delayed mLSTM scorer can defer the keep/drop decision until the protected-window exit, using near-future context. On AIME/HMMT with Qwen3-4B/8B, KVpop and a stateless MLP variant outperform StreamingLLM, TOVA, a trained StreamingLLM+, and parameter-matched DMS under matched budgets (Table 1), with near-teacher OOD results on GPQA-D and LiveCodeBench (Table 2) and favorable latency/VRAM at long generation lengths (Fig. 4).
Significance. KV-cache growth is a central systems bottleneck for long-context decoding; a learned fixed-budget eviction policy that preserves hard reasoning quality would be practically valuable. The work’s main technical contributions—boundary-aware supervision with an efficiently computed future-attention target, and delayed stateful scoring that uses protected-window context before the decision—are concrete and well-motivated relative to heuristic eviction and to DMS-style relaxed gates. Empirically, the matched-budget comparisons (especially StreamingLLM+ isolating pattern adaptation, and DMS isolating the retention objective) and the homogeneous per-head budget that yields cleaner long-generation throughput are strengths. If the reported near-teacher AIME/HMMT retention under 75–88% compression holds under broader evaluation, the method is a meaningful advance for learned eviction retrofits.
major comments (3)
- [Abstract; §1 Contributions; Table 1; §5 Conclusion] Relative-performance numbers are inconsistent across the manuscript and do not always match Table 1. The abstract reports 98%/97% of teacher on Qwen3-4B at 75%/88% compression; §1 Contributions and §5 Conclusion report 95%/94% (and 95%/99% for Qwen3-8B). Table 1 gives Rel. 0.98/0.97 for Qwen3-4B and 1.00/1.00 for Qwen3-8B. These are load-bearing headline claims; please recompute from the Abs. columns, reconcile abstract/intro/conclusion with Table 1, and state the exact aggregation (mean of four pass@1s vs. other).
- [§4.2; Figure 5; Table 1] The delayed-scoring claim is central (“uniquely among learned eviction methods”) but the only direct ablation (Fig. 5) reports token accuracy (+0.2 points after 2k steps), not AIME/HMMT pass@1 or other task metrics under the same budget. Without a task-level delayed vs. non-delayed comparison for the stateful scorer, it is hard to judge how much of Table 1’s KVpop vs. KVpopmlp gap is due to delay rather than recurrence capacity or the reduced top-k used to match memory. Please add (or clearly state the absence of) a task-level ablation.
- [§3.2; Eqs. (8)–(9); Appendix D.3] The future-attention target uses sparse-query LSE as a proxy for dense causal LSE (Eqs. 8–9; App. D.3), and the paper asserts this “matches the dense-LSE target in downstream performance.” That approximation is an axiom of the training signal, yet no main-text or appendix table quantifies target agreement or task delta under dense vs. sparse LSE. A short ablation (or correlation of targets) would make the supervision claim more solid; if already run, report it explicitly.
minor comments (5)
- [§4.1; Table 1] Pass@1 is estimated with 16 rollouts; please report uncertainty (e.g., bootstrap SE or std over seeds) for Table 1 averages, especially where gaps to DMS are a few points.
- [§4.3; Figure 6; Appendix Figure 9] Figure 6 / Appendix Fig. 9 eviction visualizations are informative but qualitative; a simple quantitative breakdown (e.g., retention rates for numeric vs. discourse/operator tokens) would strengthen §4.3.
- [§4 Baselines; Appendix F] Clarify whether StreamingLLM+ and DMS use identical sink/window/top-k (or equivalent B) and the same Nemotron-Math training mixture and KL setup as KVpop; §4 states matching budget and scorer parameter count for DMS, but a one-row hyperparameter table would help reproducibility.
- [§3.1; Eq. (3); Appendix B.1] Notation: r̂ vs. r_tgt and effective scores with decay (Eq. 5) are clear, but Agg_g is only specified as max in the main text with alternatives in App. B; a brief main-text sentence on why max was chosen would help.
- [Throughout; Algorithm 1] Typos / polish: “KVcache” spacing is inconsistent; “themLSTM” often lacks a space after “the”; Algorithm 1 line breaks and the “˜LSE” notation in the PDF are hard to read.
Circularity Check
No significant circularity: ordinary teacher-supervised distillation of a keep/drop scorer, evaluated on held-out task accuracy rather than reconstruction of the training target.
specific steps
-
other
[Sec. 3.2 / Eq. (9) and Appendix D.3]
"Using the dense causal LSE would make this identity exact. Instead, we reuse the sparse query LSE values ˜LSE already returned by the student attention pass... We find empirically that this sparse-LSE approximation matches the dense-LSE target in downstream performance"
The future-attention target used to supervise the scorer is computed with the student's own sparse-attention normalizers rather than independent dense LSE. This is a mild self-reference in the training signal (target depends on the current sparse policy). It is not a tautology of the main claim: the claim is held-out task accuracy under fixed budgets, the paper reports the approximation matches dense-LSE downstream, and baselines/ablations still separate learned ranking from the sparse pattern itself.
full rationale
KVpop trains lightweight scorers with a boundary-aware pairwise loss against a future-attention mass target derived from the dense (or sparse-LSE-approximated) teacher attention, while the base model is updated only by KL distillation to the dense teacher (Algorithm 1; Eqs. 2–3, 5–7). The sparse-LSE reuse (Eq. 9, App. D.3) is an efficiency approximation for the normalizer, not a definition of the claimed result; the paper reports that it matches dense-LSE downstream performance. Evaluation is pass@1 on AIME/HMMT and OOD GPQA-D/LCB under fixed budgets, with ablations (StreamingLLM+ vs learned ranking, DMS under matched budget/params, delayed vs non-delayed mLSTM) that isolate the supervision target from mere pattern adaptation. There is no self-definitional loop, no fitted parameter renamed as a prediction of the same quantity, and no load-bearing uniqueness theorem imported from overlapping authors. Self-citations (e.g., mLSTM background, FlexAttention) supply architecture or tooling, not the central claim. Score 1 only for the minor self-referential sparse-LSE approximation, which is not load-bearing against the task-accuracy claim.
Axiom & Free-Parameter Ledger
free parameters (4)
- per-head long-range top-k budget k (and sinks s, window w) =
s=4, w=256, k∈{2016,4032}
- per-head decay factor γ_h (log-decay range) =
γ_min=0.999, γ_max=0.999999
- scorer and base learning rates / schedule =
scorer 1e-3→8e-5; base 8e-5; 2k steps
- boundary-loss temperature τ and margin/balance weights =
τ=1.0; margin weighting enabled
axioms (4)
- domain assumption Mean future attention mass a key receives after leaving the protected window is a valid supervision signal for keep-or-drop decisions that preserve generation quality.
- ad hoc to paper Sparse-query LSE normalizers from the fixed-budget student attention closely approximate dense causal LSE for the purpose of the future-attention target.
- domain assumption Homogeneous fixed per-head budgets are preferable for GPU execution even if dynamic per-head allocation (as in DMS) could concentrate capacity.
- standard math Standard causal multi-head / grouped-query attention and KL distillation to a dense teacher are valid training substrates for a sparse retrofit.
invented entities (2)
-
Future-attention target with transposed-attention computation
no independent evidence
-
Delayed mLSTM importance scorer
no independent evidence
read the original abstract
Key-value (KV) cache growth is a major bottleneck in autoregressive decoding, as memory and bandwidth scale linearly with context length. Existing KV eviction methods often rely on static heuristics or proxy scores, which poorly track future token utility and cause brittle eviction as relevance shifts. To address this, we introduce KVpop, which learns a fixed-budget KV eviction policy by directly supervising the keep-or-drop decision. The scorer is trained against a novel future-attention target, computed efficiently without materializing dense attention maps. We further introduce a delayed memory-based scorer that, uniquely among learned eviction methods, defers scoring for a fixed number of steps to exploit near-future context. On AIME and HMMT mathematical reasoning, KVpop retains 98% of full-attention performance on Qwen3-4B at 75% KV cache compression and 97% at 88% compression, consistently outperforming established eviction baselines. Qwen3-8B shows even stronger results, reaching near-full teacher performance. These results show that supervising eviction with future-attention signals cuts memory costs while maintaining quality.
Figures
Reference graph
Works this paper leans on
-
[1]
Advances in Neural Information Processing Systems , author =. 2024 , pages =. doi:10.52202/079017-3417 , language =
-
[2]
Neural computation , keywords =
Hochreiter, Sepp and Schmidhuber, J. Neural computation , keywords =
-
[3]
Michael Zhang and Simran Arora and Rahul Chalamala and Benjamin Frederick Spector and Alan Wu and Krithik Ramesh and Aaryan Singhal and Christopher Re , booktitle=. Lo. 2025 , url=
2025
-
[4]
The Twelfth International Conference on Learning Representations , year=
The Hedgehog & the Porcupine: Expressive Linear Attentions with Softmax Mimicry , author=. The Twelfth International Conference on Learning Representations , year=
-
[5]
2023 , eprint=
Gaussian Error Linear Units (GELUs) , author=. 2023 , eprint=
2023
-
[6]
2023 , eprint=
GQA: Training Generalized Multi-Query Transformer Models from Multi-Head Checkpoints , author=. 2023 , eprint=
2023
-
[7]
2025 , eprint=
LoLA: Low-Rank Linear Attention With Sparse Caching , author=. 2025 , eprint=
2025
-
[8]
2025 , eprint=
Olmo 3 , author=. 2025 , eprint=
2025
-
[9]
Wu, Wei and Pan, Zhuoshi and Fu, Kun and Wang, Chao and Chen, Liyi and Bai, Yunchu and Wang, Tianfu and Wang, Zheng and Xiong, Hui. T oken S elect: Efficient Long-Context Inference and Length Extrapolation for LLM s via Dynamic Token-Level KV Cache Selection. Proceedings of the 2025 Conference on Empirical Methods in Natural Language Processing. 2025. doi...
-
[10]
and Soloveychik, Ilya and Kamath, Purushotham , booktitle =
Adnan, Muhammad and Arunkumar, Akhil and Jain, Gaurav and Nair, Prashant J. and Soloveychik, Ilya and Kamath, Purushotham , booktitle =. Keyformer: KV Cache reduction through key tokens selection for Efficient Generative Inference , volume =
-
[11]
Attention
Vaswani, Ashish and Shazeer, Noam and Parmar, Niki and Uszkoreit, Jakob and Jones, Llion and. Attention. 2017 , booktitle =
2017
-
[12]
2023 , pages =
Advances in Neural Information Processing Systems , author =. 2023 , pages =
2023
-
[13]
The Twelfth International Conference on Learning Representations , year=
Efficient Streaming Language Models with Attention Sinks , author=. The Twelfth International Conference on Learning Representations , year=
-
[14]
arXiv preprint arXiv:2512.15489 , year =
Nemotron-Math: Efficient Long-Context Distillation of Mathematical Reasoning from Multi-Mode Supervision , author =. arXiv preprint arXiv:2512.15489 , year =
-
[15]
Forty-second International Conference on Machine Learning , year=
Understanding and Improving Length Generalization in Recurrent Models , author=. Forty-second International Conference on Machine Learning , year=
-
[16]
Inference-
Łańcucki, Adrian and Staniszewski, Konrad and Nawrot, Piotr and Ponti, Edoardo , month = oct, year =. Inference-
-
[17]
Nawrot, Piotr and Li, Robert and Huang, Renjie and Ruder, Sebastian and Marchisio, Kelly and Ponti, Edoardo M. , month = apr, year =. The. doi:10.48550/arXiv.2504.17768 , abstract =
-
[18]
Overcoming
Zhan, Zhihao and Zhao, Jianan and Zhu, Zhaocheng and Tang, Jian , month = oct, year =. Overcoming
-
[19]
Nawrot, Piotr and Łańcucki, Adrian and Chochowski, Marcin and Tarjan, David and Ponti, Edoardo , month = jul, year =. Dynamic. Proceedings of the 41st
-
[20]
Yuhong Li and Yingbing Huang and Bowen Yang and Bharat Venkitesh and Acyr Locatelli and Hanchen Ye and Tianle Cai and Patrick Lewis and Deming Chen , booktitle=. Snap. 2024 , url=
2024
-
[21]
Proceedings of the 41st
Tang, Jiaming and Zhao, Yilong and Zhu, Kan and Xiao, Guangxuan and Kasikci, Baris and Han, Song , month = jul, year =. Proceedings of the 41st
-
[22]
doi:10.48550/arXiv.2512.02556 , abstract =
DeepSeek Team , month = dec, year =. doi:10.48550/arXiv.2512.02556 , abstract =
-
[23]
Kwon, Woosuk and Li, Zhuohan and Zhuang, Siyuan and Sheng, Ying and Zheng, Lianmin and Yu, Cody Hao and Gonzalez, Joseph and Zhang, Hao and Stoica, Ion , month = oct, year =. Efficient. Proceedings of the 29th. doi:10.1145/3600006.3613165 , abstract =
-
[24]
Proceedings of Machine Learning and Systems , author =
Keyformer:. Proceedings of Machine Learning and Systems , author =. 2024 , pages =
2024
-
[25]
doi:10.48550/arXiv.2503.08879 , abstract =
Wang, Guangtao and Upasani, Shubhangi and Wu, Chen and Gandhi, Darshan and Li, Jonathan and Hu, Changran and Li, Bo and Thakker, Urmish , month = mar, year =. doi:10.48550/arXiv.2503.08879 , abstract =
-
[26]
Efficient
Xiao, Guangxuan and Tian, Yuandong and Chen, Beidi and Han, Song and Lewis, Mike , month = oct, year =. Efficient
-
[27]
Zeng, Zihao and Lin, Bokai and Hou, Tianqi and Zhang, Hao and Deng, Zhijie , month = apr, year =. In-context. doi:10.48550/arXiv.2410.12876 , abstract =
-
[28]
and Gao, Yifei and Chang, Chi-Chih and Jain, Nilesh and Abdelfattah, Mohamed S
Akhauri, Yash and AbouElhamayed, Ahmed F. and Gao, Yifei and Chang, Chi-Chih and Jain, Nilesh and Abdelfattah, Mohamed S. , month = mar, year =. doi:10.48550/arXiv.2503.07518 , abstract =
-
[29]
, title =
Fenwick, Peter M. , title =. Software---Practice and Experience , volume =. 1994 , doi =
1994
-
[30]
, author=
FlexAttention: A Programming Model for Generating Fused Attention Variants. , author=. Eighth Conference on Machine Learning and Systems , year=
-
[31]
2025 , eprint=
Qwen3 Technical Report , author=. 2025 , eprint=
2025
-
[32]
Longformer: The Long-Document Transformer , author=. arXiv:2004.05150 , year=
Pith/arXiv arXiv 2004
-
[33]
arXiv preprint arXiv:2603.15590 , year=
Effective Distillation to Hybrid xLSTM Architectures , author=. arXiv preprint arXiv:2603.15590 , year=
-
[34]
2024 , booktitle =
Yang, Songlin and Wang, Bailin and Shen, Yikang and Panda, Rameswar and Kim, Yoon , title =. 2024 , booktitle =
2024
-
[35]
arXiv preprint arXiv:2503.13427 , year=
xlstm 7b: A recurrent llm for fast and efficient inference , author=. arXiv preprint arXiv:2503.13427 , year=
-
[36]
arXiv preprint arXiv:2411.04165 , year=
Bio-xLSTM: Generative modeling, representation and in-context learning of biological and chemical sequences , author=. arXiv preprint arXiv:2411.04165 , year=
-
[37]
arXiv preprint arXiv:2410.22391 , year=
A large recurrent action model: xLSTM enables fast inference for robotics tasks , author=. arXiv preprint arXiv:2410.22391 , year=
-
[38]
arXiv preprint arXiv:2406.04303 , year=
Vision-lstm: xlstm as generic vision backbone , author=. arXiv preprint arXiv:2406.04303 , year=
-
[39]
arXiv preprint arXiv:2505.23719 , year=
Tirex: Zero-shot forecasting across long and short horizons with enhanced in-context learning , author=. arXiv preprint arXiv:2505.23719 , year=
-
[40]
2020 , booktitle =
Katharopoulos, Angelos and Vyas, Apoorv and Pappas, Nikolaos and Fleuret, François , title =. 2020 , booktitle =
2020
-
[41]
arXiv preprint arXiv:2510.00636 , year=
Expected attention: Kv cache compression by estimating attention from future queries distribution , author=. arXiv preprint arXiv:2510.00636 , year=
-
[42]
arXiv preprint arXiv:2412.06464 , year=
Gated delta networks: Improving mamba2 with delta rule , author=. arXiv preprint arXiv:2412.06464 , year=
-
[43]
2024 , booktitle =
Gu, Albert and Dao, Tri , title =. 2024 , booktitle =
2024
-
[44]
Transformers are Multi-State RNN s
Oren, Matanel and Hassid, Michael and Yarden, Nir and Adi, Yossi and Schwartz, Roy. Transformers are Multi-State RNN s. Proceedings of the 2024 Conference on Empirical Methods in Natural Language Processing. 2024. doi:10.18653/v1/2024.emnlp-main.1043
-
[45]
Thirty-seventh Conference on Neural Information Processing Systems , year=
Random-Access Infinite Context Length for Transformers , author=. Thirty-seventh Conference on Neural Information Processing Systems , year=
-
[46]
2025 , eprint=
Native Sparse Attention: Hardware-Aligned and Natively Trainable Sparse Attention , author=. 2025 , eprint=
2025
-
[47]
Bowman , booktitle=
David Rein and Betty Li Hou and Asa Cooper Stickland and Jackson Petty and Richard Yuanzhe Pang and Julien Dirani and Julian Michael and Samuel R. Bowman , booktitle=. 2024 , url=
2024
-
[48]
The Thirteenth International Conference on Learning Representations , year=
LiveCodeBench: Holistic and Contamination Free Evaluation of Large Language Models for Code , author=. The Thirteenth International Conference on Learning Representations , year=
-
[49]
Zefan Cai and Yichi Zhang and Bofei Gao and Yuliang Liu and Yucheng Li and Tianyu Liu and Keming Lu and Wayne Xiong and Yue Dong and Junjie Hu and Wen Xiao , booktitle=. Pyramid. 2025 , url=
2025
-
[50]
2025 , eprint=
DeepSeek-V3.2: Pushing the Frontier of Open Large Language Models , author=. 2025 , eprint=
2025
-
[51]
BERT Rediscovers the Classical NLP Pipeline
Tenney, Ian and Das, Dipanjan and Pavlick, Ellie. BERT Rediscovers the Classical NLP Pipeline. Proceedings of the 57th Annual Meeting of the Association for Computational Linguistics. 2019. doi:10.18653/v1/P19-1452
-
[52]
Gonzalez and Clark Barrett and Ying Sheng , booktitle=
Lianmin Zheng and Liangsheng Yin and Zhiqiang Xie and Chuyue Sun and Jeff Huang and Cody Hao Yu and Shiyi Cao and Christos Kozyrakis and Ion Stoica and Joseph E. Gonzalez and Clark Barrett and Ying Sheng , booktitle=. 2024 , url=
2024
discussion (0)
Sign in with ORCID, Apple, or X to comment. Anyone can read and Pith papers without signing in.