Sharing Attention Weights for Fast Transformer
Not on the map. Why?
Abstract (arXiv)
Recently, the Transformer machine translation system has shown strong results by stacking attention layers on both the source and target-language sides. But the inference of this model is slow due to the heavy use of dot-product attention in auto-regressive decoding. In this paper we speed up Transformer via a fast and lightweight attention model. More specifically, we share attention weights in adjacent layers and enable the efficient re-use of hidden states in a vertical manner. Moreover, the sharing policy can be jointly learned with the MT model. We test our approach on ten WMT and NIST OpenMT tasks. Experimental results show that it yields an average of 1.3X speed-up (with almost no decrease in BLEU) on top of a state-of-the-art implementation that has already adopted a cache for fast inference. Also, our approach obtains a 1.8X speed-up when it works with the textscAan model. This is even 16 times faster than the baseline with no use of the attention cache.
Key facts. Cited by 34 papers in the map (data through 7 Oct 2026). First posted on arXiv 26 Jun 2019.
Builds on
No reference list yet.
Built on by
Citing papers per year: 2021: 5; 2022: 3; 2023: 8; 2024: 8; 2025: 9; 2026: 1.
34 papers.
Sort:
By area:
- 2023Towards Efficient Generative Large Language Model Serving: A Survey from Algorithms to Systemscited by 101
- 2021Lessons on Parameter Sharing across Layers in Transformerscited by 85
- 2023Transformers in Speech Processing: A Surveycited by 38
- 2024KVSharer: Efficient Inference via Layer-Wise Dissimilar KV Cache Sharingcited by 38
- 2024Basis Sharing: Cross-Layer Parameter Sharing for Large Language Model Compressioncited by 34
- 2025Tensor Product Attention Is All You Needcited by 27
- 2023Skip-Attention: Improving Vision Transformers by Paying Less Attentioncited by 23
- 2021LazyFormer: Self Attention with Lazy Updatecited by 16
- 2024Compressing KV Cache for Long-Context LLM Inference with Inter-Layer Attention Similaritycited by 11
- 2021An Efficient Transformer Decoder with Compressed Sub-layerscited by 10
- 2021OH-Former: Omni-Relational High-Order Transformer for Person Re-Identificationcited by 10
- 2024Dynamic Layer Tying for Parameter-Efficient Transformerscited by 10
22 more
- 2025Kascade: A Practical Sparse Attention Method for Long-Context LLM Inferencecited by 9
- 2022ACORT: A Compact Object Relation Transformer for Parameter Efficient Image Captioningcited by 6
- 2023Recycle-and-Distill: Universal Compression Strategy for Transformer-based Speech SSL Models with Attention Map Reusing and Masking Distillationcited by 6
- 2024Increasing Model Capacity for Free: A Simple Strategy for Parameter Efficient Fine-tuningcited by 6
- 2025Layer-wise dynamic rank for compressing large language modelscited by 6
- 2024Cross-layer Attention Sharing for Pre-trained Large Language Modelscited by 5
- 2024Head-wise Shareable Attention for Large Language Modelscited by 4
- 2021Bag of Tricks for Optimizing Transformer Efficiencycited by 3
- 2022Hybrid-Regressive Neural Machine Translationcited by 3
- 2023Exploring Attention Map Reuse for Efficient Transformer Neural Networkscited by 3
- 2023RecycleGPT: An Autoregressive Language Model with Recyclable Modulecited by 3
- 2024PSformer: Parameter-efficient Transformer with Segment Attention for Time Series Forecastingcited by 2
- 2025Deconstructing Attention: Investigating Design Principles for Effective Language Modelingcited by 2
- 2025AttnCache: Accelerating Self-Attention Inference for LLM Prefill via Attention Cachecited by 2
- 2022PGX: A Multi-level GNN Explanation Framework Based on Separate Knowledge Distillation Processescited by 1
- 2023Only 5% Attention Is All You Need: Efficient Long-range Document-level Neural Machine Translationcited by 1
- 2025MASA: Rethinking the Representational Bottleneck in LoRA with Multi-A Shared Adaptationcited by 1
- 2023Revisiting Implicit Models: Sparsity Trade-offs Capability in Weight-tied Model for Vision Taskscited by 0
- 2025Round Attention: A Novel Round-Level Attention Mechanism to Accelerate LLM Inferencecited by 0
- 2025ACE: A Cardinality Estimator for Set-Valued Queriescited by 0
- 2025It Takes a Good Model to Train a Good Model: Generalized Gaussian Priors for Optimized LLMscited by 0
- 2026Attention Routing Stabilizes Early: Working-Set Inference for Recurrent Language Modelscited by 0
- 2026Attention Routing Stabilizes Early: Working-Set Inference for Recurrent Language Models23 Sep 2026
- 2025Kascade: A Practical Sparse Attention Method for Long-Context LLM Inference18 Dec 2025
- 2025AttnCache: Accelerating Self-Attention Inference for LLM Prefill via Attention Cache29 Oct 2025
- 2025Deconstructing Attention: Investigating Design Principles for Effective Language Modeling13 Oct 2025
- 2025MASA: Rethinking the Representational Bottleneck in LoRA with Multi-A Shared Adaptation7 Oct 2025
- 2025Layer-wise dynamic rank for compressing large language models30 Sep 2025
- 2025It Takes a Good Model to Train a Good Model: Generalized Gaussian Priors for Optimized LLMs31 May 2025
- 2025ACE: A Cardinality Estimator for Set-Valued Queries19 Mar 2025
- 2025Round Attention: A Novel Round-Level Attention Mechanism to Accelerate LLM Inference21 Feb 2025
- 2025Tensor Product Attention Is All You Need11 Jan 2025
- 2024Compressing KV Cache for Long-Context LLM Inference with Inter-Layer Attention Similarity3 Dec 2024
- 2024PSformer: Parameter-efficient Transformer with Segment Attention for Time Series Forecasting3 Nov 2024
22 more
- 2024KVSharer: Efficient Inference via Layer-Wise Dissimilar KV Cache Sharing24 Oct 2024
- 2024Basis Sharing: Cross-Layer Parameter Sharing for Large Language Model Compression2 Oct 2024
- 2024Cross-layer Attention Sharing for Pre-trained Large Language Models4 Aug 2024
- 2024Increasing Model Capacity for Free: A Simple Strategy for Parameter Efficient Fine-tuning1 Jul 2024
- 2024Head-wise Shareable Attention for Large Language Models19 Feb 2024
- 2024Dynamic Layer Tying for Parameter-Efficient Transformers23 Jan 2024
- 2023Towards Efficient Generative Large Language Model Serving: A Survey from Algorithms to Systems23 Dec 2023
- 2023Only 5% Attention Is All You Need: Efficient Long-range Document-level Neural Machine Translation25 Sep 2023
- 2023RecycleGPT: An Autoregressive Language Model with Recyclable Module7 Aug 2023
- 2023Revisiting Implicit Models: Sparsity Trade-offs Capability in Weight-tied Model for Vision Tasks16 Jul 2023
- 2023Recycle-and-Distill: Universal Compression Strategy for Transformer-based Speech SSL Models with Attention Map Reusing and Masking Distillation19 May 2023
- 2023Transformers in Speech Processing: A Survey21 Mar 2023
- 2023Exploring Attention Map Reuse for Efficient Transformer Neural Networks29 Jan 2023
- 2023Skip-Attention: Improving Vision Transformers by Paying Less Attention5 Jan 2023
- 2022Hybrid-Regressive Neural Machine Translation19 Oct 2022
- 2022PGX: A Multi-level GNN Explanation Framework Based on Separate Knowledge Distillation Processes5 Aug 2022
- 2022ACORT: A Compact Object Relation Transformer for Parameter Efficient Image Captioning11 Feb 2022
- 2021OH-Former: Omni-Relational High-Order Transformer for Person Re-Identification23 Sep 2021
- 2021Bag of Tricks for Optimizing Transformer Efficiency9 Sep 2021
- 2021Lessons on Parameter Sharing across Layers in Transformers13 Apr 2021
- 2021LazyFormer: Self Attention with Lazy Update25 Feb 2021
- 2021An Efficient Transformer Decoder with Compressed Sub-layers3 Jan 2021
- 2023Towards Efficient Generative Large Language Model Serving: A Survey from Algorithms to Systemscited by 101
- 2024Compressing KV Cache for Long-Context LLM Inference with Inter-Layer Attention Similaritycited by 11
- 2024Cross-layer Attention Sharing for Pre-trained Large Language Modelscited by 5
- 2025Round Attention: A Novel Round-Level Attention Mechanism to Accelerate LLM Inferencecited by 0
- 2025Round Attention: A Novel Round-Level Attention Mechanism to Accelerate LLM Inference21 Feb 2025
- 2024Compressing KV Cache for Long-Context LLM Inference with Inter-Layer Attention Similarity3 Dec 2024
- 2024Cross-layer Attention Sharing for Pre-trained Large Language Models4 Aug 2024
- 2023Towards Efficient Generative Large Language Model Serving: A Survey from Algorithms to Systems23 Dec 2023
- 2024KVSharer: Efficient Inference via Layer-Wise Dissimilar KV Cache Sharingcited by 38
- 2025Kascade: A Practical Sparse Attention Method for Long-Context LLM Inferencecited by 9
- 2025MASA: Rethinking the Representational Bottleneck in LoRA with Multi-A Shared Adaptationcited by 1
- 2026Attention Routing Stabilizes Early: Working-Set Inference for Recurrent Language Modelscited by 0
- 2026Attention Routing Stabilizes Early: Working-Set Inference for Recurrent Language Models23 Sep 2026
- 2025Kascade: A Practical Sparse Attention Method for Long-Context LLM Inference18 Dec 2025
- 2025MASA: Rethinking the Representational Bottleneck in LoRA with Multi-A Shared Adaptation7 Oct 2025
- 2024KVSharer: Efficient Inference via Layer-Wise Dissimilar KV Cache Sharing24 Oct 2024
- 2021Lessons on Parameter Sharing across Layers in Transformerscited by 85
- 2023Transformers in Speech Processing: A Surveycited by 38
- 2024Basis Sharing: Cross-Layer Parameter Sharing for Large Language Model Compressioncited by 34
- 2025Tensor Product Attention Is All You Needcited by 27
- 2023Skip-Attention: Improving Vision Transformers by Paying Less Attentioncited by 23
- 2021LazyFormer: Self Attention with Lazy Updatecited by 16
- 2021An Efficient Transformer Decoder with Compressed Sub-layerscited by 10
- 2021OH-Former: Omni-Relational High-Order Transformer for Person Re-Identificationcited by 10
- 2024Dynamic Layer Tying for Parameter-Efficient Transformerscited by 10
- 2022ACORT: A Compact Object Relation Transformer for Parameter Efficient Image Captioningcited by 6
- 2023Recycle-and-Distill: Universal Compression Strategy for Transformer-based Speech SSL Models with Attention Map Reusing and Masking Distillationcited by 6
- 2024Increasing Model Capacity for Free: A Simple Strategy for Parameter Efficient Fine-tuningcited by 6
12 most cited of 26 All 26 (JSON)
- 2025AttnCache: Accelerating Self-Attention Inference for LLM Prefill via Attention Cache29 Oct 2025
- 2025Deconstructing Attention: Investigating Design Principles for Effective Language Modeling13 Oct 2025
- 2025Layer-wise dynamic rank for compressing large language models30 Sep 2025
- 2025It Takes a Good Model to Train a Good Model: Generalized Gaussian Priors for Optimized LLMs31 May 2025
- 2025ACE: A Cardinality Estimator for Set-Valued Queries19 Mar 2025
- 2025Tensor Product Attention Is All You Need11 Jan 2025
- 2024PSformer: Parameter-efficient Transformer with Segment Attention for Time Series Forecasting3 Nov 2024
- 2024Basis Sharing: Cross-Layer Parameter Sharing for Large Language Model Compression2 Oct 2024
- 2024Increasing Model Capacity for Free: A Simple Strategy for Parameter Efficient Fine-tuning1 Jul 2024
- 2024Head-wise Shareable Attention for Large Language Models19 Feb 2024
- 2024Dynamic Layer Tying for Parameter-Efficient Transformers23 Jan 2024
- 2023Only 5% Attention Is All You Need: Efficient Long-range Document-level Neural Machine Translation25 Sep 2023
12 newest of 26 All 26 (JSON)