Depth-Attention: Cross-Layer Value Mixing for Language Models
For LLM practitioners, Depth-Attention offers a parameter-free, inference-efficient method to improve model quality by reusing earlier-layer representations, addressing a known bottleneck in residual stream information flow.
Depth-Attention improves language model performance by enabling cross-layer value mixing inside the attention module without adding parameters or inference state beyond the standard KV cache. On Qwen3-style decoders from 360M to 3B parameters, it achieves up to 2.3 accuracy points improvement over vanilla Transformers with under 0.01% extra FLOPs.
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.