Gated Subspace Inference for Transformer Acceleration
Not on the map. Why?
Abstract (arXiv)
A method is presented for accelerating inference in transformer language models by exploiting the low effective rank of the token activation manifold at each layer. The method decomposes each activation vector into a subspace component and a residual, computes the linear-layer output on the subspace component via a cached low-rank weight image at reduced memory bandwidth, and applies a per-token gate that determines whether the residual correction is computed or skipped. The gate ensures that the output distribution is preserved to within a controllable tolerance. Validation on three model families (GPT-2 124M, GPT-J 6B, OPT 6.7B) on AMD MI300X demonstrates effective speedups of 3.0x to 10.5x on linear-layer weight reads with perplexity ratios below 1.00 and top-1 token agreement above 98%. The method requires no retraining, no architectural modification, and no approximation of the attention mechanism. At the operating point (k = 256, ε = 0.05) on GPT-J 14 6B, the accelerated model produces character-for-character identical output to the baseline.
Key facts. No citing papers in the map yet (data through 14 Oct 2026). Builds on 22 of its 34 references that are in the map. First posted on arXiv 4 May 2026.
Builds on
22 of 34 references.
References · 22
- 2022FlashAttention: Fast and Memory-Efficient Exact Attention with IO-Awarenesscited by 3,673
- 2022OPT: Open Pre-trained Transformer Language Modelscited by 3,512
- 2017Outrageously Large Neural Networks: The Sparsely-Gated Mixture-of-Experts Layercited by 3,509
- 2020Linformer: Self-Attention with Linear Complexitycited by 1,474
- 2022LLM.int8(): 8-bit Matrix Multiplication for Transformers at Scalecited by 906
- 2020Intrinsic Dimensionality Explains the Effectiveness of Language Model Fine-Tuningcited by 671
- 2022Efficiently Scaling Transformer Inferencecited by 539
- 2016Adaptive Computation Time for Recurrent Neural Networkscited by 506
- 2022Confident Adaptive Language Modelingcited by 310
- 2019Intrinsic dimension of data representations in deep neural networkscited by 282
- 2024Mixture-of-Depths: Dynamically allocating compute in transformer-based language modelscited by 258
- 2023Deja Vu: Contextual Sparsity for Efficient LLMs at Inference Timecited by 252
10 more
- 2024SVD-LLM: Truncation-aware Singular Value Decomposition for Large Language Model Compressioncited by 243
- 2023PowerInfer: Fast Large Language Model Serving with a Consumer-grade GPUcited by 217
- 2023ASVD: Activation-aware Singular Value Decomposition for Compressing Large Language Modelscited by 169
- 2024LLM Inference Unveiled: Survey and Roofline Model Insightscited by 152
- 2023The geometry of hidden representations of large transformer modelscited by 112
- 2024Training-Free Activation Sparsity in Large Language Modelscited by 55
- 2024CATS: Contextually-Aware Thresholding for Sparsity in Large Language Modelscited by 41
- 2025FLAT-LLM: Fine-grained Low-rank Activation Space Transformation for Large Language Model Compressioncited by 5
- 2026RooflineBench: A Benchmarking Framework for On-Device LLMs via Roofline Analysiscited by 5
- 2025Adaptive Rank Allocation: Speeding Up Modern Transformers with RaNA Adapterscited by 3
Built on by
Not cited yet.