Do Transformers Need Three Projections? Systematic Study of QKV Variants
Not on the map. Why?
Abstract (arXiv)
Transformers have become the standard solution for various AI tasks, with the query, key, and value (QKV) attention formulation playing a central role. However, the individual contribution of these three projections and the impact of omitting some remain poorly understood. We systematically evaluate three projection sharing constraints: a) Q-K=V (shared key-value), b) Q=K-V (shared query-key), and c) Q=K=V (single projection). The last two variants produce symmetric attention maps; to address this, we also explore asymmetric attention via 2D positional encodings. Through experiments spanning synthetic tasks, vision (MNIST, CIFAR, TinyImageNet, anomaly), and language modeling (300M and 1.2B parameter models on 10B tokens), we discovered that our transformers perform on par or occasionally better than the QKV transformer. In language modeling, Q-K=V projection sharing achieves 50% KV cache reduction with only 3.1% perplexity degradation. Crucially, projection sharing is complementary to head sharing (GQA/MQA): combining Q-K=V with GQA-4 yields 87.5% cache reduction, while Q-K=V + MQA achieves 96.9%, enabling practical on-device inference. We show that Q-K=V preserves quality because keys and values can occupy similar representational spaces and attention operates in a low-rank regime, whereas Q=K-V breaks attention directionality. Our results systematically characterize projection sharing as an underexplored instance of weight tying in attention, with direct, quantifiable inference memory benefits, particularly valuable for edge deployment. The code is publicly available at https://github.com/Brainchip-Inc/Do-Transformers-Need-3-Projections
Key facts. Cited by 3 papers in the map (data through 7 Oct 2026). Builds on 51 of its 63 references that are in the map. First posted on arXiv 1 Jun 2026.
Builds on
51 of 63 references.
References · 51
- 2017Attention Is All You Needcited by 59,932
- 2015Deep Residual Learning for Image Recognitioncited by 51,809
- 2021Learning Transferable Visual Models From Natural Language Supervisioncited by 33,456
- 2020Language Models are Few-Shot Learnerscited by 32,199
- 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
- 2023Llama 2: Open Foundation and Fine-Tuned Chat Modelscited by 12,865
- 2017Graph Attention Networkscited by 8,854
- 2020Scaling Laws for Neural Language Modelscited by 6,411
- 2016Layer Normalizationcited by 5,579
- 2022PaLM: Scaling Language Modeling with Pathwayscited by 5,304
- 2023Mamba: Linear-Time Sequence Modeling with Selective State Spacescited by 4,616
38 more
- 2022Flamingo: a Visual Language Model for Few-Shot Learningcited by 4,337
- 2017Fashion-MNIST: a Novel Image Dataset for Benchmarking Machine Learning Algorithmscited by 4,283
- 2022FlashAttention: Fast and Memory-Efficient Exact Attention with IO-Awarenesscited by 3,611
- 2020Longformer: The Long-Document Transformercited by 3,362
- 2022Training Compute-Optimal Large Language Modelscited by 2,927
- 2020Transformers are RNNs: Fast Autoregressive Transformers with Linear Attentioncited by 1,979
- 2020Big Bird: Transformers for Longer Sequencescited by 1,673
- 2020Rethinking Attention with Performerscited by 1,584
- 2021TransUNet: Transformers Make Strong Encoders for Medical Image Segmentationcited by 1,552
- 2019Generating Long Sequences with Sparse Transformerscited by 1,523
- 2020Rethinking Semantic Segmentation from a Sequence-to-Sequence Perspective with Transformerscited by 1,368
- 2022SmoothQuant: Accurate and Efficient Post-Training Quantization for Large Language Modelscited by 1,338
- 2018Self-Attention with Relative Position Representationscited by 1,112
- 2024DeepSeek-V2: A Strong, Economical, and Efficient Mixture-of-Experts Language Modelcited by 1,093
- 2021CvT: Introducing Convolutions to Vision Transformerscited by 946
- 2023A Survey on Multimodal Large Language Modelscited by 898
- 2020Efficient Transformers: A Surveycited by 784
- 2019Fast Transformer Decoding: One Write-Head is All You Needcited by 700
- 2023Scaling Vision Transformers to 22 Billion Parameterscited by 690
- 2020A Survey on Visual Transformercited by 653
- 2023FlexGen: High-Throughput Generative Inference of Large Language Models with a Single GPUcited by 567
- 2022Efficiently Scaling Transformer Inferencecited by 529
- 2023The Falcon Series of Open Language Modelscited by 526
- 2023Scissorhands: Exploiting the Persistence of Importance Hypothesis for LLM KV Cache Compression at Test Timecited by 428
- 2023SpQR: A Sparse-Quantized Representation for Near-Lossless LLM Weight Compressioncited by 359
- 2017A simple neural network module for relational reasoningcited by 345
- 2016Using the Output Embedding to Improve Language Modelscited by 266
- 2021SOFT: Softmax-free Transformer with Linear Complexitycited by 128
- 2016Visual Question Answering: A Survey of Methods and Datasetscited by 123
- 2021An Attention Free Transformercited by 111
- 2020Improve Transformer Models with Better Relative Position Embeddingscited by 73
- 2019An Empirical Study of Spatial Attention Mechanisms in Deep Networkscited by 58
- 2020Attention Flows: Analyzing and Comparing Attention Mechanisms in Language Modelscited by 49
- 2023Simplifying Transformer Blockscited by 37
- 2022SimA: Simple Softmax-free Attention for Vision Transformerscited by 32
- 2022pNLP-Mixer: an Efficient all-MLP Architecture for Languagecited by 19
- 2022HyperMixer: An MLP-based Low Cost Alternative to Transformerscited by 13
- 2024Does Self-Attention Need Separate Weights in Transformers?cited by 4
50 most cited of 51 All 51 (JSON)
Built on by
Citing papers per year: 2026: 3.
3 papers.
- 2026Gemma 4 Technical Reportcited by 187
- 2026Keyless Attention: Value-Space Routing and Value-Only Caching for Efficient Transformerscited by 0
- 2026All Routes Lead to Collapsecited by 0
- 2026Gemma 4 Technical Report2 Jul 2026
- 2026All Routes Lead to Collapse21 Jun 2026
- 2026Keyless Attention: Value-Space Routing and Value-Only Caching for Efficient Transformers20 Jun 2026
Similar, not cited
Close in wording, no citation either way
- 2026Bridging KV-Cache Quantization and Linear Attention: From Theory to Pretrained Weight Migration
- 2024MLKV: Multi-Layer Key-Value Heads for Memory Efficient Transformer Decoding
- 2026Low-Rank Key Value Attention
- 2025The End of Transformers? On Challenging Attention and the Rise of Sub-Quadratic Architectures
- 2026Mixture-of-Top-k Attention: Efficient Attention via Scalable Fast Weights
- 2026Symmetry Breaking in Transformers for Efficient and Interpretable Training
- 2023SwitchHead: Accelerating Transformers with Mixture-of-Experts Attention
- 2026QVLA: Not All Channels Are Equal in Vision-Language-Action Model's Quantization