mlsanityData to 7 Oct 2026

Learning What Matters: Supervising Global Context Pruning with Causal Evidence Sets

arXiv 2607.21692 ↗First posted 23 Jul 2026cs.LG cs.CLcited by 0

Not on the map. Why?

Abstract (arXiv)

Pruning a long context means committing to the blocks a model will keep, and the usual selector is distilled from a dense teacher's attention. That assumes attention shows which context the answer depends on. We test the assumption on retrieval tasks where the evidence is known exactly, by masking context and measuring whether the answer changes. Attention and causal dependence disagree. Teachers attend to outdated facts that the answer does not depend on, and they attend differently across training runs that use the same evidence. Selectors trained on that attention copy both failures. On a multi-hop retrieval task, a selector distilled from attention routes at 36% to 98% depending on the training run. The same selector trained on causal evidence sets reaches 99% or better on every run. Dense accuracy does not tell the teachers apart. Masking the frozen teacher recovers the causal sets of these tasks without annotations. Frozen pretrained models show the same conflict, and selectors supervised with known evidence labels beat attention-based eviction through 32B when context must be pruned before the question arrives.

Key facts. No citing papers in the map yet (data through 7 Oct 2026). Builds on 31 of its 33 references that are in the map. First posted on arXiv 23 Jul 2026.

Builds on

31 of 33 references.

References · 31

  1. 2020Longformer: The Long-Document Transformercited by 3,362
  2. 2022Locating and Editing Factual Associations in GPTcited by 2,609
  3. 2023Efficient Streaming Language Models with Attention Sinkscited by 2,115
  4. 2020Big Bird: Transformers for Longer Sequencescited by 1,673
  5. 2020Reformer: The Efficient Transformercited by 1,658
  6. 2024RULER: What's the Real Context Size of Your Long-Context Language Models?cited by 1,270
  7. 2022In-context Learning and Induction Headscited by 930
  8. 2019Attention is not Explanationcited by 916
  9. 2024SnapKV: LLM Knows What You are Looking for Before Generationcited by 848
  10. 2020Quantifying Attention Flow in Transformerscited by 759
  11. 2023Towards Automated Circuit Discovery for Mechanistic Interpretabilitycited by 700
  12. 2019Attention is not not Explanationcited by 629
19 more
  1. 2024Quest: Query-Aware Sparsity for Efficient Long-Context LLM Inferencecited by 487
  2. 2021A Dataset of Information-Seeking Questions and Answers Anchored in Research Paperscited by 475
  3. 2020Efficient Content-Based Sparse Attention with Routing Transformerscited by 469
  4. 2025Native Sparse Attention: Hardware-Aligned and Natively Trainable Sparse Attentioncited by 391
  5. 2019Is Attention Interpretable?cited by 387
  6. 2024MInference 1.0: Accelerating Pre-filling for Long-Context LLMs via Dynamic Sparse Attentioncited by 375
  7. 2024Retrieval Head Mechanistically Explains Long-Context Factualitycited by 219
  8. 2024When Attention Sink Emerges in Language Models: An Empirical Viewcited by 200
  9. 2023Zoology: Measuring and Improving Recall in Efficient Language Modelscited by 188
  10. 2025MoBA: Mixture of Block Attention for Long-Context LLMscited by 188
  11. 2024SeerAttention: Learning Intrinsic Sparse Attention in Your LLMscited by 121
  12. 2024Finding Transformer Circuits with Edge Pruningcited by 58
  13. 2025The Sparse Frontier: Sparse Attention Trade-offs in Transformer LLMscited by 28
  14. 2025Learning to Focus: Causal Attention Distillation via Gradient-Guided Token Pruningcited by 13
  15. 2026ReST-KV: Robust KV Cache Eviction with Layer-wise Output Reconstruction and Spatial-Temporal Smoothingcited by 9
  16. 2026MomentKV: Closing the Directional Gap in KV Cache Eviction for Long-Context Inferencecited by 5
  17. 2026SpotAttention: Plug-In Block-Sparse Routing for Pretrained Long-Context Transformerscited by 4
  18. 2026Quantifying LLM Attention-Head Stability: Implications for Circuit Universalitycited by 2
  19. 2026Routing Absorption in Sparse Attention: Why Random Gates Are Hard to Beatcited by 2

Built on by

Not cited yet.