Rethinking LLM Inference Bottlenecks: Insights from Latent Attention and Mixture-of-Experts
On the map: Mixture-of-experts → Efficient MoE training and inference systems
Abstract (arXiv)
Computational workloads composing traditional transformer models are starkly bifurcated. Multi-Head Attention (MHA) and Grouped-Query Attention are memory-bound due to low arithmetic intensity, while FeedForward Networks are compute-bound. This dichotomy has long motivated research into specialized hardware to mitigate the attention bottleneck. This paper argues that recent architectural advances in transformer models -- Multi-head Latent Attention (MLA) and Mixture of Experts (MoE) -- introduce new dominant bottlenecks, shifting the challenge away from memory-intensive attention. We make two key observations. First, the arithmetic intensity of MLA is over two orders of magnitude higher than that of MHA, moving it toward a compute-bound regime well-matched to modern accelerators such as GPUs. Second, distributing MoE experts across a pool of accelerators allows batching to tune their arithmetic intensity to that of dense layers, producing a more balanced computational profile. Consequently, the focus of hardware and system optimization should shift from attention acceleration to high-bandwidth interconnects and balancing expert workloads across accelerators.
Key facts. Cited by 4 papers in the map (data through 7 Oct 2026). Placed in Efficient MoE training and inference systems, one of 12 directions in Mixture-of-experts. Builds on 34 of its 59 references that are in the map. First posted on arXiv 21 Jul 2025.
Builds on
34 of 59 references.
Closest ancestors · 4
Same direction, newest first
- 2025FAST: An Efficient Scheduler for All-to-All GPU Communication14 May
- 2025Insights into DeepSeek-V3: Scaling Challenges and Reflections on Hardware for AI Architectures14 May
- 2025Comet: Fine-grained Computation-communication Overlapping for Mixture-of-Experts27 Feb
- 2024Duplex: A Device for Large Language Models with Mixture of Experts, Grouped Query Attention, and Continuous Batching2 Sep
Other references · 30
- 2017Attention Is All You Needcited by 59,932
- 2020Scaling Laws for Neural Language Modelscited by 6,411
- 2023Mamba: Linear-Time Sequence Modeling with Selective State Spacescited by 4,616
- 2021RoFormer: Enhanced Transformer with Rotary Position Embeddingcited by 4,581
- 2022FlashAttention: Fast and Memory-Efficient Exact Attention with IO-Awarenesscited by 3,611
- 2017Outrageously Large Neural Networks: The Sparsely-Gated Mixture-of-Experts Layercited by 3,413
- 2021Switch Transformers: Scaling to Trillion Parameter Models with Simple and Efficient Sparsitycited by 3,014
- 2019Megatron-LM: Training Multi-Billion Parameter Language Models Using Model Parallelismcited by 1,829
- 2024Mixtral of Expertscited by 1,785
- 2020GLU Variants Improve Transformercited by 1,724
- 2022SmoothQuant: Accurate and Efficient Post-Training Quantization for Large Language Modelscited by 1,338
- 2023GQA: Training Generalized Multi-Query Transformer Models from Multi-Head Checkpointscited by 1,311
18 more
- 2023AWQ: Activation-aware Weight Quantization for LLM Compression and Accelerationcited by 1,234
- 2022LLM.int8(): 8-bit Matrix Multiplication for Transformers at Scalecited by 881
- 2025Large Language Diffusion Modelscited by 872
- 2024DeepSeekMoE: Towards Ultimate Expert Specialization in Mixture-of-Experts Language Modelscited by 834
- 2019Fast Transformer Decoding: One Write-Head is All You Needcited by 700
- 2024DistServe: Disaggregating Prefill and Decoding for Goodput-optimized Large Language Model Servingcited by 595
- 2023Splitwise: Efficient generative LLM inference using phase splittingcited by 536
- 2022Efficiently Scaling Transformer Inferencecited by 529
- 2024Taming Throughput-Latency Tradeoff in LLM Inference with Sarathi-Servecited by 507
- 2022DeepSpeed Inference: Enabling Efficient Inference of Transformer Models at Unprecedented Scalecited by 403
- 2023Atom: Low-bit Quantization for Efficient and Accurate LLM Servingcited by 255
- 2024QServe: W4A8KV4 Quantization and System Co-design for Efficient LLM Servingcited by 200
- 2023FP8-LM: Training FP8 Large Language Modelscited by 75
- 2024NeuPIMs: NPU-PIM Heterogeneous Acceleration for Batched LLM Inferencingcited by 48
- 2025PIM Is All You Need: A CXL-Enabled GPU-Free System for Large Language Model Inferencecited by 22
- 2025PrefillOnly: An Inference Engine for Prefill-only Workloads in Large Language Model Applicationscited by 21
- 2024QQQ: Quality Quattuor-Bit Quantization for Large Language Modelscited by 14
- 2024Demystifying AI Platform Design for Distributed Inference of Next-Generation LLM modelscited by 12
Built on by
Citing papers per year: 2025: 2; 2026: 2.
4 papers.
- 2025From Tokens to Layers: Redefining Stall-Free Scheduling for MoE Serving with Layered Prefillcited by 3
- 2025Adaptive and Resource-efficient Agentic AI Systems for Mobile and Embedded Devices: A Surveycited by 2
- 2026Rethinking Network Topologies for Cost-Effective Mixture-of-Experts LLM Servingcited by 0
- 2026Think Before You Grid-Search: Floor-First Triage for LLM Servingcited by 0
- 2026Think Before You Grid-Search: Floor-First Triage for LLM Serving7 Jul 2026
- 2026Rethinking Network Topologies for Cost-Effective Mixture-of-Experts LLM Serving30 Apr 2026
- 2025From Tokens to Layers: Redefining Stall-Free Scheduling for MoE Serving with Layered Prefill9 Oct 2025
- 2025Adaptive and Resource-efficient Agentic AI Systems for Mobile and Embedded Devices: A Survey30 Sep 2025
Similar, not cited
Close in wording, no citation either way
- 2023SwitchHead: Accelerating Transformers with Mixture-of-Experts Attention
- 2023AttMEMO : Accelerating Transformers with Memoization on Big Memory Systems
- 2026Beyond Capacity: Scalable MoE LLM Inference via High-Bandwidth Flash with Direct GPU and HBM Paths
- 2026Architecture-Aware LLM Inference Optimization on AMD Instinct GPUs: A Comprehensive Benchmark and Deployment Study
- 2025Patterns behind Chaos: Forecasting Data Movement for Efficient Large-Scale MoE LLM Inference
- 2026AI Hardware Accelerators for Large Language Models: Architectures and the Memory Wall ↗
- 2026Understanding Inference Scaling for LLMs: Bottlenecks, Trade-offs, and Performance Principles
- 2026Revealing the Challenges of Attention-FFN Disaggregation for Modern MoE Models and Hardware Systems