Linear Complexity Randomized Self-attention Mechanism
Not on the map. Why?
Abstract (arXiv)
Recently, random feature attentions (RFAs) are proposed to approximate the softmax attention in linear time and space complexity by linearizing the exponential kernel. In this paper, we first propose a novel perspective to understand the bias in such approximation by recasting RFAs as self-normalized importance samplers. This perspective further sheds light on an unbiased estimator for the whole softmax attention, called randomized attention (RA). RA constructs positive random features via query-specific distributions and enjoys greatly improved approximation fidelity, albeit exhibiting quadratic complexity. By combining the expressiveness in RA and the efficiency in RFA, we develop a novel linear complexity self-attention mechanism called linear randomized attention (LARA). Extensive experiments across various domains demonstrate that RA and LARA significantly improve the performance of RFAs by a substantial margin.
Key facts. Cited by 31 papers in the map (data through 7 Oct 2026). Builds on 68 of its 100 references that are in the map. First posted on arXiv 10 Apr 2022.
Builds on
68 of 100 references.
References · 68
- 2017Attention Is All You Needcited by 59,932
- 2018BERT: Pre-training of Deep Bidirectional Transformers for Language Understandingcited by 40,204
- 2014Adam: A Method for Stochastic Optimizationcited by 36,960
- 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
- 2021Swin Transformer: Hierarchical Vision Transformer using Shifted Windowscited by 12,082
- 2020End-to-End Object Detection with Transformerscited by 7,384
- 2017mixup: Beyond Empirical Risk Minimizationcited by 5,155
- 2016SGDR: Stochastic Gradient Descent with Warm Restartscited by 4,993
- 2020Training data-efficient image transformers & distillation through attentioncited by 4,560
- 2020Longformer: The Long-Document Transformercited by 3,362
- 2015Neural Machine Translation of Rare Words with Subword Unitscited by 2,898
38 more
- 2019CutMix: Regularization Strategy to Train Strong Classifiers with Localizable Featurescited by 2,875
- 2019RandAugment: Practical automated data augmentation with a reduced search spacecited by 2,261
- 2020Transformers are RNNs: Fast Autoregressive Transformers with Linear Attentioncited by 1,979
- 2021Pyramid Vision Transformer: A Versatile Backbone for Dense Prediction without Convolutionscited by 1,960
- 2017The Kinetics Human Action Video Datasetcited by 1,894
- 2020Big Bird: Transformers for Longer Sequencescited by 1,673
- 2020Reformer: The Efficient Transformercited by 1,658
- 2020Rethinking Attention with Performerscited by 1,584
- 2021Decision Transformer: Reinforcement Learning via Sequence Modelingcited by 1,528
- 2019Generating Long Sequences with Sparse Transformerscited by 1,523
- 2020Linformer: Self-Attention with Linear Complexitycited by 1,462
- 2017Random Erasing Data Augmentationcited by 1,235
- 2017The "something something" video database for learning and evaluating visual common sensecited by 1,092
- 2021Perceiver: General Perception with Iterative Attentioncited by 1,089
- 2016Deep Networks with Stochastic Depthcited by 973
- 2020Designing Network Design Spacescited by 954
- 2021CvT: Introducing Convolutions to Vision Transformerscited by 946
- 2020Efficient Transformers: A Surveycited by 784
- 2018Image Transformercited by 770
- 2021PVT v2: Improved Baselines with Pyramid Vision Transformercited by 749
- 2020Long Range Arena: A Benchmark for Efficient Transformerscited by 696
- 2021Perceiver IO: A General Architecture for Structured Inputs & Outputscited by 616
- 2018Universal Transformerscited by 613
- 2021Twins: Revisiting the Design of Spatial Attention in Vision Transformerscited by 565
- 2021Linear Transformers Are Secretly Fast Weight Programmerscited by 471
- 2020Efficient Content-Based Sparse Attention with Routing Transformerscited by 469
- 2021Nystr"omformer: A Nystr"om-Based Algorithm for Approximating Self-Attentioncited by 468
- 2021A Survey of Transformerscited by 399
- 2021Early Convolutions Help Transformers See Bettercited by 395
- 2021Random Feature Attentioncited by 350
- 2021FNet: Mixing Tokens with Fourier Transformscited by 343
- 2021XCiT: Cross-Covariance Image Transformerscited by 335
- 2018Generating Wikipedia by Summarizing Long Sequencescited by 333
- 2020Sparse Sinkhorn Attentioncited by 250
- 2021Keeping Your Eye on the Ball: Trajectory Attention in Video Transformerscited by 224
- 2021Multi-Scale Vision Longformer: A New Vision Transformer for High-Resolution Image Encodingcited by 207
- 2018Scaling Neural Machine Translationcited by 204
- 2020Synthesizer: Rethinking Self-Attention in Transformer Modelscited by 204
50 most cited of 68 All 68 (JSON)
Built on by
Citing papers per year: 2022: 5; 2023: 10; 2024: 11; 2025: 1; 2026: 4.
31 papers.
Sort:
By area:
- 2023Adapting Language Models to Compress Contextscited by 350
- 2023In-context Autoencoder for Context Compression in a Large Language Modelcited by 190
- 2022The Devil in Linear Transformercited by 98
- 2024Lightning Attention-2: A Free Lunch for Handling Unlimited Sequence Lengths in Large Language Modelscited by 57
- 2023In-Context Learning with Many Demonstration Examplescited by 39
- 2023TransNormerLLM: A Faster and Better Large Language Model with Improved TransNormercited by 34
- 2024Various Lengths, Constant Speed: Efficient Language Modeling with Lightning Attentioncited by 31
- 2024Training Tensor Attention Efficiently: From Cubic to Almost Linear Timecited by 29
- 2023Faster Causal Attention Over Large Sequences Through Sparse Flash Attentioncited by 24
- 2023Efficient Attention via Control Variatescited by 23
- 2023Why Softmax Attention Outperforms Linear Attentioncited by 15
- 2024UncertaintyRAG: Span-Level Uncertainty Enhanced Long-Context Modeling for Retrieval-Augmented Generationcited by 14
19 more
- 2022CAB: Comprehensive Attention Benchmarking on Long Sequence Modelingcited by 10
- 2024Linear Attention Sequence Parallelismcited by 8
- 2024IceFormer: Accelerated Inference with Long-Sequence Transformers on CPUscited by 8
- 2024SelfCP: Compressing Over-Limit Prompt via the Frozen Large Language Model Itselfcited by 8
- 2022Linear Video Transformer with Feature Fixationcited by 6
- 2024Squid: Long Context as a New Modality for Energy-Efficient On-Device Language Modelscited by 6
- 2024Theoretical Understanding of In-Context Learning in Shallow Transformers with Unstructured Datacited by 5
- 2022Attention and Self-Attention in Random Forestscited by 4
- 2023Linear Attention via Orthogonal Memorycited by 4
- 2022DBA: Efficient Transformer with Dynamic Bilinear Low-Rank Attentioncited by 3
- 2023FLuRKA: Fast and accurate unified Low-Rank & Kernel Attentioncited by 3
- 2024FactorLLM: Factorizing Knowledge via Mixture of Experts for Large Language Modelscited by 2
- 2024Interactive Multi-Head Self-Attention with Linear Complexitycited by 1
- 2025FlashEVA: Accelerating LLM inference via Efficient Attentioncited by 1
- 2026Analytical Correction for Subsampling Bias in Drifting Modelscited by 1
- 2026Scratchpad Patching: Decoupling Compute from Patch Size in Byte-Level Language Modelscited by 1
- 2023Attentive Multi-Layer Perceptron for Non-autoregressive Generationcited by 0
- 2026Finite-Time Analysis of Gradient Descent for Shallow Transformerscited by 0
- 2026Temporal Preservation over Processing: Diagnosing and Designing Spatiotemporal Single-Stage Video Detectorscited by 0
- 2026Temporal Preservation over Processing: Diagnosing and Designing Spatiotemporal Single-Stage Video Detectors30 Jun 2026
- 2026Scratchpad Patching: Decoupling Compute from Patch Size in Byte-Level Language Models10 May 2026
- 2026Analytical Correction for Subsampling Bias in Drifting Models29 Apr 2026
- 2026Finite-Time Analysis of Gradient Descent for Shallow Transformers23 Jan 2026
- 2025FlashEVA: Accelerating LLM inference via Efficient Attention1 Nov 2025
- 2024UncertaintyRAG: Span-Level Uncertainty Enhanced Long-Context Modeling for Retrieval-Augmented Generation3 Oct 2024
- 2024Squid: Long Context as a New Modality for Energy-Efficient On-Device Language Models28 Aug 2024
- 2024FactorLLM: Factorizing Knowledge via Mixture of Experts for Large Language Models15 Aug 2024
- 2024Various Lengths, Constant Speed: Efficient Language Modeling with Lightning Attention27 May 2024
- 2024SelfCP: Compressing Over-Limit Prompt via the Frozen Large Language Model Itself27 May 2024
- 2024Training Tensor Attention Efficiently: From Cubic to Almost Linear Time26 May 2024
- 2024IceFormer: Accelerated Inference with Long-Sequence Transformers on CPUs5 May 2024
19 more
- 2024Linear Attention Sequence Parallelism3 Apr 2024
- 2024Interactive Multi-Head Self-Attention with Linear Complexity27 Feb 2024
- 2024Theoretical Understanding of In-Context Learning in Shallow Transformers with Unstructured Data1 Feb 2024
- 2024Lightning Attention-2: A Free Lunch for Handling Unlimited Sequence Lengths in Large Language Models9 Jan 2024
- 2023Linear Attention via Orthogonal Memory18 Dec 2023
- 2023Why Softmax Attention Outperforms Linear Attention18 Oct 2023
- 2023Attentive Multi-Layer Perceptron for Non-autoregressive Generation14 Oct 2023
- 2023TransNormerLLM: A Faster and Better Large Language Model with Improved TransNormer27 Jul 2023
- 2023In-context Autoencoder for Context Compression in a Large Language Model13 Jul 2023
- 2023FLuRKA: Fast and accurate unified Low-Rank & Kernel Attention27 Jun 2023
- 2023Faster Causal Attention Over Large Sequences Through Sparse Flash Attention1 Jun 2023
- 2023Adapting Language Models to Compress Contexts24 May 2023
- 2023In-Context Learning with Many Demonstration Examples9 Feb 2023
- 2023Efficient Attention via Control Variates9 Feb 2023
- 2022DBA: Efficient Transformer with Dynamic Bilinear Low-Rank Attention24 Nov 2022
- 2022The Devil in Linear Transformer19 Oct 2022
- 2022Linear Video Transformer with Feature Fixation15 Oct 2022
- 2022CAB: Comprehensive Attention Benchmarking on Long Sequence Modeling14 Oct 2022
- 2022Attention and Self-Attention in Random Forests9 Jul 2022
- 2023Adapting Language Models to Compress Contextscited by 350
- 2022The Devil in Linear Transformercited by 98
- 2024Lightning Attention-2: A Free Lunch for Handling Unlimited Sequence Lengths in Large Language Modelscited by 57
- 2023In-Context Learning with Many Demonstration Examplescited by 39
- 2023TransNormerLLM: A Faster and Better Large Language Model with Improved TransNormercited by 34
- 2024Various Lengths, Constant Speed: Efficient Language Modeling with Lightning Attentioncited by 31
- 2024Training Tensor Attention Efficiently: From Cubic to Almost Linear Timecited by 29
- 2023Faster Causal Attention Over Large Sequences Through Sparse Flash Attentioncited by 24
- 2023Efficient Attention via Control Variatescited by 23
- 2023Why Softmax Attention Outperforms Linear Attentioncited by 15
- 2022CAB: Comprehensive Attention Benchmarking on Long Sequence Modelingcited by 10
- 2024Linear Attention Sequence Parallelismcited by 8
12 most cited of 27 All 27 (JSON)
- 2026Temporal Preservation over Processing: Diagnosing and Designing Spatiotemporal Single-Stage Video Detectors30 Jun 2026
- 2026Analytical Correction for Subsampling Bias in Drifting Models29 Apr 2026
- 2026Finite-Time Analysis of Gradient Descent for Shallow Transformers23 Jan 2026
- 2025FlashEVA: Accelerating LLM inference via Efficient Attention1 Nov 2025
- 2024Squid: Long Context as a New Modality for Energy-Efficient On-Device Language Models28 Aug 2024
- 2024Various Lengths, Constant Speed: Efficient Language Modeling with Lightning Attention27 May 2024
- 2024SelfCP: Compressing Over-Limit Prompt via the Frozen Large Language Model Itself27 May 2024
- 2024Training Tensor Attention Efficiently: From Cubic to Almost Linear Time26 May 2024
- 2024IceFormer: Accelerated Inference with Long-Sequence Transformers on CPUs5 May 2024
- 2024Linear Attention Sequence Parallelism3 Apr 2024
- 2024Interactive Multi-Head Self-Attention with Linear Complexity27 Feb 2024
- 2024Theoretical Understanding of In-Context Learning in Shallow Transformers with Unstructured Data1 Feb 2024
12 newest of 27 All 27 (JSON)