Transformer Quality in Linear Time
Not on the map. Why?
Abstract (arXiv)
We revisit the design choices in Transformers, and propose methods to address their weaknesses in handling long sequences. First, we propose a simple layer named gated attention unit, which allows the use of a weaker single-head attention with minimal quality loss. We then propose a linear approximation method complementary to this new layer, which is accelerator-friendly and highly competitive in quality. The resulting model, named FLASH, matches the perplexity of improved Transformers over both short (512) and long (8K) context lengths, achieving training speedups of up to 4.9× on Wiki-40B and 12.1× on PG-19 for auto-regressive language modeling, and 4.8× on C4 for masked language modeling.
Key facts. Cited by 189 papers in the map (data through 7 Oct 2026). Builds on 35 of its 39 references that are in the map. First posted on arXiv 21 Feb 2022.
Builds on
35 of 39 references.
References · 35
- 2017Attention Is All You Needcited by 59,932
- 2018BERT: Pre-training of Deep Bidirectional Transformers for Language Understandingcited by 40,204
- 2020Language Models are Few-Shot Learnerscited by 32,199
- 2020An Image is Worth 16x16 Words: Transformers for Image Recognition at Scalecited by 29,915
- 2019Exploring the Limits of Transfer Learning with a Unified Text-to-Text Transformercited by 14,491
- 2016Layer Normalizationcited by 5,579
- 2021RoFormer: Enhanced Transformer with Rotary Position Embeddingcited by 4,581
- 2016Gaussian Error Linear Units (GELUs)cited by 3,431
- 2020Longformer: The Long-Document Transformercited by 3,362
- 2017TriviaQA: A Large Scale Distantly Supervised Challenge Dataset for Reading Comprehensioncited by 2,895
- 2020Transformers are RNNs: Fast Autoregressive Transformers with Linear Attentioncited by 1,979
- 2019Transformer-XL: Attentive Language Models Beyond a Fixed-Length Contextcited by 1,830
23 more
- 2020GLU Variants Improve Transformercited by 1,724
- 2020Big Bird: Transformers for Longer Sequencescited by 1,673
- 2020Reformer: The Efficient Transformercited by 1,658
- 2020Rethinking Attention with Performerscited by 1,584
- 2019Generating Long Sequences with Sparse Transformerscited by 1,523
- 2020Linformer: Self-Attention with Linear Complexitycited by 1,462
- 2022LaMDA: Language Models for Dialog Applicationscited by 1,218
- 2021Perceiver: General Perception with Iterative Attentioncited by 1,089
- 2017Sigmoid-Weighted Linear Units for Neural Network Function Approximation in Reinforcement Learningcited by 935
- 2021GLaM: Efficient Scaling of Language Models with Mixture-of-Expertscited by 890
- 2018CCNet: Criss-Cross Attention for Semantic Segmentationcited by 717
- 2019Compressive Transformers for Long-Range Sequence Modellingcited by 699
- 2016Language Modeling with Gated Convolutional Networkscited by 688
- 2019Enhancing the Locality and Breaking the Memory Bottleneck of Transformer on Time Series Forecastingcited by 606
- 2017Convolutional Sequence to Sequence Learningcited by 550
- 2020Efficient Content-Based Sparse Attention with Routing Transformerscited by 469
- 2021Pay Attention to MLPscited by 367
- 2019Axial Attention in Multidimensional Transformerscited by 350
- 2021Random Feature Attentioncited by 350
- 2021Primer: Searching for Efficient Transformers for Language Modelingcited by 162
- 2019Transformers without Tears: Improving the Normalization of Self-Attentioncited by 153
- 2021Do Transformer Modifications Transfer Across Implementations and Applications?cited by 87
- 2021Combiner: Full Attention Transformer with Sparse Computation Costcited by 64
Built on by
Citing papers per year: 2022: 25; 2023: 35; 2024: 48; 2025: 41; 2026: 40.
189 papers.
Sort:
By area:
- 2022FlashAttention: Fast and Memory-Efficient Exact Attention with IO-Awarenesscited by 3,611
- 2022Simple Baselines for Image Restorationcited by 701
- 2023Gated Linear Attention Transformers with Hardware-Efficient Trainingcited by 490
- 2024Gated Delta Networks: Improving Mamba2 with Delta Rulecited by 483
- 2023Symbolic Discovery of Optimization Algorithmscited by 423
- 2024Parallelizing Linear Transformers with the Delta Rule over Sequence Lengthcited by 373
- 2025Gated Attention for Large Language Models: Non-linearity, Sparsity, and Attention-Sink-Freecited by 266
- 2022Long Range Language Modeling via Gated State Spacescited by 265
- 2024A Survey on Efficient Inference for Large Language Modelscited by 197
- 2024Actions Speak Louder than Words: Trillion-Parameter Sequential Transducers for Generative Recommendationscited by 184
- 2023Small-scale proxies for large-scale Transformer training instabilitiescited by 182
- 2022Mega: Moving Average Equipped Gated Attentioncited by 178
38 more
- 2023EfficientViT: Memory Efficient Vision Transformer with Cascaded Group Attentioncited by 172
- 2023RTMPose: Real-Time Multi-Person Pose Estimation based on MMPosecited by 170
- 2025Kimi Linear: An Expressive, Efficient Attention Architecturecited by 169
- 2022Decoupling Features in Hierarchical Propagation for Video Object Segmentationcited by 146
- 2024PlainMamba: Improving Non-Hierarchical Mamba in Visual Recognitioncited by 141
- 2024A Survey on Large Language Model Acceleration based on KV Cache Managementcited by 125
- 2024HGRN2: Gated Linear RNNs with State Expansioncited by 122
- 2025Test-Time Training Done Rightcited by 121
- 2022Block-Recurrent Transformerscited by 107
- 2024Demystify Mamba in Vision: A Linear Attention Perspectivecited by 101
- 2023Hierarchically Gated Recurrent Neural Network for Sequence Modelingcited by 100
- 2022The Devil in Linear Transformercited by 98
- 2023Modeling Dense Multimodal Interactions Between Biological Pathways and Histology for Survival Predictioncited by 92
- 2024The Hidden Attention of Mamba Modelscited by 88
- 2024In-Context Language Learning: Architectures and Algorithmscited by 86
- 2022Cramming: Training a Language Model on a Single GPU in One Daycited by 81
- 2023Diffusion Models Without Attentioncited by 75
- 2024Gated Slot Attention for Efficient Linear-Time Sequence Modelingcited by 73
- 2023Conditional Adapters: Parameter-efficient Transfer Learning with Fast Inferencecited by 69
- 2025DeltaProduct: Improving State-Tracking in Linear RNNs via Householder Productscited by 69
- 2024Mamba-360: Survey of State Space Models as Transformer Alternative for Long Sequence Modelling: Methods, Applications, and Challengescited by 63
- 2025Test-time regression: a unifying framework for designing sequence models with associative memorycited by 63
- 2024Model Compression and Efficient Inference for Large Language Models: A Surveycited by 62
- 2022MogaNet: Multi-order Gated Aggregation Networkcited by 59
- 2024Lightning Attention-2: A Free Lunch for Handling Unlimited Sequence Lengths in Large Language Modelscited by 57
- 2024Mechanistic Design and Scaling of Hybrid Architecturescited by 57
- 2025ATLAS: Learning to Optimally Memorize the Context at Test Timecited by 57
- 2022Pretraining Without Attentioncited by 51
- 2022Efficient Movie Scene Detection using State-Space Transformerscited by 51
- 2024Sparser is Faster and Less is More: Efficient Sparse Attention for Long-Range Transformerscited by 51
- 2024Megalodon: Efficient LLM Pretraining and Inference with Unlimited Context Lengthcited by 46
- 2025Rope to Nope and Back Again: A New Hybrid Attention Strategycited by 46
- 2023Toeplitz Neural Network for Sequence Modelingcited by 43
- 2023Replacing softmax with ReLU in Vision Transformerscited by 43
- 2024Diffusion-RWKV: Scaling RWKV-Like Architectures for Diffusion Modelscited by 41
- 2025Log-Linear Attentioncited by 39
- 2025Jet-Nemotron: Efficient Language Model with Post Neural Architecture Searchcited by 39
- 2022Efficient Long Sequence Modeling via State Space Augmented Transformercited by 37
50 most cited of 189 All 189 (JSON)
- 2026Attention Kernels for Learning Maps Between Heavy-Tailed Measures30 Sep 2026
- 2026Triadic Linear Attention: Three-Dimensional Recurrent States for Long-Context Sequence Modeling29 Sep 2026
- 2026SMat-Attention: Structured Long-Context Sequence Modeling28 Sep 2026
- 2026MS-GLA: Multi-Scale Gated Linear Attention for Addressing Representational Bottlenecks via Multi-Temporal Resolution28 Sep 2026
- 2026Abstention and Noise Filtering: Two Missing Primitives of Softmax Attention18 Sep 2026
- 2026RunningTensor: Generalizing Linear Attention to Higher-Order Recurrent States11 Sep 2026
- 2026Modern Transformers Are Implicit Hybrids: From Functional Differentiation to Principled Hybrid Architecture Design2 Sep 2026
- 2026A Top-Down Framework for Metric-Scale Athlete Localization from Single Broadcast Frames2 Sep 2026
- 2026TANGO: Treating Tokens as Operators22 Aug 2026
- 2026Rethinking Expressivity and Efficiency in Test-Time Training21 Aug 2026
- 2026Ask Self, Ask Others: Relation Is All You Need20 Aug 2026
- 2026Hand Visibility Detector: Per-Keypoint Visibility Estimation for Hands12 Aug 2026
38 more
- 2026Kernelized Linear Attention: Breaking the Capacity Wall with Symmetric Cones19 Jul 2026
- 2026Semidirect Fourier Delta Attention: Phase-Controlled Delta Memory with Constructive Chunk-WY Kernels12 Jun 2026
- 2026On Subquadratic Architectures: From Applications to Principles10 Jun 2026
- 2026When Good Enough Is Optimal: Multiplication-Only Matrix Inversion Approximation for Quantized Gated DeltaNet4 Jun 2026
- 2026Gated Graph Attention Networks with Learnable Temperature28 May 2026
- 2026Multi-Mixer Models: Flexible Sequence Modeling with Shared Representations27 May 2026
- 2026Gated DeltaNet-2: Decoupling Erase and Write in Linear Attention21 May 2026
- 2026Fast and Stable Triangular Inversion for Delta-Rule Linear Transformers20 May 2026
- 2026Multi-Headed Transformer Architectures as Time-dependent Wasserstein Gradient Flows15 May 2026
- 2026FG2-GDN: Enhancing Long-Context Gated Delta Networks with Doubly Fine-Grained Control21 Apr 2026
- 2026AdaCluster: Adaptive Query-Key Clustering for Sparse Attention in Video Generation20 Apr 2026
- 2026MambaSL: Exploring Single-Layer Mamba for Time Series Classification16 Apr 2026
- 2026On The Application of Linear Attention in Multimodal Transformers11 Apr 2026
- 2026Attention to Mamba: A Recipe for Cross-Architecture Distillation1 Apr 2026
- 2026Screening Is Enough1 Apr 2026
- 2026M2RNN: Non-Linear RNNs with Matrix-Valued States for Scalable Language Modeling15 Mar 2026
- 2026VGG-T3: Offline Feed-Forward 3D Reconstruction at Scale26 Feb 2026
- 2026RAT+: Train Dense, Infer Sparse -- Recurrence Augmented Attention for Dilated Inference20 Feb 2026
- 2026Learning State-Tracking from Code Using Linear RNNs16 Feb 2026
- 2026HyperMLP: An Integrated Perspective for Sequence Modeling13 Feb 2026
- 2026Free Energy Mixer6 Feb 2026
- 2026ZeroS: Zero-Sum Linear Attention for Efficient Transformers5 Feb 2026
- 2026On State Reduction in Linear Attention4 Feb 2026
- 2026Neural Attention Search Linear: Towards Adaptive Token-Level Hybrid Attention Models3 Feb 2026
- 2026Auto-Regressive Masked Diffusion Models23 Jan 2026
- 2026MHLA: Restoring Expressivity of Linear Attention via Token-Level Multi-Head12 Jan 2026
- 2026Gecko: An Efficient Neural Architecture Inherently Processing Sequences with Arbitrary Lengths10 Jan 2026
- 2026PackCache: A Training-Free Acceleration Method for Unified Autoregressive Video Generation via Compact KV-Cache7 Jan 2026
- 2025Trellis: Learning to Compress Key-Value Memory in Attention Models29 Dec 2025
- 2025MS-SSM: A Multi-Scale State Space Model for Efficient Sequence Modeling29 Dec 2025
- 2025KOSS: Kalman-Optimal Selective State Spaces for Long-Term Sequence Modeling18 Dec 2025
- 2025TNT: Improving Chunkwise Training for Test-Time Memorization10 Nov 2025
- 2025GroupKAN: Efficient Kolmogorov-Arnold Networks via Grouped Spline Modeling7 Nov 2025
- 2025FlashEVA: Accelerating LLM inference via Efficient Attention1 Nov 2025
- 2025Kimi Linear: An Expressive, Efficient Attention Architecture30 Oct 2025
- 2025Attentive Convolution: Unifying the Expressivity of Self-Attention with Convolutional Efficiency23 Oct 2025
- 2025Artificial Hippocampus Networks for Efficient Long-Context Modeling8 Oct 2025
- 2025Hybrid Architectures for Language Models: Systematic Analysis and Design Insights6 Oct 2025
50 newest of 189 All 189 (JSON)
- 2023Gated Linear Attention Transformers with Hardware-Efficient Trainingcited by 490
- 2024PlainMamba: Improving Non-Hierarchical Mamba in Visual Recognitioncited by 141
- 2024Demystify Mamba in Vision: A Linear Attention Perspectivecited by 101
- 2024The Hidden Attention of Mamba Modelscited by 88
- 2024Mamba-360: Survey of State Space Models as Transformer Alternative for Long Sequence Modelling: Methods, Applications, and Challengescited by 63
- 2024LKM-UNet: Large Kernel Vision Mamba UNet for Medical Image Segmentationcited by 34
- 2024Towards Universality: Studying Mechanistic Similarity Across Language Model Architecturescited by 31
- 2026Gated DeltaNet-2: Decoupling Erase and Write in Linear Attentioncited by 24
- 2025Hybrid Architectures for Language Models: Systematic Analysis and Design Insightscited by 13
- 2024SliceMamba with Neural Architecture Search for Medical Image Segmentationcited by 11
- 2025PRE-Mamba: A 4D State Space Model for Ultra-High-Frequent Event Camera Derainingcited by 5
- 2025VSRM: A Robust Mamba-Based Framework for Video Super-Resolutioncited by 5
12 most cited of 16 All 16 (JSON)
- 2026Gated DeltaNet-2: Decoupling Erase and Write in Linear Attention21 May 2026
- 2026MambaSL: Exploring Single-Layer Mamba for Time Series Classification16 Apr 2026
- 2026Attention to Mamba: A Recipe for Cross-Architecture Distillation1 Apr 2026
- 2025KOSS: Kalman-Optimal Selective State Spaces for Long-Term Sequence Modeling18 Dec 2025
- 2025Hybrid Architectures for Language Models: Systematic Analysis and Design Insights6 Oct 2025
- 2025VSRM: A Robust Mamba-Based Framework for Video Super-Resolution28 Jun 2025
- 2025PRE-Mamba: A 4D State Space Model for Ultra-High-Frequent Event Camera Deraining8 May 2025
- 2025On the Expressivity of Selective State-Space Layers: A Multivariate Polynomial Approach4 Feb 2025
- 2024Towards Universality: Studying Mechanistic Similarity Across Language Model Architectures9 Oct 2024
- 2024SliceMamba with Neural Architecture Search for Medical Image Segmentation11 Jul 2024
- 2024Demystify Mamba in Vision: A Linear Attention Perspective26 May 2024
- 2024Mamba-360: Survey of State Space Models as Transformer Alternative for Long Sequence Modelling: Methods, Applications, and Challenges24 Apr 2024
12 newest of 16 All 16 (JSON)
- 2025Test-Time Training Done Rightcited by 121
- 2023Diffusion Models Without Attentioncited by 75
- 2024Diffusion-RWKV: Scaling RWKV-Like Architectures for Diffusion Modelscited by 41
- 2024CLEAR: Conv-Like Linearization Revs Pre-Trained Diffusion Transformers Upcited by 26
- 2026AdaCluster: Adaptive Query-Key Clustering for Sparse Attention in Video Generationcited by 5
- 2023Diffusing Gaussian Mixtures for Generating Categorical Datacited by 4
- 2026Auto-Regressive Masked Diffusion Modelscited by 4
- 2026AdaCluster: Adaptive Query-Key Clustering for Sparse Attention in Video Generation20 Apr 2026
- 2026Auto-Regressive Masked Diffusion Models23 Jan 2026
- 2025Test-Time Training Done Right29 May 2025
- 2024CLEAR: Conv-Like Linearization Revs Pre-Trained Diffusion Transformers Up20 Dec 2024
- 2024Diffusion-RWKV: Scaling RWKV-Like Architectures for Diffusion Models6 Apr 2024
- 2023Diffusion Models Without Attention30 Nov 2023
- 2023Diffusing Gaussian Mixtures for Generating Categorical Data8 Mar 2023
- 2024A Survey on Efficient Inference for Large Language Modelscited by 197
- 2024SpeechPrune: Context-aware Token Pruning for Speech Information Retrievalcited by 20
- 2024When Linear Attention Meets Autoregressive Decoding: Towards More Effective and Efficient Linearized Large Language Modelscited by 9
- 2026PackCache: A Training-Free Acceleration Method for Unified Autoregressive Video Generation via Compact KV-Cachecited by 9
- 2026RAT+: Train Dense, Infer Sparse -- Recurrence Augmented Attention for Dilated Inferencecited by 2
- 2026Semidirect Fourier Delta Attention: Phase-Controlled Delta Memory with Constructive Chunk-WY Kernelscited by 0
- 2026Semidirect Fourier Delta Attention: Phase-Controlled Delta Memory with Constructive Chunk-WY Kernels12 Jun 2026
- 2026RAT+: Train Dense, Infer Sparse -- Recurrence Augmented Attention for Dilated Inference20 Feb 2026
- 2026PackCache: A Training-Free Acceleration Method for Unified Autoregressive Video Generation via Compact KV-Cache7 Jan 2026
- 2024SpeechPrune: Context-aware Token Pruning for Speech Information Retrieval16 Dec 2024
- 2024When Linear Attention Meets Autoregressive Decoding: Towards More Effective and Efficient Linearized Large Language Models11 Jun 2024
- 2024A Survey on Efficient Inference for Large Language Models22 Apr 2024
- 2022FlashAttention: Fast and Memory-Efficient Exact Attention with IO-Awarenesscited by 3,611
- 2022Simple Baselines for Image Restorationcited by 701
- 2024Gated Delta Networks: Improving Mamba2 with Delta Rulecited by 483
- 2023Symbolic Discovery of Optimization Algorithmscited by 423
- 2024Parallelizing Linear Transformers with the Delta Rule over Sequence Lengthcited by 373
- 2022Long Range Language Modeling via Gated State Spacescited by 265
- 2024Actions Speak Louder than Words: Trillion-Parameter Sequential Transducers for Generative Recommendationscited by 184
- 2023Small-scale proxies for large-scale Transformer training instabilitiescited by 182
- 2022Mega: Moving Average Equipped Gated Attentioncited by 178
- 2023EfficientViT: Memory Efficient Vision Transformer with Cascaded Group Attentioncited by 172
- 2023RTMPose: Real-Time Multi-Person Pose Estimation based on MMPosecited by 170
- 2025Kimi Linear: An Expressive, Efficient Attention Architecturecited by 169
12 most cited of 154 All 154 (JSON)
- 2026Attention Kernels for Learning Maps Between Heavy-Tailed Measures30 Sep 2026
- 2026Triadic Linear Attention: Three-Dimensional Recurrent States for Long-Context Sequence Modeling29 Sep 2026
- 2026SMat-Attention: Structured Long-Context Sequence Modeling28 Sep 2026
- 2026MS-GLA: Multi-Scale Gated Linear Attention for Addressing Representational Bottlenecks via Multi-Temporal Resolution28 Sep 2026
- 2026Abstention and Noise Filtering: Two Missing Primitives of Softmax Attention18 Sep 2026
- 2026RunningTensor: Generalizing Linear Attention to Higher-Order Recurrent States11 Sep 2026
- 2026Modern Transformers Are Implicit Hybrids: From Functional Differentiation to Principled Hybrid Architecture Design2 Sep 2026
- 2026A Top-Down Framework for Metric-Scale Athlete Localization from Single Broadcast Frames2 Sep 2026
- 2026TANGO: Treating Tokens as Operators22 Aug 2026
- 2026Ask Self, Ask Others: Relation Is All You Need20 Aug 2026
- 2026Hand Visibility Detector: Per-Keypoint Visibility Estimation for Hands12 Aug 2026
- 2026Kernelized Linear Attention: Breaking the Capacity Wall with Symmetric Cones19 Jul 2026
12 newest of 154 All 154 (JSON)