mlsanityData to 7 Oct 2026

Screening Is Enough

arXiv 2604.01178 ↗First posted 1 Apr 2026cs.LG cs.AI cs.CLcited by 1

Not on the map. Why?

Abstract (arXiv)

We call query--key relevance absolute when its values lie on a fixed bounded scale, depend on neither competing keys nor sequence length, require no sequence-length-dependent calibration, and can all be zero. To realize this notion, we introduce screening, whose explicit threshold transforms bounded query--key similarities into relevance values, enabling exact rejection, empty selection, and direct inspection on a common scale. In a controlled comparison of 12 attention mechanisms on a matched Transformer backbone, only screening maintains both low long-context perplexity and robust retrieval beyond the training context; notably, it does so without inference-time scaling. Building on screening, we introduce Multiscreen, a language-model architecture composed of parallel gated screening tiles. Multiscreen retains these long-context gains while achieving greater parameter efficiency, stronger general zero-shot downstream performance, lower training cost at larger scales, and lower model-side time to first token than Transformer baselines. We further develop a normalization design that keeps Multiscreen training stable even at a learning rate of 1 and show that an adapted version likewise stabilizes Transformer at the same learning rate.

Key facts. Cited by 1 paper in the map (data through 7 Oct 2026). Builds on 58 of its 73 references that are in the map. First posted on arXiv 1 Apr 2026.

Builds on

58 of 73 references.

References · 58

  1. 2017Attention Is All You Needcited by 59,932
  2. 2018BERT: Pre-training of Deep Bidirectional Transformers for Language Understandingcited by 40,204
  3. 2019PyTorch: An Imperative Style, High-Performance Deep Learning Librarycited by 20,160
  4. 2023LLaMA: Open and Efficient Foundation Language Modelscited by 13,958
  5. 2021Training Verifiers to Solve Math Word Problemscited by 9,743
  6. 2020Measuring Massive Multitask Language Understandingcited by 7,768
  7. 2018Think you have Solved Question Answering? Try ARC, the AI2 Reasoning Challengecited by 4,710
  8. 2023Mamba: Linear-Time Sequence Modeling with Selective State Spacescited by 4,616
  9. 2021RoFormer: Enhanced Transformer with Rotary Position Embeddingcited by 4,581
  10. 2019HellaSwag: Can a Machine Really Finish Your Sentence?cited by 4,018
  11. 2022FlashAttention: Fast and Memory-Efficient Exact Attention with IO-Awarenesscited by 3,611
  12. 2023Lost in the Middle: How Language Models Use Long Contextscited by 3,383
38 more
  1. 2020Longformer: The Long-Document Transformercited by 3,362
  2. 2019PIQA: Reasoning about Physical Commonsense in Natural Languagecited by 2,818
  3. 2023FlashAttention-2: Faster Attention with Better Parallelism and Work Partitioningcited by 2,551
  4. 2019BoolQ: Exploring the Surprising Difficulty of Natural Yes/No Questionscited by 2,140
  5. 2018Can a Suit of Armor Conduct Electricity? A New Dataset for Open Book Question Answeringcited by 2,114
  6. 2023Pythia: A Suite for Analyzing Large Language Models Across Training and Scalingcited by 1,734
  7. 2020GLU Variants Improve Transformercited by 1,724
  8. 2020Big Bird: Transformers for Longer Sequencescited by 1,673
  9. 2021Train Short, Test Long: Attention with Linear Biases Enables Input Length Extrapolationcited by 1,085
  10. 2017Sigmoid-Weighted Linear Units for Neural Network Function Approximation in Reinforcement Learningcited by 935
  11. 2022In-context Learning and Induction Headscited by 930
  12. 2016Conditional Image Generation with PixelCNN Decoderscited by 859
  13. 2017Crowdsourcing Multiple Choice Science Questionscited by 742
  14. 2023Extending Context Window of Large Language Models via Positional Interpolationcited by 721
  15. 2022GPT-NeoX-20B: An Open-Source Autoregressive Language Modelcited by 711
  16. 2019Compressive Transformers for Long-Range Sequence Modellingcited by 699
  17. 2016Language Modeling with Gated Convolutional Networkscited by 688
  18. 2014Neural Turing Machinescited by 630
  19. 2023Retentive Network: A Successor to Transformer for Large Language Modelscited by 594
  20. 2023YaRN: Efficient Context Window Extension of Large Language Modelscited by 584
  21. 2025Native Sparse Attention: Hardware-Aligned and Natively Trainable Sparse Attentioncited by 391
  22. 2023Hyena Hierarchy: Towards Larger Convolutional Language Modelscited by 388
  23. 2016From Softmax to Sparsemax: A Sparse Model of Attention and Multi-Label Classificationcited by 337
  24. 2023The Impact of Positional Encoding on Length Generalization in Transformerscited by 319
  25. 2024LongRoPE: Extending LLM Context Window Beyond 2 Million Tokenscited by 302
  26. 2020Query-Key Normalization for Transformerscited by 266
  27. 2025Gated Attention for Large Language Models: Non-linearity, Sparsity, and Attention-Sink-Freecited by 266
  28. 2023Landmark Attention: Random-Access Infinite Context Length for Transformerscited by 206
  29. 2022Transformer Quality in Linear Timecited by 189
  30. 2023Zoology: Measuring and Improving Recall in Efficient Language Modelscited by 188
  31. 2024RedPajama: an Open Dataset for Training Large Language Modelscited by 178
  32. 2019Sparse Sequence-to-Sequence Modelscited by 167
  33. 2024Samba: Simple Hybrid State Space Models for Efficient Unlimited Context Language Modelingcited by 167
  34. 2024Scaling Laws and Compute-Optimal Training Beyond Fixed Training Durationscited by 160
  35. 2019Adaptively Sparse Transformerscited by 154
  36. 2024RetrievalAttention: Accelerating Long-Context LLM Inference via Vector Retrievalcited by 114
  37. 2024Attention Score is not All You Need for Token Importance Indicator in KV Cache Reduction: Value Also Matterscited by 73
  38. 2024Theory, Analysis, and Best Practices for Sigmoid Self-Attentioncited by 62

50 most cited of 58 All 58 (JSON)

Built on by

Cited by year (2026 to 7 Oct)

Citing papers per year: 2026: 1.

1 paper.