Sparse is Enough in Scaling Transformers
Not on the map. Why?
Abstract (arXiv)
Large Transformer models yield impressive results on many tasks, but are expensive to train, or even fine-tune, and so slow at decoding that their use and study becomes out of reach. We address this problem by leveraging sparsity. We study sparse variants for all layers in the Transformer and propose Scaling Transformers, a family of next generation Transformer models that use sparse layers to scale efficiently and perform unbatched decoding much faster than the standard Transformer as we scale up the model size. Surprisingly, the sparse layers are enough to obtain the same perplexity as the standard Transformer with the same number of parameters. We also integrate with prior sparsity approaches to attention and enable fast inference on long sequences even with limited memory. This results in performance competitive to the state-of-the-art on long text summarization.
Key facts. Cited by 86 papers in the map (data through 14 Oct 2026). Builds on 40 of its 47 references that are in the map. First posted on arXiv 24 Nov 2021.
Builds on
40 of 47 references.
References · 40
- 2017Attention Is All You Needcited by 60,361
- 2018BERT: Pre-training of Deep Bidirectional Transformers for Language Understandingcited by 40,399
- 2020Language Models are Few-Shot Learnerscited by 32,419
- 2019Exploring the Limits of Transfer Learning with a Unified Text-to-Text Transformercited by 14,599
- 2020Scaling Laws for Neural Language Modelscited by 6,586
- 2019DistilBERT, a distilled version of BERT: smaller, faster, cheaper and lightercited by 4,307
- 2017Outrageously Large Neural Networks: The Sparsely-Gated Mixture-of-Experts Layercited by 3,509
- 2021Switch Transformers: Scaling to Trillion Parameter Models with Simple and Efficient Sparsitycited by 3,073
- 2016Categorical Reparameterization with Gumbel-Softmaxcited by 2,739
- 2020mT5: A massively multilingual pre-trained text-to-text transformercited by 1,818
- 2020Big Bird: Transformers for Longer Sequencescited by 1,684
- 2020Reformer: The Efficient Transformercited by 1,668
28 more
- 2020GShard: Scaling Giant Models with Conditional Computation and Automatic Shardingcited by 1,631
- 2020Rethinking Attention with Performerscited by 1,601
- 2019Generating Long Sequences with Sparse Transformerscited by 1,546
- 2019Energy and Policy Considerations for Deep Learning in NLPcited by 1,254
- 2019PEGASUS: Pre-training with Extracted Gap-sentences for Abstractive Summarizationcited by 1,102
- 2020Efficient Transformers: A Surveycited by 790
- 2019Enhancing the Locality and Breaking the Memory Bottleneck of Transformer on Time Series Forecastingcited by 607
- 2021Carbon Emissions and Large Neural Network Trainingcited by 567
- 2020MobileBERT: a Compact Task-Agnostic BERT for Resource-Limited Devicescited by 482
- 2018A Discourse-Aware Attention Model for Abstractive Summarization of Long Documentscited by 478
- 2019The State of Sparsity in Deep Neural Networkscited by 476
- 2020Efficient Content-Based Sparse Attention with Routing Transformerscited by 471
- 2019Q-BERT: Hessian Based Ultra Low Precision Quantization of BERTcited by 359
- 2020Sparse Sinkhorn Attentioncited by 251
- 2021Learning N:M Fine-grained Structured Sparse Neural Networks From Scratchcited by 179
- 2018Mesh-TensorFlow: Deep Learning for Supercomputerscited by 169
- 2019Adaptive Attention Span in Transformerscited by 149
- 2020HAT: Hardware-Aware Transformers for Efficient Natural Language Processingcited by 136
- 2015Neural GPUs Learn Algorithmscited by 121
- 2021Do Transformer Modifications Transfer Across Implementations and Applications?cited by 87
- 2020Compression of Deep Learning Models for Text: A Surveycited by 64
- 2020SqueezeBERT: What can computer vision teach NLP about efficient neural networks?cited by 50
- 2020Efficient Transformer-based Large Scale Language Representations using Hardware-friendly Block Structured Pruningcited by 39
- 2019Fully Quantized Transformer for Machine Translationcited by 36
- 2020FastFormers: Highly Efficient Transformer Models for Natural Language Understandingcited by 28
- 2018Discrete Autoencoders for Sequence Modelscited by 24
- 2020Successfully Applying the Stabilized Lottery Ticket Hypothesis to the Transformer Architecturecited by 23
- 2019Flexible Multi-task Networks by Learning Parameter Allocationcited by 7
Built on by
Citing papers per year: 2021: 1; 2022: 18; 2023: 29; 2024: 24; 2025: 14.
86 papers.
Sort:
By area:
- 2022Fast Inference from Transformers via Speculative Decodingcited by 1,653
- 2024RULER: What's the Real Context Size of Your Long-Context Language Models?cited by 1,346
- 2022Efficiently Scaling Transformer Inferencecited by 539
- 2024OLMoE: Open Mixture-of-Experts Language Modelscited by 301
- 2022The case for 4-bit precision: k-bit Inference Scaling Lawscited by 251
- 2022EfficientFormer: Vision Transformers at MobileNet Speedcited by 241
- 2021Adaptive Token Sampling For Efficient Vision Transformerscited by 183
- 2022A Review of Sparse Expert Models in Deep Learningcited by 148
- 2022The Lazy Neuron Phenomenon: On Emergence of Activation Sparsity in Transformerscited by 122
- 2022M3ViT: Mixture-of-Experts Vision Transformer for Efficient Multi-task Learning with Model-Accelerator Co-designcited by 98
- 2023ReLU Strikes Back: Exploiting Activation Sparsity in Large Language Modelscited by 94
- 2023Learning to Search Feasible and Infeasible Regions of Routing Problems with Flexible Neural k-Optcited by 81
38 more
- 2024All-in-one simulation-based inferencecited by 74
- 2023Advancing Transformer Architecture in Long-Context Large Language Models: A Comprehensive Surveycited by 72
- 2022CATs++: Boosting Cost Aggregation with Convolutions and Transformerscited by 53
- 2024Revisiting Knowledge Distillation for Autoregressive Language Modelscited by 42
- 2024CATS: Contextually-Aware Thresholding for Sparsity in Large Language Modelscited by 41
- 2023Dynamic Sparse Training with Structured Sparsitycited by 38
- 2023Towards MoE Deployment: Mitigating Inefficiencies in Mixture-of-Expert (MoE) Inferencecited by 37
- 2023Brainformers: Trading Simplicity for Efficiencycited by 31
- 2024Multi-Candidate Speculative Decodingcited by 31
- 2023SPDF: Sparse Pre-training and Dense Fine-tuning for Large Language Modelscited by 28
- 2024XMoE: Sparse Models with Fine-grained and Adaptive Expert Selectioncited by 28
- 2023On Efficient Training of Large-Scale Deep Learning Models: A Literature Reviewcited by 26
- 2024Selective Attention Improves Transformercited by 26
- 2024Configurable Foundation Models: Building LLMs from a Modular Perspectivecited by 22
- 2025Orion-MSP: Multi-Scale Sparse Attention for Tabular In-Context Learningcited by 19
- 2022Modular and On-demand Bias Mitigation with Attribute-Removal Subnetworkscited by 18
- 2023Beyond Classical Attention: Quantum Attention for Scalable Computationcited by 18
- 2024Progressive Gradient Flow for Robust N:M Sparsity Training in Transformerscited by 18
- 2024Attention layers provably solve single-location regressioncited by 17
- 2023Exploiting Activation Sparsity with Dense to Dynamic-k Mixture-of-Experts Conversioncited by 16
- 2023MetaBEV: Solving Sensor Failures for BEV Detection and Map Segmentationcited by 14
- 2023Eventful Transformers: Leveraging Temporal Redundancy in Vision Transformerscited by 13
- 2022DCT-Former: Efficient Self-Attention with Discrete Cosine Transformcited by 12
- 2022Sparse Mixers: Combining MoE and Mixing to build a more efficient BERTcited by 12
- 2024CFSP: An Efficient Structured Pruning Framework for LLMs with Coarse-to-Fine Activation Informationcited by 11
- 2024Multi-Draft Speculative Sampling: Canonical Decomposition and Theoretical Limitscited by 11
- 2024Conditional computation in neural networks: principles and research trendscited by 10
- 2022Minimum Variance Unbiased N:M Sparsity for the Neural Gradientscited by 9
- 2023Towards A Unified View of Sparse Feed-Forward Network in Pretraining Large Language Modelcited by 9
- 2025FastGRPO: Accelerating Policy Optimization via Concurrency-aware Speculative Decoding and Online Draft Learningcited by 9
- 2022Efficient Language Modeling with Sparse all-MLPcited by 8
- 2023Sparse Binary Transformers for Multivariate Time Series Modelingcited by 8
- 2024Revisiting SMoE Language Models by Evaluating Inefficiencies with Task Specific Expert Pruningcited by 8
- 2023Adaptivity and Modularity for Efficient Generalization Over Task Complexitycited by 7
- 2023MMGPL: Multimodal Medical Data Analysis with Graph Prompt Learningcited by 7
- 2024HiRE: High Recall Approximate Top-k Estimation for Efficient LLM Inferencecited by 7
- 2022A Theoretical View on Sparsely Activated Networkscited by 6
- 2022Pretraining the Vision Transformer using self-supervised methods for vision based Deep Reinforcement Learningcited by 5
50 most cited of 86 All 86 (JSON)
- 2025Memory as Resonance: A Biomimetic Architecture for Infinite Context Memory on Ergodic Phonetic Manifolds23 Dec 2025
- 2025Unlocking the Address Book: Dissecting the Sparse Semantic Structure of LLM Key-Value Caches via Sparse Autoencoders11 Dec 2025
- 2025Orion-MSP: Multi-Scale Sparse Attention for Tabular In-Context Learning4 Nov 2025
- 2025Contextual Attention Modulation: Towards Efficient Multi-Task Adaptation in Large Language Models20 Oct 2025
- 2025FastGRPO: Accelerating Policy Optimization via Concurrency-aware Speculative Decoding and Online Draft Learning26 Sep 2025
- 2025Dynamic Adaptive Shared Experts with Grouped Multi-Head Attention Mixture of Experts5 Sep 2025
- 2025Spatial-Temporal-Spectral Mamba with Sparse Deformable Token Sequence for Enhanced MODIS Time Series Classification29 Jul 2025
- 2025GNN-CNN: An Efficient Hybrid Model of Convolutional and Graph Neural Networks for Text Representation10 Jul 2025
- 2025TopK Language Models26 Jun 2025
- 2025Spark Transformer: Reactivating Sparsity in FFN and Attention7 Jun 2025
- 2025ZenFlow: Enabling Stall-Free Offloading for LLM Training18 May 2025
- 2025M-DocSum: Do LVLMs Genuinely Comprehend Interleaved Image-Text in Document Summarization?27 Mar 2025
38 more
- 2025UbiMoE: A Ubiquitous Mixture-of-Experts Vision Transformer Accelerator With Hybrid Computation Pattern on FPGA8 Feb 2025
- 2025CURing Large Models: Compression via CUR Decomposition8 Jan 2025
- 2024KEDformer:Knowledge Extraction Seasonal Trend Decomposition for Long-term Sequence Prediction6 Dec 2024
- 2024Multi-Draft Speculative Sampling: Canonical Decomposition and Theoretical Limits23 Oct 2024
- 2024Selective Attention Improves Transformer3 Oct 2024
- 2024Attention layers provably solve single-location regression2 Oct 2024
- 2024CFSP: An Efficient Structured Pruning Framework for LLMs with Coarse-to-Fine Activation Information20 Sep 2024
- 2024Configurable Foundation Models: Building LLMs from a Modular Perspective4 Sep 2024
- 2024OLMoE: Open Mixture-of-Experts Language Models3 Sep 2024
- 2024Revisiting SMoE Language Models by Evaluating Inefficiencies with Task Specific Expert Pruning2 Sep 2024
- 2024Uncovering Layer-Dependent Activation Sparsity Patterns in ReLU Transformers10 Jul 2024
- 2024Decoding at the Speed of Thought: Harnessing Parallel Decoding of Lexical Units for LLMs24 May 2024
- 2024A Comprehensive Survey of Accelerated Generation Techniques in Large Language Models15 May 2024
- 2024DynaMo: Accelerating Language Model Inference with Dynamic Multi-Token Sampling1 May 2024
- 2024All-in-one simulation-based inference15 Apr 2024
- 2024CATS: Contextually-Aware Thresholding for Sparsity in Large Language Models12 Apr 2024
- 2024RULER: What's the Real Context Size of Your Long-Context Language Models?9 Apr 2024
- 2024Conditional computation in neural networks: principles and research trends12 Mar 2024
- 2024Segmentation Guided Sparse Transformer for Under-Display Camera Image Restoration9 Mar 2024
- 2024MediSwift: Efficient Sparse Pre-trained Biomedical Language Models1 Mar 2024
- 2024XMoE: Sparse Models with Fine-grained and Adaptive Expert Selection27 Feb 2024
- 2024Masked Matrix Multiplication for Emergent Sparsity21 Feb 2024
- 2024Revisiting Knowledge Distillation for Autoregressive Language Models19 Feb 2024
- 2024HiRE: High Recall Approximate Top-k Estimation for Efficient LLM Inference14 Feb 2024
- 2024Progressive Gradient Flow for Robust N:M Sparsity Training in Transformers7 Feb 2024
- 2024Multi-Candidate Speculative Decoding12 Jan 2024
- 2023Heterogeneous Encoders Scaling In The Transformer For Neural Machine Translation26 Dec 2023
- 2023MMGPL: Multimodal Medical Data Analysis with Graph Prompt Learning22 Dec 2023
- 2023Advancing Transformer Architecture in Long-Context Large Language Models: A Comprehensive Survey21 Nov 2023
- 2023Learning to Search Feasible and Infeasible Regions of Routing Problems with Flexible Neural k-Opt27 Oct 2023
- 2023Mixture of Tokens: Continuous MoE through Cross-Example Aggregation24 Oct 2023
- 2023Manifold-Preserving Transformers are Effective for Short-Long Range Encoding22 Oct 2023
- 2023G-SPEED: General SParse Efficient Editing MoDel16 Oct 2023
- 2023Adaptivity and Modularity for Efficient Generalization Over Task Complexity13 Oct 2023
- 2023ReLU Strikes Back: Exploiting Activation Sparsity in Large Language Models6 Oct 2023
- 2023Exploiting Activation Sparsity with Dense to Dynamic-k Mixture-of-Experts Conversion6 Oct 2023
- 2023Memory-efficient particle filter recurrent neural network for object localization2 Oct 2023
- 2023Eventful Transformers: Leveraging Temporal Redundancy in Vision Transformers25 Aug 2023
50 newest of 86 All 86 (JSON)
- 2024OLMoE: Open Mixture-of-Experts Language Modelscited by 301
- 2022A Review of Sparse Expert Models in Deep Learningcited by 148
- 2022M3ViT: Mixture-of-Experts Vision Transformer for Efficient Multi-task Learning with Model-Accelerator Co-designcited by 98
- 2023Towards MoE Deployment: Mitigating Inefficiencies in Mixture-of-Expert (MoE) Inferencecited by 37
- 2024XMoE: Sparse Models with Fine-grained and Adaptive Expert Selectioncited by 28
- 2023Exploiting Activation Sparsity with Dense to Dynamic-k Mixture-of-Experts Conversioncited by 16
- 2022Sparse Mixers: Combining MoE and Mixing to build a more efficient BERTcited by 12
- 2024Conditional computation in neural networks: principles and research trendscited by 10
- 2023Towards A Unified View of Sparse Feed-Forward Network in Pretraining Large Language Modelcited by 9
- 2022Efficient Language Modeling with Sparse all-MLPcited by 8
- 2024Revisiting SMoE Language Models by Evaluating Inefficiencies with Task Specific Expert Pruningcited by 8
- 2022A Theoretical View on Sparsely Activated Networkscited by 6
12 most cited of 15 All 15 (JSON)
- 2025Dynamic Adaptive Shared Experts with Grouped Multi-Head Attention Mixture of Experts5 Sep 2025
- 2025UbiMoE: A Ubiquitous Mixture-of-Experts Vision Transformer Accelerator With Hybrid Computation Pattern on FPGA8 Feb 2025
- 2024OLMoE: Open Mixture-of-Experts Language Models3 Sep 2024
- 2024Revisiting SMoE Language Models by Evaluating Inefficiencies with Task Specific Expert Pruning2 Sep 2024
- 2024Conditional computation in neural networks: principles and research trends12 Mar 2024
- 2024XMoE: Sparse Models with Fine-grained and Adaptive Expert Selection27 Feb 2024
- 2023Mixture of Tokens: Continuous MoE through Cross-Example Aggregation24 Oct 2023
- 2023Exploiting Activation Sparsity with Dense to Dynamic-k Mixture-of-Experts Conversion6 Oct 2023
- 2023Towards A Unified View of Sparse Feed-Forward Network in Pretraining Large Language Model23 May 2023
- 2023Towards MoE Deployment: Mitigating Inefficiencies in Mixture-of-Expert (MoE) Inference10 Mar 2023
- 2022M3ViT: Mixture-of-Experts Vision Transformer for Efficient Multi-task Learning with Model-Accelerator Co-design26 Oct 2022
- 2022A Review of Sparse Expert Models in Deep Learning4 Sep 2022
12 newest of 15 All 15 (JSON)
- 2022Fast Inference from Transformers via Speculative Decodingcited by 1,653
- 2024Multi-Candidate Speculative Decodingcited by 31
- 2024Multi-Draft Speculative Sampling: Canonical Decomposition and Theoretical Limitscited by 11
- 2024HiRE: High Recall Approximate Top-k Estimation for Efficient LLM Inferencecited by 7
- 2024A Comprehensive Survey of Accelerated Generation Techniques in Large Language Modelscited by 5
- 2025Unlocking the Address Book: Dissecting the Sparse Semantic Structure of LLM Key-Value Caches via Sparse Autoencoderscited by 1
- 2025Unlocking the Address Book: Dissecting the Sparse Semantic Structure of LLM Key-Value Caches via Sparse Autoencoders11 Dec 2025
- 2024Multi-Draft Speculative Sampling: Canonical Decomposition and Theoretical Limits23 Oct 2024
- 2024A Comprehensive Survey of Accelerated Generation Techniques in Large Language Models15 May 2024
- 2024HiRE: High Recall Approximate Top-k Estimation for Efficient LLM Inference14 Feb 2024
- 2024Multi-Candidate Speculative Decoding12 Jan 2024
- 2022Fast Inference from Transformers via Speculative Decoding30 Nov 2022
- 2025TopK Language Modelscited by 0
- 2025TopK Language Models26 Jun 2025
- 2024RULER: What's the Real Context Size of Your Long-Context Language Models?cited by 1,346
- 2022Efficiently Scaling Transformer Inferencecited by 539
- 2022The case for 4-bit precision: k-bit Inference Scaling Lawscited by 251
- 2022EfficientFormer: Vision Transformers at MobileNet Speedcited by 241
- 2021Adaptive Token Sampling For Efficient Vision Transformerscited by 183
- 2022The Lazy Neuron Phenomenon: On Emergence of Activation Sparsity in Transformerscited by 122
- 2023ReLU Strikes Back: Exploiting Activation Sparsity in Large Language Modelscited by 94
- 2023Learning to Search Feasible and Infeasible Regions of Routing Problems with Flexible Neural k-Optcited by 81
- 2024All-in-one simulation-based inferencecited by 74
- 2023Advancing Transformer Architecture in Long-Context Large Language Models: A Comprehensive Surveycited by 72
- 2022CATs++: Boosting Cost Aggregation with Convolutions and Transformerscited by 53
- 2024CATS: Contextually-Aware Thresholding for Sparsity in Large Language Modelscited by 41
12 most cited of 59 All 59 (JSON)
- 2025Memory as Resonance: A Biomimetic Architecture for Infinite Context Memory on Ergodic Phonetic Manifolds23 Dec 2025
- 2025Orion-MSP: Multi-Scale Sparse Attention for Tabular In-Context Learning4 Nov 2025
- 2025Contextual Attention Modulation: Towards Efficient Multi-Task Adaptation in Large Language Models20 Oct 2025
- 2025GNN-CNN: An Efficient Hybrid Model of Convolutional and Graph Neural Networks for Text Representation10 Jul 2025
- 2025Spark Transformer: Reactivating Sparsity in FFN and Attention7 Jun 2025
- 2025ZenFlow: Enabling Stall-Free Offloading for LLM Training18 May 2025
- 2025CURing Large Models: Compression via CUR Decomposition8 Jan 2025
- 2024KEDformer:Knowledge Extraction Seasonal Trend Decomposition for Long-term Sequence Prediction6 Dec 2024
- 2024Selective Attention Improves Transformer3 Oct 2024
- 2024Attention layers provably solve single-location regression2 Oct 2024
- 2024CFSP: An Efficient Structured Pruning Framework for LLMs with Coarse-to-Fine Activation Information20 Sep 2024
- 2024Configurable Foundation Models: Building LLMs from a Modular Perspective4 Sep 2024
12 newest of 59 All 59 (JSON)