mlsanityData to 7 Oct 2026

EntroPrefill: Renyi-Guided Context Pruning with Conditional Stability Guarantees for Retrieval-Augmented Generation

arXiv 2610.09757 ↗First posted 7 Oct 2026cs.LG stat.MLcited by 0

On the map: Retrieval-augmented generation → Generation and context integration

Abstract (arXiv)

Mid-prefill pruning can reduce the sequence processed by deeper transformer layers, but attention concentration alone does not certify that discarded context is dispensable. We formulate EntroPrefill as a Renyi-guided proposal mechanism coupled to explicit constraints on discarded attention mass. Sink-isolated, regularized head pooling respects grouped-query attention while exposing a quantitative trade-off between specialization and worst-head coverage. We derive a mixture-to-head deletion envelope, a computable upper bound on feasible token removal, and a finite-sample observer guarantee that remains valid when the pruning layer is selected adaptively. We then establish a conditional transformer perturbation bound with explicit sufficient Lipschitz constants and a first-token decision-margin corollary. A counterexample shows why shallow observations alone cannot imply an unconditional future-output guarantee. The systems analysis distinguishes query-head unions, physical page allocation, and KV-transfer payload, and gives an arithmetic break-even condition for pruning. This manuscript is theoretical in scope: it defines the procedure, its assumptions, and its formal limits, but does not report measured acceleration or task-accuracy preservation. Experiments are reserved for subsequent validation of the assumptions, approximation tightness, and end-to-end resource trade-offs.

Key facts. No citing papers in the map yet (data through 7 Oct 2026). Placed in Generation and context integration, one of 15 directions in Retrieval-augmented generation. Builds on 15 of its 18 references that are in the map. First posted on arXiv 7 Oct 2026.

Builds on

15 of 18 references.

Closest ancestors · 1

Same direction, newest first

  1. 2025AttentionRAG: Attention-Guided Context Pruning in Retrieval-Augmented Generation13 Mar

Other references · 14

  1. 2023Efficient Memory Management for Large Language Model Serving with PagedAttentioncited by 5,767
  2. 2023FlashAttention-2: Faster Attention with Better Parallelism and Work Partitioningcited by 2,551
  3. 2023Efficient Streaming Language Models with Attention Sinkscited by 2,115
  4. 2023GQA: Training Generalized Multi-Query Transformer Models from Multi-Head Checkpointscited by 1,311
  5. 2024SnapKV: LLM Knows What You are Looking for Before Generationcited by 848
  6. 2023H2O: Heavy-Hitter Oracle for Efficient Generative Inference of Large Language Modelscited by 805
  7. 2024DistServe: Disaggregating Prefill and Decoding for Goodput-optimized Large Language Model Servingcited by 595
  8. 2024Taming Throughput-Latency Tradeoff in LLM Inference with Sarathi-Servecited by 507
  9. 2023LLMLingua: Compressing Prompts for Accelerated Inference of Large Language Modelscited by 279
  10. 2024Ada-KV: Optimizing KV Cache Eviction by Adaptive Budget Allocation for Efficient LLM Inferencecited by 230
  11. 2024Retrieval Head Mechanistically Explains Long-Context Factualitycited by 219
  12. 2024LazyLLM: Dynamic Token Pruning for Efficient Long Context LLM Inferencecited by 70
2 more
  1. 2025SlimInfer: Accelerating Long-Context LLM Inference via Dynamic Token Pruningcited by 13
  2. 2026Adaptive Layer Selection for Layer-Wise Token Pruning in LLM Inferencecited by 3

Built on by

Not cited yet.

Similar, not cited

Close in wording, no citation either way

  1. 2026SpotAttention: Plug-In Block-Sparse Routing for Pretrained Long-Context Transformers
  2. 2026Retrieval-Aware Distillation for Transformer-SSM Hybrids
  3. 2026Rethinking Retrieval-Augmentation as Synthesis: A Query-Aware Context Merging Approach
  4. 2026A Self-Pruning Transformer: Extreme KV-Cache Compression with Universal Attention
  5. 2026Simplified Sparse Attention via Gist Tokens
  6. 2026From Saliency to Discriminability: Rank-Preserving Visual Token Pruning for VLM Rerankers
  7. 2025Towards Lossless Token Pruning in Late-Interaction Retrieval Models
  8. 2024A Training-free Sub-quadratic Cost Transformer Model Serving Framework With Hierarchically Pruned Attention