Nystr"omformer: A Nystr"om-Based Algorithm for Approximating Self-Attention
Not on the map. Why?
Abstract (arXiv)
Transformers have emerged as a powerful tool for a broad range of natural language processing tasks. A key component that drives the impressive performance of Transformers is the self-attention mechanism that encodes the influence or dependence of other tokens on each specific token. While beneficial, the quadratic complexity of self-attention on the input sequence length has limited its application to longer sequences -- a topic being actively studied in the community. To address this limitation, we propose Nystr"omformer -- a model that exhibits favorable scalability as a function of sequence length. Our idea is based on adapting the Nystr"om method to approximate standard self-attention with O(n) complexity. The scalability of Nystr"omformer enables application to longer sequences with thousands of tokens. We perform evaluations on multiple downstream tasks on the GLUE benchmark and IMDB reviews with standard sequence length, and find that our Nystr"omformer performs comparably, or in a few cases, even slightly better, than standard self-attention. On longer sequence tasks in the Long Range Arena (LRA) benchmark, Nystr"omformer performs favorably relative to other efficient self-attention methods. Our code is available at https://github.com/mlpen/Nystromformer.
Key facts. Cited by 468 papers in the map (data through 7 Oct 2026). Builds on 34 of its 65 references that are in the map. First posted on arXiv 7 Feb 2021.
Builds on
34 of 65 references.
References · 34
- 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
- 2019RoBERTa: A Robustly Optimized BERT Pretraining Approachcited by 13,266
- 2019BART: Denoising Sequence-to-Sequence Pre-training for Natural Language Generation, Translation, and Comprehensioncited by 5,874
- 2018GLUE: A Multi-Task Benchmark and Analysis Platform for Natural Language Understandingcited by 4,854
- 2019DistilBERT, a distilled version of BERT: smaller, faster, cheaper and lightercited by 4,243
- 2016SQuAD: 100,000+ Questions for Machine Comprehension of Textcited by 4,115
- 2020Longformer: The Long-Document Transformercited by 3,362
- 2019ALBERT: A Lite BERT for Self-supervised Learning of Language Representationscited by 2,772
- 2017A Broad-Coverage Challenge Corpus for Sentence Understanding through Inferencecited by 2,665
- 2018Deep contextualized word representationscited by 2,285
22 more
- 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
- 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
- 2015Aligning Books and Movies: Towards Story-like Visual Explanations by Watching Movies and Reading Bookscited by 1,041
- 2016Training Deep Nets with Sublinear Memory Costcited by 825
- 2019Are Sixteen Heads Really Better than One?cited by 803
- 2020Long Range Arena: A Benchmark for Efficient Transformerscited by 696
- 2018Efficient Attention: Attention with Linear Complexitiescited by 366
- 2019Q8BERT: Quantized 8Bit BERTcited by 296
- 2018ListOps: A Diagnostic Dataset for Latent Tree Learningcited by 128
- 2020Fast Transformers with Clustered Attentioncited by 123
- 2019Large Memory Layers with Product Keyscited by 120
- 2018Learning long-range spatial dependencies with horizontal gated-recurrent unitscited by 98
- 2018Baseline Needs More Love: On Simple Word-Embedding-Based Models and Associated Pooling Mechanismscited by 53
- 2017Adaptive Sampled Softmax with Kernel Based Samplingcited by 38
- 2019Sampled Softmax with Random Fourier Featurescited by 33
- 2013Improving CUR Matrix Decomposition and the Nystr"om Approximation via Adaptive Samplingcited by 21
- 2019Nystr"om landmark sampling and regularized Christoffel functionscited by 8
Built on by
Citing papers per year: 2021: 48; 2022: 79; 2023: 87; 2024: 97; 2025: 80; 2026: 77.
468 papers.
Sort:
By area:
- 2022FlashAttention: Fast and Memory-Efficient Exact Attention with IO-Awarenesscited by 3,611
- 2022FEDformer: Frequency Enhanced Decomposed Transformer for Long-term Series Forecastingcited by 1,200
- 2021Perceiver: General Perception with Iterative Attentioncited by 1,089
- 2022Simplified State Space Layers for Sequence Modelingcited by 776
- 2021Perceiver IO: A General Architecture for Structured Inputs & Outputscited by 616
- 2021TransMIL: Transformer based Correlated Multiple Instance Learning for Whole Slide Image Classificationcited by 549
- 2024FlashAttention-3: Fast and Accurate Attention with Asynchrony and Low-precisioncited by 507
- 2022XMem: Long-Term Video Object Segmentation with an Atkinson-Shiffrin Memory Modelcited by 436
- 2021A Survey of Transformerscited by 399
- 2021XCiT: Cross-Covariance Image Transformerscited by 335
- 2021Choose a Transformer: Fourier or Galerkincited by 333
- 2022ActionFormer: Localizing Moments of Actions with Transformerscited by 308
38 more
- 2025YOLOv12: Attention-Centric Real-Time Object Detectorscited by 300
- 2024Transolver: A Fast Transformer Solver for PDEs on General Geometriescited by 299
- 2022Transformers in Medical Imaging: A Surveycited by 245
- 2021Adaptive Fourier Neural Operators: Efficient Token Mixers for Transformerscited by 237
- 2021Keeping Your Eye on the Ball: Trajectory Attention in Video Transformerscited by 224
- 2024MobileLLM: Optimizing Sub-billion Parameter Language Models for On-Device Use Casescited by 216
- 2022cosFormer: Rethinking Softmax in Attentioncited by 212
- 2022Pure Transformers are Powerful Graph Learnerscited by 188
- 2023FLatten Transformer: Vision Transformer using Focused Linear Attentioncited by 170
- 2021TransGAN: Two Pure Transformers Can Make One Strong GAN, and That Can Scale Upcited by 162
- 2024Eagle and Finch: RWKV with Matrix-Valued States and Dynamic Recurrencecited by 155
- 2022FiLM: Frequency improved Legendre Memory Model for Long-term Time Series Forecastingcited by 147
- 2022On The Computational Complexity of Self-Attentioncited by 137
- 2022Liquid Structural State-Space Modelscited by 134
- 2021SOFT: Softmax-free Transformer with Linear Complexitycited by 128
- 2021Scatterbrain: Unifying Sparse and Low-rank Attention Approximationcited by 128
- 2023Scalable Transformer for PDE Surrogate Modelingcited by 123
- 2024The Hedgehog & the Porcupine: Expressive Linear Attentions with Softmax Mimicrycited by 119
- 2022EdgeViTs: Competing Light-weight CNNs on Mobile Devices with Vision Transformerscited by 116
- 2022Cost Aggregation with 4D Convolutional Swin Transformer for Few-Shot Segmentationcited by 114
- 2023Retrieval meets Long Context Large Language Modelscited by 114
- 2023SoundStorm: Efficient Parallel Audio Generationcited by 111
- 2021LambdaNetworks: Modeling Long-Range Interactions Without Attentioncited by 107
- 2021Long-Short Transformer: Efficient Transformers for Language and Visioncited by 102
- 2024Demystify Mamba in Vision: A Linear Attention Perspectivecited by 101
- 2023Modeling Dense Multimodal Interactions Between Biological Pathways and Histology for Survival Predictioncited by 92
- 2023Agent Attention: On the Integration of Softmax and Linear Attentioncited by 90
- 2022Learning Operators with Coupled Attentioncited by 88
- 2022A graph-transformer for whole slide image classificationcited by 84
- 2022KERPLE: Kernelized Relative Positional Embedding for Length Extrapolationcited by 84
- 2021U-Net Transformer: Self and Cross Attention for Medical Image Segmentationcited by 81
- 2021Pixelated Butterfly: Simple and Efficient Sparse training for Neural Network Modelscited by 75
- 2021A Practical Survey on Faster and Lighter Transformerscited by 73
- 2023Cross-Modal Translation and Alignment for Survival Analysiscited by 70
- 2022Flowformer: Linearizing Transformers with Conservation Flowscited by 69
- 2024Feature Re-Embedding: Towards Foundation Model-Level Performance in Computational Pathologycited by 66
- 2021Closed-form Continuous-time Neural Modelscited by 65
- 2024Mamba-360: Survey of State Space Models as Transformer Alternative for Long Sequence Modelling: Methods, Applications, and Challengescited by 63
50 most cited of 468 All 468 (JSON)
- 2026Efficiently Approximating Attention Is Hard29 Sep 2026
- 2026Quasi Linear Kernel Attention with Infinite Capacity28 Sep 2026
- 2026DimPO: Dimensionality Reduction for Attention using Preference Optimization26 Sep 2026
- 2026Where Compute Matters: Heterogeneous Attention for Efficient Video Diffusion25 Sep 2026
- 2026ARM: Attention with Routed-Memory for Learnable Sparse Control21 Sep 2026
- 2026Prescriptive SVD-Inspired Attention via Spectral Energy Retention21 Sep 2026
- 2026CAR-MIL: Counterfactual Attention Regularization for Multiple Instance Learning8 Sep 2026
- 2026Nystr"om Attention Matches Full Attention for Cross-Sectional Stock Prediction8 Sep 2026
- 2026Conditioned Initialization for Attention7 Sep 2026
- 2026Safin-1: Safety from Within through Memory-Native State Evolution31 Aug 2026
- 2026Efficient and High-Quality Depth Estimation via Pixel-Space Diffusion with Linear Attention31 Aug 2026
- 2026Intrinsic Interaction Geometry Controls the Low-Rank Complexity of Softmax Attention28 Aug 2026
38 more
- 2026ClusterAttention: A training-free speedup of bidirectional attention27 Aug 2026
- 2026Prefix Sliding for efficient test-time scaling26 Aug 2026
- 2026Beyond Sparse Weights: When Is Attention Compressible?21 Aug 2026
- 2026Partition the Support, Reconstruct the Residual: Training-Free Sparse Attention for Video Generation and World Models19 Aug 2026
- 2026CG-GLORE: A Conjugate Gradient-Based Global-Local Regularization Network for Sparse-View CT Reconstruction15 Aug 2026
- 2026MARCH: Scaling Recurrent Memory with Content-Routed State Anchors12 Aug 2026
- 2026FLARE++: Low-rank attention with dynamic attention routing12 Aug 2026
- 2026MixFormer: Linear Transformer with Mixture of Memory Experts10 Aug 2026
- 2026Gated Spatial Redundancy Projection for Pathology Transformer Attentions8 Aug 2026
- 2026HSMLA: Hierarchical Softmax Multi-scale Linear Attention for Efficient Vision Transformers7 Aug 2026
- 2026bioMoR: Biology-Guided Mixture-of-Recursions for Effective Genomic Learning7 Aug 2026
- 2026Placement Is Free, Composition Is Not: The Latin Square as a Provably-Balanced Construction for Heterogeneous Sequence-Mixer Stacks29 Jul 2026
- 2026A Coulomb Particle Model for Learning Kernel Attention in Transformers26 Jul 2026
- 2026ELSAA: Efficient Low-Rank and Sparse Attention Approximation for Training Transformers22 Jul 2026
- 2026Norm or Direction? Decoding Vision Mambas for High-Resolution Vision21 Jul 2026
- 2026EMAGN: Efficient Multi-Attention Graph Network via Learned Clustering for Scalable Traffic Forecasting14 Jul 2026
- 2026MobileWan: Closing the Quality Gap for Mobile Video Diffusion7 Jul 2026
- 2026ELiTeFormer: An Efficient Transformer for FPGAs4 Jul 2026
- 2026BrainWorld: A Structural-Prior-Conditioned Generative Model for Whole-Brain 4D fMRI Dynamics16 Jun 2026
- 2026Spectral-LSH: Sub-Quadratic Prompt Compression via Krylov-Projected Locality-Sensitive Hashing12 Jun 2026
- 2026The Word and the Way: Strategies for Domain-Specific BERT Pre-Training in German Medical NLP2 Jun 2026
- 2026PortBERT: Navigating the Depths of Portuguese Language Models1 Jun 2026
- 2026Functional Attention: From Pairwise Affinities to Functional Correspondences29 May 2026
- 2026Trading Complexity for Expressivity Through Structured Generalized Linear Token Mixing29 May 2026
- 2026Multi-Mixer Models: Flexible Sequence Modeling with Shared Representations27 May 2026
- 2026Transformer-Enhanced Reinforcement Learning: Fundamentals and Applications in Communication Networks26 May 2026
- 2026DunbaaBERT: From Sacrifice to Semantics26 May 2026
- 2026Approaching I/O-optimality for Approximate Attention22 May 2026
- 2026Accelerating Vision Foundation Models with Drop-in Depthwise Convolution21 May 2026
- 2026GCE-MIL: Faithful and Recoverable Evidence for Multiple Instance Learning in Whole-Slide Imaging17 May 2026
- 2026Spatial Blindness in Whole-Slide Multiple Instance Learning17 May 2026
- 2026Olivia: Harmonizing Time Series Foundation Models with Power Spectral Density17 May 2026
- 2026Representative Attention For Vision Transformers14 May 2026
- 2026ASAP: Amortized Doubly-Stochastic Attention via Sliced Dual Projection13 May 2026
- 2026Do Vision Transformers Need All-to-All Attention? Global Communication Through Elastic Learned Cores12 May 2026
- 2026Beyond Position Bias: Shifting Context Compression from Position-Driven to Semantic-Driven10 May 2026
- 2026UMEDA: Unified Multi-modal Efficient Data Fusion for Privacy-Preserving Graph Federated Learning via Spectral-Gated Attention and Diffusion-Based Operator Alignment8 May 2026
- 2026Linear-Time Global Visual Modeling without Explicit Attention3 May 2026
50 newest of 468 All 468 (JSON)
- 2022Simplified State Space Layers for Sequence Modelingcited by 776
- 2024Demystify Mamba in Vision: A Linear Attention Perspectivecited by 101
- 2024Mamba-360: Survey of State Space Models as Transformer Alternative for Long Sequence Modelling: Methods, Applications, and Challengescited by 63
- 20242DMamba: Efficient State Space Model for Image Representation with Applications on Giga-Pixel Whole Slide Image Classificationcited by 35
- 2024EfficientViM: Efficient Vision Mamba with Hidden State Mixer based State Space Dualitycited by 17
- 2024MamMIL: Multiple Instance Learning for Whole Slide Images with State Space Modelscited by 12
- 2024A Mamba Foundation Model for Time Series Forecastingcited by 9
- 2026Attention to Mamba: A Recipe for Cross-Architecture Distillationcited by 3
- 2025On Structured State-Space Dualitycited by 1
- 2024Efficient State Space Model via Fast Tensor Convolution and Block Diagonalizationcited by 0
- 2026Norm or Direction? Decoding Vision Mambas for High-Resolution Visioncited by 0
- 2026Norm or Direction? Decoding Vision Mambas for High-Resolution Vision21 Jul 2026
- 2026Attention to Mamba: A Recipe for Cross-Architecture Distillation1 Apr 2026
- 2025On Structured State-Space Duality6 Oct 2025
- 20242DMamba: Efficient State Space Model for Image Representation with Applications on Giga-Pixel Whole Slide Image Classification1 Dec 2024
- 2024EfficientViM: Efficient Vision Mamba with Hidden State Mixer based State Space Duality22 Nov 2024
- 2024A Mamba Foundation Model for Time Series Forecasting5 Nov 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
- 2024MamMIL: Multiple Instance Learning for Whole Slide Images with State Space Models8 Mar 2024
- 2024Efficient State Space Model via Fast Tensor Convolution and Block Diagonalization23 Feb 2024
- 2022Simplified State Space Layers for Sequence Modeling9 Aug 2022
- 2024LoCoCo: Dropping In Convolutions for Long Context Compressioncited by 18
- 2024NoMAD-Attention: Efficient LLM Inference on CPUs Through Multiply-add-free Attentioncited by 12
- 2024When Linear Attention Meets Autoregressive Decoding: Towards More Effective and Efficient Linearized Large Language Modelscited by 9
- 2025KVLinC : KV Cache Quantization with Hadamard Rotation and Linear Correctioncited by 9
- 2026CARE: Covariance-Aware and Rank-Enhanced Decomposition for Enabling Multi-Head Latent Attentioncited by 5
- 2026Beyond Position Bias: Shifting Context Compression from Position-Driven to Semantic-Drivencited by 3
- 2026Representative Attention For Vision Transformerscited by 0
- 2026Spectral-LSH: Sub-Quadratic Prompt Compression via Krylov-Projected Locality-Sensitive Hashingcited by 0
- 2026Beyond Sparse Weights: When Is Attention Compressible?cited by 0
- 2026Beyond Sparse Weights: When Is Attention Compressible?21 Aug 2026
- 2026Spectral-LSH: Sub-Quadratic Prompt Compression via Krylov-Projected Locality-Sensitive Hashing12 Jun 2026
- 2026Representative Attention For Vision Transformers14 May 2026
- 2026Beyond Position Bias: Shifting Context Compression from Position-Driven to Semantic-Driven10 May 2026
- 2026CARE: Covariance-Aware and Rank-Enhanced Decomposition for Enabling Multi-Head Latent Attention18 Mar 2026
- 2025KVLinC : KV Cache Quantization with Hadamard Rotation and Linear Correction6 Oct 2025
- 2024When Linear Attention Meets Autoregressive Decoding: Towards More Effective and Efficient Linearized Large Language Models11 Jun 2024
- 2024LoCoCo: Dropping In Convolutions for Long Context Compression8 Jun 2024
- 2024NoMAD-Attention: Efficient LLM Inference on CPUs Through Multiply-add-free Attention2 Mar 2024
- 2024CLEAR: Conv-Like Linearization Revs Pre-Trained Diffusion Transformers Upcited by 26
- 2024Efficient Diffusion Transformer with Step-wise Dynamic Attention Mediatorscited by 24
- 2024CaLMFlow: Volterra Flow Matching using Causal Language Modelscited by 3
- 2026VMonarch: Efficient Video Diffusion Transformers with Structured Attentioncited by 3
- 2026MobileWan: Closing the Quality Gap for Mobile Video Diffusioncited by 3
- 2024Efficient generative adversarial networks using linear additive-attention Transformerscited by 2
- 2026FreqFormer: Hierarchical Frequency-Domain Attention with Adaptive Spectral Routing for Long-Sequence Video Diffusion Transformerscited by 0
- 2026Where Compute Matters: Heterogeneous Attention for Efficient Video Diffusioncited by 0
- 2026Where Compute Matters: Heterogeneous Attention for Efficient Video Diffusion25 Sep 2026
- 2026MobileWan: Closing the Quality Gap for Mobile Video Diffusion7 Jul 2026
- 2026FreqFormer: Hierarchical Frequency-Domain Attention with Adaptive Spectral Routing for Long-Sequence Video Diffusion Transformers14 Apr 2026
- 2026VMonarch: Efficient Video Diffusion Transformers with Structured Attention29 Jan 2026
- 2024CLEAR: Conv-Like Linearization Revs Pre-Trained Diffusion Transformers Up20 Dec 2024
- 2024CaLMFlow: Volterra Flow Matching using Causal Language Models3 Oct 2024
- 2024Efficient Diffusion Transformer with Step-wise Dynamic Attention Mediators11 Aug 2024
- 2024Efficient generative adversarial networks using linear additive-attention Transformers17 Jan 2024
- 2022cosFormer: Rethinking Softmax in Attentioncited by 212
- 2024Reducing Reasoning Costs: The Path of Optimization for Chain of Thought via Sparse Attention Mechanismcited by 4
- 2023Do large language models and humans have similar behaviors in causal inference with script knowledge?cited by 3
- 2022Dense but Efficient VideoQA for Intricate Compositional Reasoningcited by 2
- 2026Prefix Sliding for efficient test-time scalingcited by 2
- 2026Prefix Sliding for efficient test-time scaling26 Aug 2026
- 2024Reducing Reasoning Costs: The Path of Optimization for Chain of Thought via Sparse Attention Mechanism14 Nov 2024
- 2023Do large language models and humans have similar behaviors in causal inference with script knowledge?13 Nov 2023
- 2022Dense but Efficient VideoQA for Intricate Compositional Reasoning19 Oct 2022
- 2022cosFormer: Rethinking Softmax in Attention17 Feb 2022
- 2022FlashAttention: Fast and Memory-Efficient Exact Attention with IO-Awarenesscited by 3,611
- 2022FEDformer: Frequency Enhanced Decomposed Transformer for Long-term Series Forecastingcited by 1,200
- 2021Perceiver: General Perception with Iterative Attentioncited by 1,089
- 2021Perceiver IO: A General Architecture for Structured Inputs & Outputscited by 616
- 2021TransMIL: Transformer based Correlated Multiple Instance Learning for Whole Slide Image Classificationcited by 549
- 2024FlashAttention-3: Fast and Accurate Attention with Asynchrony and Low-precisioncited by 507
- 2022XMem: Long-Term Video Object Segmentation with an Atkinson-Shiffrin Memory Modelcited by 436
- 2021A Survey of Transformerscited by 399
- 2021XCiT: Cross-Covariance Image Transformerscited by 335
- 2021Choose a Transformer: Fourier or Galerkincited by 333
- 2022ActionFormer: Localizing Moments of Actions with Transformerscited by 308
- 2025YOLOv12: Attention-Centric Real-Time Object Detectorscited by 300
12 most cited of 429 All 429 (JSON)
- 2026Efficiently Approximating Attention Is Hard29 Sep 2026
- 2026Quasi Linear Kernel Attention with Infinite Capacity28 Sep 2026
- 2026ARM: Attention with Routed-Memory for Learnable Sparse Control21 Sep 2026
- 2026Prescriptive SVD-Inspired Attention via Spectral Energy Retention21 Sep 2026
- 2026CAR-MIL: Counterfactual Attention Regularization for Multiple Instance Learning8 Sep 2026
- 2026Nystr"om Attention Matches Full Attention for Cross-Sectional Stock Prediction8 Sep 2026
- 2026Conditioned Initialization for Attention7 Sep 2026
- 2026Safin-1: Safety from Within through Memory-Native State Evolution31 Aug 2026
- 2026Efficient and High-Quality Depth Estimation via Pixel-Space Diffusion with Linear Attention31 Aug 2026
- 2026Intrinsic Interaction Geometry Controls the Low-Rank Complexity of Softmax Attention28 Aug 2026
- 2026ClusterAttention: A training-free speedup of bidirectional attention27 Aug 2026
- 2026CG-GLORE: A Conjugate Gradient-Based Global-Local Regularization Network for Sparse-View CT Reconstruction15 Aug 2026
12 newest of 429 All 429 (JSON)