SAS: Simple Attention Sparsification via End-to-End Optimization of Context Ranking

  • 类型:arxiv
  • 标识:2609.13141
  • 链接:https://arxiv.org/abs/2609.13141
  • 主分类:engineering
  • 形态:method
  • TLDR:Post-training attention sparsification reduces the quadratic cumulative attention cost of pretrained Transformers by selecting a small set of context units (tokens or blocks) for each query. Existing trainable methods usually use a lightweight selector to score context units, followed by hard Top-K selection that blocks gradients from the language modeling loss. Consequently, these methods commonly distill layer-wise dense attention distributions. Although this encourages the selector to rank context units by dense attention weights in the original model, the ranking is not directly aligned wi
  • 待LLM分类:否
  • 来源文件
  • /inbox/tom/_candidates/2026-09-14-agent-rag-longcontext-candidates.json