Depth-Attention: Cross-Layer Value Mixing for Language Models
On the map: Efficient LLM inference and serving → KV cache optimization
Abstract (arXiv)
Self-attention selects information freely across the sequence, but across depth, Transformers merely add each layer's output to the residual stream, so later layers cannot selectively reuse earlier-layer representations. Recent cross-layer methods improve this flow but operate on hidden states outside attention, adding state beyond the key-value cache at inference--a cost that becomes increasingly salient as modern LLMs compress the cache with grouped-query and multi-head latent attention. We introduce Depth-Attention, which performs this selection inside the attention module itself: before a layer attends over the sequence, its query attends over the keys of earlier layers at the same token position and mixes their values into the value that self-attention then reads. Because Depth-Attention reuses the standard attention queries, keys, and value-cache slots, storing depth-mixed values in place of the original values, it adds no parameters and introduces no persistent inference state beyond the standard key-value cache--the same cache size as a vanilla decoder and less than hidden-state-based cross-layer methods. On Qwen3-style decoders at 1.5B and 3B parameters, Depth-Attention attains the lowest perplexity and the highest average downstream accuracy, improving over the vanilla Transformer by up to 2.3 accuracy points and surpassing strong cross-layer baselines in perplexity and average accuracy, while adding under 0.01% extra arithmetic FLOPs and no additional persistent inference state. The gains hold from 360M to 3B parameters and extend to looped Transformers.
Key facts. Cited by 3 papers in the map (data through 7 Oct 2026). Placed in KV cache optimization, one of 12 directions in Efficient LLM inference and serving. Builds on 25 of its 31 references that are in the map. First posted on arXiv 3 Jun 2026.
Builds on
25 of 31 references.
Closest ancestors · 1
Same direction, newest first
- 2024Value Residual Learning23 Oct
Other references · 24
- 2017Attention Is All You Needcited by 59,932
- 2015Deep Residual Learning for Image Recognitioncited by 51,809
- 2016Densely Connected Convolutional Networkscited by 6,691
- 2020Scaling Laws for Neural Language Modelscited by 6,411
- 2018Think you have Solved Question Answering? Try ARC, the AI2 Reasoning Challengecited by 4,710
- 2019HellaSwag: Can a Machine Really Finish Your Sentence?cited by 4,018
- 2019PIQA: Reasoning about Physical Commonsense in Natural Languagecited by 2,818
- 2018Deep contextualized word representationscited by 2,285
- 2023GQA: Training Generalized Multi-Query Transformer Models from Multi-Head Checkpointscited by 1,311
- 2024DeepSeek-V2: A Strong, Economical, and Efficient Mixture-of-Experts Language Modelcited by 1,093
- 2016The LAMBADA dataset: Word prediction requiring a broad discourse contextcited by 960
- 2020On Layer Normalization in the Transformer Architecturecited by 906
12 more
- 2017RACE: Large-scale ReAding Comprehension Dataset From Examinationscited by 851
- 2017Crowdsourcing Multiple Choice Science Questionscited by 742
- 2020ReZero is All You Need: Fast Convergence at Large Depthcited by 260
- 2022DeepNet: Scaling Transformers to 1,000 Layerscited by 176
- 2025Mixture-of-Recursions: Learning Dynamic Recursive Depths for Adaptive Token-Level Computationcited by 120
- 2020RealFormer: Transformer Likes Residual Attentioncited by 62
- 2024DenseFormer: Enhancing Information Flow in Transformers via Depth Weighted Averagingcited by 48
- 2025PonderLM: Pretraining Language Models to Ponder in Continuous Spacecited by 44
- 2023ResiDual: Transformer with Dual Residual Connectionscited by 30
- 2025MUDDFormer: Breaking Residual Bottlenecks in Transformers via Multiway Dynamic Dense Connectionscited by 27
- 2026Depth-Recurrent Attention Mixtures: Giving Latent Reasoning the Attention it Deservescited by 11
- 2026ANCRe: Adaptive Neural Connection Reassignment for Efficient Depth Scalingcited by 4
Built on by
Citing papers per year: 2026: 3.
3 papers.
- 2026Role-Decoupled Attention Residuals: Separating Matching and Content Retrieval Across Depthcited by 0
- 2026NCP-ArchPreview Technical Report: Moving towards Latent Space Language Models through Next Concept Predictioncited by 0
- 2026DepthBench: Measuring How Residual Connections Enable More Computational Depthcited by 0
- 2026DepthBench: Measuring How Residual Connections Enable More Computational Depth26 Sep 2026
- 2026NCP-ArchPreview Technical Report: Moving towards Latent Space Language Models through Next Concept Prediction9 Sep 2026
- 2026Role-Decoupled Attention Residuals: Separating Matching and Content Retrieval Across Depth2 Aug 2026
Similar, not cited
Close in wording, no citation either way
- 2024Cross-layer Attention Sharing for Pre-trained Large Language Models
- 2024Beyond KV Caching: Shared Attention for Efficient LLMs
- 2026Q Cache: Visual Attention is Valuable in Less than Half of Decode Layers for Multimodal Large Language Model
- 2024Reducing Transformer Key-Value Cache Size with Cross-Layer Attention
- 2026ART: Attention Run-time Termination for Efficient Large Language Model Decoding
- 2026Sliding-window beats linear attention
- 2026IndexCache: Accelerating Sparse Attention via Cross-Layer Index Reuse
- 2026GLIDE: Guided Layerwise Hybrid Attention for Efficient LLM Inference