A3: Accelerating Attention Mechanisms in Neural Networks with Approximation
Not on the map. Why?
Abstract (arXiv)
With the increasing computational demands of neural networks, many hardware accelerators for the neural networks have been proposed. Such existing neural network accelerators often focus on popular neural network types such as convolutional neural networks (CNNs) and recurrent neural networks (RNNs); however, not much attention has been paid to attention mechanisms, an emerging neural network primitive that enables neural networks to retrieve most relevant information from a knowledge-base, external memory, or past states. The attention mechanism is widely adopted by many state-of-the-art neural networks for computer vision, natural language processing, and machine translation, and accounts for a large portion of total execution time. We observe today's practice of implementing this mechanism using matrix-vector multiplication is suboptimal as the attention mechanism is semantically a content-based search where a large portion of computations ends up not being used. Based on this observation, we design and architect A3, which accelerates attention mechanisms in neural networks with algorithmic approximation and hardware specialization. Our proposed accelerator achieves multiple orders of magnitude improvement in energy efficiency (performance/watt) as well as substantial speedup over the state-of-the-art conventional hardware.
Key facts. Cited by 97 papers in the map (data through 7 Oct 2026). First posted on arXiv 22 Feb 2020.
Builds on
No reference list yet.
Built on by
Citing papers per year: 2021: 5; 2022: 16; 2023: 13; 2024: 29; 2025: 29; 2026: 5.
97 papers.
Sort:
By area:
- 2024QServe: W4A8KV4 Quantization and System Co-design for Efficient LLM Servingcited by 200
- 2022A Fast Post-Training Pruning Framework for Transformerscited by 161
- 2023Full Stack Optimization of Transformer Inference: a Surveycited by 79
- 2022Efficient Methods for Natural Language Processing: A Surveycited by 78
- 2024FlightLLM: Efficient Large Language Model Inference with a Complete Mapping Flow on FPGAscited by 60
- 2022DFX: A Low-latency Multi-FPGA Appliance for Accelerating Transformer-based Text Generationcited by 46
- 2024Mixture of Attention Spans: Optimizing LLM Inference Efficiency with Heterogeneous Sliding-Window Lengthscited by 44
- 2022ViTCoD: Vision Transformer Acceleration via Dedicated Algorithm and Accelerator Co-Designcited by 41
- 2021FLAT: An Optimized Dataflow for Mitigating Attention Bottleneckscited by 38
- 2024Lightweight Deep Learning for Resource-Constrained Environments: A Surveycited by 38
- 2023A Survey on Efficient Training of Transformerscited by 37
- 2023A Survey of Techniques for Optimizing Transformer Inferencecited by 35
38 more
- 2024Duplex: A Device for Large Language Models with Mixture of Experts, Grouped Query Attention, and Continuous Batchingcited by 33
- 2022Enable Deep Learning on Mobile Devices: Methods, Systems, and Applicationscited by 32
- 2022SimA: Simple Softmax-free Attention for Vision Transformerscited by 32
- 2024LLMServingSim: A HW/SW Co-Simulation Infrastructure for LLM Inference Serving at Scalecited by 30
- 2022Adaptable Butterfly Accelerator for Attention-based NNs via Hardware and Algorithm Co-designcited by 29
- 2022ViTALiTy: Unifying Low-rank and Sparse Approximation for Vision Transformer Acceleration with a Linear Taylor Attentioncited by 29
- 2022An Algorithm-Hardware Co-Optimized Framework for Accelerating N:M Sparse Transformerscited by 28
- 2021Accelerating Framework of Transformer by Hardware Design and Model Compression Co-Optimizationcited by 27
- 2021Energon: Towards Efficient Acceleration of Transformers Using Dynamic Sparse Attentioncited by 26
- 2023X-Former: In-Memory Acceleration of Transformerscited by 25
- 2023The Efficiency Spectrum of Large Language Models: An Algorithmic Surveycited by 25
- 2023AccelTran: A Sparsity-Aware Accelerator for Dynamic Inference with Transformerscited by 24
- 2023Chameleon: a Heterogeneous and Disaggregated Accelerator System for Retrieval-Augmented Language Modelscited by 23
- 2022A Length Adaptive Algorithm-Hardware Co-design of Transformer on FPGA Through Sparse Attention and Dynamic Pipeliningcited by 21
- 2021Transformer Acceleration with Dynamic Sparse Attentioncited by 19
- 2025PAPI: Exploiting Dynamic Parallelism in Large Language Model Decoding with a Processing-In-Memory-Enabled Computing Systemcited by 19
- 2023Integer or Floating Point? New Outlooks for Low-Bit Quantization on Large Language Modelscited by 18
- 2023SwiftTron: An Efficient Hardware Accelerator for Quantized Transformerscited by 17
- 2024A Survey on Hardware Accelerators for Large Language Modelscited by 15
- 2024ConSmax: Hardware-Friendly Alternative Softmax with Learnable Parameterscited by 14
- 2024CHAI: Clustered Head Attention for Efficient LLM Inferencecited by 13
- 2022Sparse Attention Acceleration with Synergistic In-Memory Pruning and On-Chip Recomputationcited by 12
- 2022SALO: An Efficient Spatial Accelerator Enabling Hybrid Sparse Attention Mechanisms for Long Sequencescited by 10
- 2025QUARK: Quantization-Enabled Circuit Sharing for Transformer Acceleration by Exploiting Common Patterns in Nonlinear Operationscited by 10
- 2025Stratum: System-Hardware Co-Design with Tiered Monolithic 3D-Stackable DRAM for Efficient MoE Servingcited by 9
- 2022Accelerating Attention through Gradient-Based Learned Runtime Pruningcited by 8
- 2025QuartDepth: Post-Training Quantization for Real-Time Depth Estimation on the Edgecited by 7
- 2023Sparse-DySta: Sparsity-Aware Dynamic and Static Scheduling for Sparse Multi-DNN Workloadscited by 6
- 2024BETA: Binarized Energy-Efficient Transformer Accelerator at the Edgecited by 6
- 20241.5-Pints Technical Report: Pretraining in Days, Not Months -- Your Language Model Thrives on Quality Datacited by 6
- 2024Cross-layer Attention Sharing for Pre-trained Large Language Modelscited by 5
- 2025EXION: Exploiting Inter- and Intra-Iteration Output Sparsity for Diffusion Modelscited by 5
- 2025Low-Cost FlashAttention with Fused Exponential and Multiplication Hardware Operatorscited by 5
- 2025SOLE: Hardware-Software Co-design of Softmax and LayerNorm for Efficient Transformer Inferencecited by 5
- 2022GROW: A Row-Stationary Sparse-Dense GEMM Accelerator for Memory-Efficient Graph Convolutional Neural Networkscited by 4
- 2024Deploying Foundation Model Powered Agent Services: A Surveycited by 4
- 2025Fine-Grained Image-Text Correspondence with Cost Aggregation for Open-Vocabulary Part Segmentationcited by 4
- 2025MCBP: A Memory-Compute Efficient LLM Inference Accelerator Leveraging Bit-Slice-enabled Sparsity and Repetitivenesscited by 4
50 most cited of 97 All 97 (JSON)
- 2026Understanding and Exploiting Diagonal Attention Sparsity in Autoregressive Image Generation17 Sep 2026
- 2026MiX: Micro-Inverted-Scaling for End-to-End Low-Bit Vision-Language Model Acceleration17 Sep 2026
- 2026Attention at the Theoretical Minimum: A Mathematics of Arrays Framework for Memory-Optimal Transformer Kernels5 Jun 2026
- 2026Salca: A Sparsity-Aware Hardware Accelerator for Efficient Long-Context Attention Decoding27 Apr 2026
- 2026Focus Session: Hardware and Software Techniques for Accelerating Multimodal Foundation Models23 Apr 2026
- 2025BitStopper: An Efficient Transformer Attention Accelerator via Stage-fusion and Early Termination6 Dec 2025
- 2025ESACT: An End-to-End Sparse Accelerator for Compute-Intensive Transformers via Local Similarity2 Dec 2025
- 2025LAPA: Log-Domain Prediction-Driven Dynamic Sparsity Accelerator for Transformer Model26 Nov 2025
- 2025CAMformer: Associative Memory is All You Need24 Nov 2025
- 2025End-to-End Transformer Acceleration Through Processing-in-Memory Architectures21 Nov 2025
- 2025QUARK: Quantization-Enabled Circuit Sharing for Transformer Acceleration by Exploiting Common Patterns in Nonlinear Operations10 Nov 2025
- 2025AttnCache: Accelerating Self-Attention Inference for LLM Prefill via Attention Cache29 Oct 2025
38 more
- 2025SOLE: Hardware-Software Co-design of Softmax and LayerNorm for Efficient Transformer Inference20 Oct 2025
- 2025Low Power Vision Transformer Accelerator with Hardware-Aware Pruning and Optimized Dataflow16 Oct 2025
- 2025Bhasha-Rupantarika: Algorithm-Hardware Co-design approach for Multilingual Neural Machine Translation12 Oct 2025
- 2025Vectorized FlashAttention with Low-cost Exponential Computation in RISC-V Vector Processors8 Oct 2025
- 2025Stratum: System-Hardware Co-Design with Tiered Monolithic 3D-Stackable DRAM for Efficient MoE Serving6 Oct 2025
- 2025LEGO: Spatial Accelerator Generation and Optimization for Tensor Applications15 Sep 2025
- 2025MCBP: A Memory-Compute Efficient LLM Inference Accelerator Leveraging Bit-Slice-enabled Sparsity and Repetitiveness12 Sep 2025
- 2025VEDA: Efficient LLM Generation Through Voting-based KV Cache Eviction and Dataflow-flexible Accelerator1 Jul 2025
- 2025Accelerating Sparse Transformer Inference on GPU6 Jun 2025
- 2025Titanus: Enabling KV Cache Pruning and Quantization On-the-Fly for LLM Acceleration23 May 2025
- 2025Low-Cost FlashAttention with Fused Exponential and Multiplication Hardware Operators20 May 2025
- 2025FLASH-D: FlashAttention with Hidden Softmax Division20 May 2025
- 2025LightNobel: Improving Sequence Length Limitation in Protein Structure Prediction Model via Adaptive Activation Quantization9 May 2025
- 2025Uni-Render: A Unified Accelerator for Real-Time Rendering Across Diverse Neural Renderers31 Mar 2025
- 2025A Low-Power Streaming Speech Enhancement Accelerator For Edge Devices27 Mar 2025
- 2025QuartDepth: Post-Training Quantization for Real-Time Depth Estimation on the Edge20 Mar 2025
- 2025Attention Condensation via Sparsity Induced Regularized Training3 Mar 2025
- 2025PAPI: Exploiting Dynamic Parallelism in Large Language Model Decoding with a Processing-In-Memory-Enabled Computing System21 Feb 2025
- 2025Top-Theta Attention: Sparsifying Transformers by Compensated Thresholding12 Feb 2025
- 2025Fine-Grained Image-Text Correspondence with Cost Aggregation for Open-Vocabulary Part Segmentation16 Jan 2025
- 2025ANSR-DT: A Neuro-Symbolic Framework for Adaptive and Explainable Digital Twins15 Jan 2025
- 2025EXION: Exploiting Inter- and Intra-Iteration Output Sparsity for Diffusion Models10 Jan 2025
- 2024Deploying Foundation Model Powered Agent Services: A Survey18 Dec 2024
- 2024A Runtime-Adaptive Transformer Neural Network Accelerator on FPGAs27 Nov 2024
- 2024Unified Error Correction Code Transformer with Low Complexity4 Oct 2024
- 2024EXAQ: Exponent Aware Quantization For LLMs Acceleration4 Oct 2024
- 2024FAMOUS: Flexible Accelerator for the Attention Mechanism of Transformer on UltraScale+ FPGAs21 Sep 2024
- 2024ProTEA: Programmable Transformer Encoder Acceleration on FPGA21 Sep 2024
- 2024An Analog and Digital Hybrid Attention Accelerator for Transformers with Charge-based In-memory Computing8 Sep 2024
- 2024Hardware Acceleration of LLMs: A comprehensive survey and comparison5 Sep 2024
- 2024Duplex: A Device for Large Language Models with Mixture of Experts, Grouped Query Attention, and Continuous Batching2 Sep 2024
- 2024LLMServingSim: A HW/SW Co-Simulation Infrastructure for LLM Inference Serving at Scale10 Aug 2024
- 20241.5-Pints Technical Report: Pretraining in Days, Not Months -- Your Language Model Thrives on Quality Data7 Aug 2024
- 2024Cross-layer Attention Sharing for Pre-trained Large Language Models4 Aug 2024
- 2024KWT-Tiny: RISC-V Accelerated, Embedded Keyword Spotting Transformer22 Jul 2024
- 2024Hybrid Dynamic Pruning: A Pathway to Efficient Transformer Inference17 Jul 2024
- 2024Co-Designing Binarized Transformer and Hardware Accelerator for Efficient End-to-End Edge Deployment16 Jul 2024
- 2024Mixture of Attention Spans: Optimizing LLM Inference Efficiency with Heterogeneous Sliding-Window Lengths21 Jun 2024
50 newest of 97 All 97 (JSON)
- 2024QServe: W4A8KV4 Quantization and System Co-design for Efficient LLM Servingcited by 200
- 2024FlightLLM: Efficient Large Language Model Inference with a Complete Mapping Flow on FPGAscited by 60
- 2024LLMServingSim: A HW/SW Co-Simulation Infrastructure for LLM Inference Serving at Scalecited by 30
- 2025PAPI: Exploiting Dynamic Parallelism in Large Language Model Decoding with a Processing-In-Memory-Enabled Computing Systemcited by 19
- 2024CHAI: Clustered Head Attention for Efficient LLM Inferencecited by 13
- 2024Cross-layer Attention Sharing for Pre-trained Large Language Modelscited by 5
- 2025MCBP: A Memory-Compute Efficient LLM Inference Accelerator Leveraging Bit-Slice-enabled Sparsity and Repetitivenesscited by 4
- 2024The CAP Principle for LLM Serving: A Survey of Long-Context Large Language Model Servingcited by 3
- 2025Titanus: Enabling KV Cache Pruning and Quantization On-the-Fly for LLM Accelerationcited by 2
- 2025VEDA: Efficient LLM Generation Through Voting-based KV Cache Eviction and Dataflow-flexible Acceleratorcited by 1
- 2025End-to-End Transformer Acceleration Through Processing-in-Memory Architecturescited by 1
- 2026Understanding and Exploiting Diagonal Attention Sparsity in Autoregressive Image Generationcited by 0
- 2026Understanding and Exploiting Diagonal Attention Sparsity in Autoregressive Image Generation17 Sep 2026
- 2025End-to-End Transformer Acceleration Through Processing-in-Memory Architectures21 Nov 2025
- 2025MCBP: A Memory-Compute Efficient LLM Inference Accelerator Leveraging Bit-Slice-enabled Sparsity and Repetitiveness12 Sep 2025
- 2025VEDA: Efficient LLM Generation Through Voting-based KV Cache Eviction and Dataflow-flexible Accelerator1 Jul 2025
- 2025Titanus: Enabling KV Cache Pruning and Quantization On-the-Fly for LLM Acceleration23 May 2025
- 2025PAPI: Exploiting Dynamic Parallelism in Large Language Model Decoding with a Processing-In-Memory-Enabled Computing System21 Feb 2025
- 2024LLMServingSim: A HW/SW Co-Simulation Infrastructure for LLM Inference Serving at Scale10 Aug 2024
- 2024Cross-layer Attention Sharing for Pre-trained Large Language Models4 Aug 2024
- 2024The CAP Principle for LLM Serving: A Survey of Long-Context Large Language Model Serving18 May 2024
- 2024QServe: W4A8KV4 Quantization and System Co-design for Efficient LLM Serving7 May 2024
- 2024CHAI: Clustered Head Attention for Efficient LLM Inference12 Mar 2024
- 2024FlightLLM: Efficient Large Language Model Inference with a Complete Mapping Flow on FPGAs8 Jan 2024
- 2024Deploying Foundation Model Powered Agent Services: A Survey18 Dec 2024
- 2022A Fast Post-Training Pruning Framework for Transformerscited by 161
- 2023Full Stack Optimization of Transformer Inference: a Surveycited by 79
- 2022Efficient Methods for Natural Language Processing: A Surveycited by 78
- 2022DFX: A Low-latency Multi-FPGA Appliance for Accelerating Transformer-based Text Generationcited by 46
- 2024Mixture of Attention Spans: Optimizing LLM Inference Efficiency with Heterogeneous Sliding-Window Lengthscited by 44
- 2022ViTCoD: Vision Transformer Acceleration via Dedicated Algorithm and Accelerator Co-Designcited by 41
- 2021FLAT: An Optimized Dataflow for Mitigating Attention Bottleneckscited by 38
- 2024Lightweight Deep Learning for Resource-Constrained Environments: A Surveycited by 38
- 2023A Survey on Efficient Training of Transformerscited by 37
- 2023A Survey of Techniques for Optimizing Transformer Inferencecited by 35
- 2022Enable Deep Learning on Mobile Devices: Methods, Systems, and Applicationscited by 32
- 2022SimA: Simple Softmax-free Attention for Vision Transformerscited by 32
12 most cited of 77 All 77 (JSON)
- 2026Attention at the Theoretical Minimum: A Mathematics of Arrays Framework for Memory-Optimal Transformer Kernels5 Jun 2026
- 2026Salca: A Sparsity-Aware Hardware Accelerator for Efficient Long-Context Attention Decoding27 Apr 2026
- 2026Focus Session: Hardware and Software Techniques for Accelerating Multimodal Foundation Models23 Apr 2026
- 2025BitStopper: An Efficient Transformer Attention Accelerator via Stage-fusion and Early Termination6 Dec 2025
- 2025ESACT: An End-to-End Sparse Accelerator for Compute-Intensive Transformers via Local Similarity2 Dec 2025
- 2025LAPA: Log-Domain Prediction-Driven Dynamic Sparsity Accelerator for Transformer Model26 Nov 2025
- 2025CAMformer: Associative Memory is All You Need24 Nov 2025
- 2025QUARK: Quantization-Enabled Circuit Sharing for Transformer Acceleration by Exploiting Common Patterns in Nonlinear Operations10 Nov 2025
- 2025AttnCache: Accelerating Self-Attention Inference for LLM Prefill via Attention Cache29 Oct 2025
- 2025SOLE: Hardware-Software Co-design of Softmax and LayerNorm for Efficient Transformer Inference20 Oct 2025
- 2025Low Power Vision Transformer Accelerator with Hardware-Aware Pruning and Optimized Dataflow16 Oct 2025
- 2025Bhasha-Rupantarika: Algorithm-Hardware Co-design approach for Multilingual Neural Machine Translation12 Oct 2025
12 newest of 77 All 77 (JSON)