mlsanityData to 7 Oct 2026

Do Transformers Need Three Projections? Systematic Study of QKV Variants

arXiv 2606.04032 ↗First posted 1 Jun 2026cs.LG cs.AI cs.CLcited by 3

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

  1. 2017Attention Is All You Needcited by 59,932
  2. 2015Deep Residual Learning for Image Recognitioncited by 51,809
  3. 2021Learning Transferable Visual Models From Natural Language Supervisioncited by 33,456
  4. 2020Language Models are Few-Shot Learnerscited by 32,199
  5. 2020An Image is Worth 16x16 Words: Transformers for Image Recognition at Scalecited by 29,915
  6. 2019Exploring the Limits of Transfer Learning with a Unified Text-to-Text Transformercited by 14,491
  7. 2023Llama 2: Open Foundation and Fine-Tuned Chat Modelscited by 12,865
  8. 2017Graph Attention Networkscited by 8,854
  9. 2020Scaling Laws for Neural Language Modelscited by 6,411
  10. 2016Layer Normalizationcited by 5,579
  11. 2022PaLM: Scaling Language Modeling with Pathwayscited by 5,304
  12. 2023Mamba: Linear-Time Sequence Modeling with Selective State Spacescited by 4,616
38 more
  1. 2022Flamingo: a Visual Language Model for Few-Shot Learningcited by 4,337
  2. 2017Fashion-MNIST: a Novel Image Dataset for Benchmarking Machine Learning Algorithmscited by 4,283
  3. 2022FlashAttention: Fast and Memory-Efficient Exact Attention with IO-Awarenesscited by 3,611
  4. 2020Longformer: The Long-Document Transformercited by 3,362
  5. 2022Training Compute-Optimal Large Language Modelscited by 2,927
  6. 2020Transformers are RNNs: Fast Autoregressive Transformers with Linear Attentioncited by 1,979
  7. 2020Big Bird: Transformers for Longer Sequencescited by 1,673
  8. 2020Rethinking Attention with Performerscited by 1,584
  9. 2021TransUNet: Transformers Make Strong Encoders for Medical Image Segmentationcited by 1,552
  10. 2019Generating Long Sequences with Sparse Transformerscited by 1,523
  11. 2020Rethinking Semantic Segmentation from a Sequence-to-Sequence Perspective with Transformerscited by 1,368
  12. 2022SmoothQuant: Accurate and Efficient Post-Training Quantization for Large Language Modelscited by 1,338
  13. 2018Self-Attention with Relative Position Representationscited by 1,112
  14. 2024DeepSeek-V2: A Strong, Economical, and Efficient Mixture-of-Experts Language Modelcited by 1,093
  15. 2021CvT: Introducing Convolutions to Vision Transformerscited by 946
  16. 2023A Survey on Multimodal Large Language Modelscited by 898
  17. 2020Efficient Transformers: A Surveycited by 784
  18. 2019Fast Transformer Decoding: One Write-Head is All You Needcited by 700
  19. 2023Scaling Vision Transformers to 22 Billion Parameterscited by 690
  20. 2020A Survey on Visual Transformercited by 653
  21. 2023FlexGen: High-Throughput Generative Inference of Large Language Models with a Single GPUcited by 567
  22. 2022Efficiently Scaling Transformer Inferencecited by 529
  23. 2023The Falcon Series of Open Language Modelscited by 526
  24. 2023Scissorhands: Exploiting the Persistence of Importance Hypothesis for LLM KV Cache Compression at Test Timecited by 428
  25. 2023SpQR: A Sparse-Quantized Representation for Near-Lossless LLM Weight Compressioncited by 359
  26. 2017A simple neural network module for relational reasoningcited by 345
  27. 2016Using the Output Embedding to Improve Language Modelscited by 266
  28. 2021SOFT: Softmax-free Transformer with Linear Complexitycited by 128
  29. 2016Visual Question Answering: A Survey of Methods and Datasetscited by 123
  30. 2021An Attention Free Transformercited by 111
  31. 2020Improve Transformer Models with Better Relative Position Embeddingscited by 73
  32. 2019An Empirical Study of Spatial Attention Mechanisms in Deep Networkscited by 58
  33. 2020Attention Flows: Analyzing and Comparing Attention Mechanisms in Language Modelscited by 49
  34. 2023Simplifying Transformer Blockscited by 37
  35. 2022SimA: Simple Softmax-free Attention for Vision Transformerscited by 32
  36. 2022pNLP-Mixer: an Efficient all-MLP Architecture for Languagecited by 19
  37. 2022HyperMixer: An MLP-based Low Cost Alternative to Transformerscited by 13
  38. 2024Does Self-Attention Need Separate Weights in Transformers?cited by 4

50 most cited of 51 All 51 (JSON)

Built on by

Cited by year (2026 to 7 Oct)

Citing papers per year: 2026: 3.

3 papers.

Similar, not cited

Close in wording, no citation either way

  1. 2026Bridging KV-Cache Quantization and Linear Attention: From Theory to Pretrained Weight Migration
  2. 2024MLKV: Multi-Layer Key-Value Heads for Memory Efficient Transformer Decoding
  3. 2026Low-Rank Key Value Attention
  4. 2025The End of Transformers? On Challenging Attention and the Rise of Sub-Quadratic Architectures
  5. 2026Mixture-of-Top-k Attention: Efficient Attention via Scalable Fast Weights
  6. 2026Symmetry Breaking in Transformers for Efficient and Interpretable Training
  7. 2023SwitchHead: Accelerating Transformers with Mixture-of-Experts Attention
  8. 2026QVLA: Not All Channels Are Equal in Vision-Language-Action Model's Quantization