mlsanityData to 7 Oct 2026

CoWindow Attention: Full Causal Coverage Is a Collective Property

arXiv 2609.32704 ↗First posted 26 Sep 2026cs.AIcited by 0

Not on the map. Why?

Abstract (arXiv)

FullAttn repeatedly exposes the complete causal history to every attention head, creating substantial redundant computation and memory traffic even with IO-efficient dense kernels. We introduce CoWA, a structured attention architecture that distributes access to the causal history across KV heads. All heads share near-diagonal and prefix-sink windows, while complementary long-range windows partition the remaining history. Their union provides full causal coverage although each head attends sparsely to distant tokens. This position-defined attention pattern requires no learned router or indexer, is used consistently during training and inference, and aligns with KV-head tensor parallelism. A window-matched ablation at 8K isolates the effect of complementary long-range allocation: CoWA with 100% collective coverage reaches 89.73% accuracy, compared with 89.97% for FullAttn, while duplicated long-range windows perform substantially worse. Across a broader controlled associative-recall comparison with matched token budgets, CoWA closely tracks FullAttn as the context grows, whereas other sparse patterns lose a substantial fraction of the associations. In an attention-operator benchmark at 128K tokens with tensor parallelism, CoWA reduces forward and backward latency during training by 7.4x and 8.6x and decoding latency during inference by 3.0x over FullAttn. Its per-rank peak operator memory matches FullAttn during training and is 7.6x lower during decoding. Across scaling-law training from 0.6B to 14B parameters, CoWA closely tracks FullAttn in perplexity while reducing total training FLOPs. The resulting 14B models and 32B models from separate continued training achieve comparable knowledge, reasoning, and long-context retrieval scores to FullAttn. These results show that full causal coverage can be a collective property of the head ensemble rather than a duplicated property of every head.

Key facts. No citing papers in the map yet (data through 7 Oct 2026). Builds on 51 of its 67 references that are in the map. First posted on arXiv 26 Sep 2026.

Builds on

51 of 67 references.

References · 51

  1. 2017Attention Is All You Needcited by 59,932
  2. 2021Training Verifiers to Solve Math Word Problemscited by 9,743
  3. 2020Measuring Massive Multitask Language Understandingcited by 7,768
  4. 2020Scaling Laws for Neural Language Modelscited by 6,411
  5. 2021Measuring Mathematical Problem Solving With the MATH Datasetcited by 5,203
  6. 2018Think you have Solved Question Answering? Try ARC, the AI2 Reasoning Challengecited by 4,710
  7. 2019HellaSwag: Can a Machine Really Finish Your Sentence?cited by 4,018
  8. 2022FlashAttention: Fast and Memory-Efficient Exact Attention with IO-Awarenesscited by 3,611
  9. 2023Generative Agents: Interactive Simulacra of Human Behaviorcited by 3,500
  10. 2020Longformer: The Long-Document Transformercited by 3,362
  11. 2019PIQA: Reasoning about Physical Commonsense in Natural Languagecited by 2,818
  12. 2023GPQA: A Graduate-Level Google-Proof Q&A Benchmarkcited by 2,748
38 more
  1. 2018Can a Suit of Armor Conduct Electricity? A New Dataset for Open Book Question Answeringcited by 2,114
  2. 2024Scaling LLM Test-Time Compute Optimally can be More Effective than Scaling Model Parameterscited by 1,928
  3. 2019Megatron-LM: Training Multi-Billion Parameter Language Models Using Model Parallelismcited by 1,829
  4. 2022Challenging BIG-Bench Tasks and Whether Chain-of-Thought Can Solve Themcited by 1,767
  5. 2017Decoupled Weight Decay Regularizationcited by 1,722
  6. 2020Big Bird: Transformers for Longer Sequencescited by 1,673
  7. 2019Generating Long Sequences with Sparse Transformerscited by 1,523
  8. 2023GQA: Training Generalized Multi-Query Transformer Models from Multi-Head Checkpointscited by 1,311
  9. 2024RULER: What's the Real Context Size of Your Long-Context Language Models?cited by 1,270
  10. 2024SnapKV: LLM Knows What You are Looking for Before Generationcited by 848
  11. 2023H2O: Heavy-Hitter Oracle for Efficient Generative Inference of Large Language Modelscited by 805
  12. 2023AGIEval: A Human-Centric Benchmark for Evaluating Foundation Modelscited by 701
  13. 2023YaRN: Efficient Context Window Extension of Large Language Modelscited by 584
  14. 2024FlashAttention-3: Fast and Accurate Attention with Asynchrony and Low-precisioncited by 507
  15. 2024Quest: Query-Aware Sparsity for Efficient Long-Context LLM Inferencecited by 487
  16. 2025Native Sparse Attention: Hardware-Aligned and Natively Trainable Sparse Attentioncited by 391
  17. 2023Effective Long-Context Scaling of Foundation Modelscited by 301
  18. 2024DuoAttention: Efficient Long-Context LLM Inference with Retrieval and Streaming Headscited by 283
  19. 2024CodeAgent: Enhancing Code Generation with Tool-Integrated Agent Systems for Real-World Repo-level Coding Challengescited by 278
  20. 2025FlashInfer: Efficient and Customizable Attention Engine for LLM Inference Servingcited by 248
  21. 2024When Attention Sink Emerges in Language Models: An Empirical Viewcited by 200
  22. 2024InfLLM: Training-Free Long-Context Extrapolation for LLMs with an Efficient Context Memorycited by 192
  23. 2023Zoology: Measuring and Improving Recall in Efficient Language Modelscited by 188
  24. 2025MoBA: Mixture of Block Attention for Long-Context LLMscited by 188
  25. 2024Flex Attention: A Programming Model for Generating Optimized Attention Kernelscited by 180
  26. 2024Scaling Laws and Compute-Optimal Training Beyond Fixed Training Durationscited by 160
  27. 2018Online normalizer calculation for softmaxcited by 154
  28. 2025Why do LLMs attend to the first token?cited by 122
  29. 2024SeerAttention: Learning Intrinsic Sparse Attention in Your LLMscited by 121
  30. 2025FlexPrefill: A Context-Aware Sparse Attention Mechanism for Efficient Long-Sequence Inferencecited by 121
  31. 2024ClusterKV: Manipulating LLM KV Cache in Semantic Space for Recallable Compressioncited by 65
  32. 2024Active-Dormant Attention Heads: Mechanistically Demystifying Extreme-Token Phenomena in LLMscited by 54
  33. 2025InfLLM-V2: Dense-Sparse Switchable Attention for Seamless Short-to-Long Adaptationcited by 31
  34. 2024HashAttention: Semantic Sparsity for Faster Inferencecited by 23
  35. 2025Sliding Window Attention Training for Efficient Large Language Modelscited by 21
  36. 2024Beyond Probabilities: Unveiling the Misalignment in Evaluating Large Language Modelscited by 19
  37. 2025Identifying and Evaluating Inactive Heads in Pretrained LLMscited by 8
  38. 2025Making Every Head Count: Sparse Attention Without the Speed-Performance Trade-offcited by 6

50 most cited of 51 All 51 (JSON)

Built on by

Not cited yet.