Improving Model Representation and Reducing KV Cache via Skip Connections with First Value Heads
On the map: Efficient LLM inference and serving → KV cache optimization
Abstract (arXiv)
Transformer models have driven breakthroughs across various language tasks by their strong capability to learn rich contextual representations. Scaling them to improve representation, however, often demands substantial memory and compute costs, such as the Key-Value (KV) cache used during auto-regressive decoding. Skip connections offer a promising way to improve representation without bloating resource usage, yet most prior works either improve expressivity while leaving KV costs unchanged, or reduce memory at the cost of weaker representation. In this work, we propose SkipV1Former, a Transformer variant that uses skip connections from the first layer's Value heads to strengthen model representation and reduce KV cache. Specifically, from the second block onward, each layer reuses half of its Value heads from the very first layer, while computing the other half as usual-cutting Value projections and V cache by nearly 50 %. Theoretically, we show that routing uncompressed first-layer Values into deeper layers restores information lost to compression and accelerates the model's implicit mesa-optimization-a key pattern of Transformer in auto-regressive tasks. Empirically, across different model scales, SkipV1Former delivers consistent reductions of approximately 25 % in KV cache while improving perplexity relative to standard Multi-Head Attention (MHA) Transformers and some advanced variants. Moreover, we propose a recipe for uptraining existing MHA Transformer checkpoints to SkipV1Former with only 10-15% additional compute. Finally, SkipV1Former can seamlessly combine advanced methods like Group-Query Attention and Multi-Latent Attention to achieve further KV cache savings and performance improvement. When combined with YOCO, it cuts KV cache size by nearly 50 % while still improving performance.
Key facts. Cited by 4 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 55 of its 66 references that are in the map. First posted on arXiv 19 Oct 2025.
Builds on
55 of 66 references.
Closest ancestors · 8
Same direction, newest first
- 2024KVSharer: Efficient Inference via Layer-Wise Dissimilar KV Cache Sharing24 Oct
- 2024Lossless KV Cache Compression to 2%20 Oct
- 2024A Systematic Study of Cross-Layer KV Sharing for Efficient LLM Inference18 Oct
- 2024Reducing Transformer Key-Value Cache Size with Cross-Layer Attention21 May
- 2024Layer-Condensed KV Cache for Efficient Inference of Large Language Models17 May
- 2024GEAR: An Efficient KV Cache Compression Recipe for Near-Lossless Generative Inference of LLM8 Mar
- 2023H2O: Heavy-Hitter Oracle for Efficient Generative Inference of Large Language Models24 Jun
- 2023Scissorhands: Exploiting the Persistence of Importance Hypothesis for LLM KV Cache Compression at Test Time26 May
Other references · 47
- 2017Attention Is All You Needcited by 59,932
- 2020Language Models are Few-Shot Learnerscited by 32,199
- 2020An Image is Worth 16x16 Words: Transformers for Image Recognition at Scalecited by 29,915
- 2015U-Net: Convolutional Networks for Biomedical Image Segmentationcited by 19,604
- 2019Exploring the Limits of Transfer Learning with a Unified Text-to-Text Transformercited by 14,491
- 2023LLaMA: Open and Efficient Foundation Language Modelscited by 13,958
- 2016Densely Connected Convolutional Networkscited by 6,691
- 2021RoFormer: Enhanced Transformer with Rotary Position Embeddingcited by 4,581
- 2022FlashAttention: Fast and Memory-Efficient Exact Attention with IO-Awarenesscited by 3,611
- 2020The Pile: An 800GB Dataset of Diverse Text for Language Modelingcited by 2,346
- 2023Efficient Streaming Language Models with Attention Sinkscited by 2,115
- 2020Transformers are RNNs: Fast Autoregressive Transformers with Linear Attentioncited by 1,979
35 more
- 2019Generating Long Sequences with Sparse Transformerscited by 1,523
- 2023GQA: Training Generalized Multi-Query Transformer Models from Multi-Head Checkpointscited by 1,311
- 2016Understanding intermediate layers using linear classifier probescited by 1,279
- 2024DeepSeek-V2: A Strong, Economical, and Efficient Mixture-of-Experts Language Modelcited by 1,093
- 2022In-context Learning and Induction Headscited by 930
- 2022Transformers learn in-context by gradient descentcited by 714
- 2019Fast Transformer Decoding: One Write-Head is All You Needcited by 700
- 2024KVQuant: Towards 10 Million Context Length LLM Inference with KV Cache Quantizationcited by 571
- 2023FlexGen: High-Throughput Generative Inference of Large Language Models with a Single GPUcited by 567
- 2022Efficiently Scaling Transformer Inferencecited by 529
- 2024GaLore: Memory-Efficient LLM Training by Gradient Low-Rank Projectioncited by 447
- 2021Attention is Not All You Need: Pure Attention Loses Rank Doubly Exponentially with Depthcited by 411
- 2025Native Sparse Attention: Hardware-Aligned and Natively Trainable Sparse Attentioncited by 391
- 2023Trained Transformers Learn Linear Models In-Contextcited by 331
- 2023Transformers learn to implement preconditioned gradient descent for in-context learningcited by 274
- 2023A Survey on Model Compression for Large Language Modelscited by 263
- 2022TinyViT: Fast Pretraining Distillation for Small Vision Transformerscited by 232
- 2020SpAtten: Efficient Sparse Attention Architecture with Cascade Token and Head Pruningcited by 228
- 2023One Step of Gradient Descent is Provably the Optimal In-Context Learner with One Layer of Linear Self-Attentioncited by 168
- 2022Anti-Oversmoothing in Deep Vision Transformers via the Fourier Domain Analysis: From Theory to Practicecited by 136
- 2024You Only Cache Once: Decoder-Decoder Architectures for Language Modelscited by 135
- 2023Uncovering mesa-optimization algorithms in Transformerscited by 94
- 2024Hyper-Connectionscited by 81
- 2023How Do Transformers Learn In-Context Beyond Simple Functions? A Case Study on Learning with Representationscited by 72
- 2020RealFormer: Transformer Likes Residual Attentioncited by 62
- 2024DenseFormer: Enhancing Information Flow in Transformers via Depth Weighted Averagingcited by 48
- 2023Transformers Learn to Achieve Second-Order Convergence Rates for In-Context Linear Regressioncited by 47
- 2023Mitigating Over-smoothing in Transformers via Regularized Nonlocal Functionalscited by 31
- 2023ResiDual: Transformer with Dual Residual Connectionscited by 30
- 2023Skip-Attention: Improving Vision Transformers by Paying Less Attentioncited by 23
- 2021Rethinking Skip Connection with Layer Normalization in Transformers and ResNetscited by 22
- 2024MLKV: Multi-Layer Key-Value Heads for Memory Efficient Transformer Decodingcited by 20
- 2024How Transformers Utilize Multi-Head Attention in In-Context Learning? A Case Study on Sparse Linear Regressioncited by 13
- 2023A Meta-Learning Perspective on Transformers for Causal Language Modelingcited by 5
- 2024Skip-Layer Attention: Bridging Abstract and Detailed Dependencies in Transformerscited by 4
Built on by
Citing papers per year: 2026: 4.
4 papers.
- 2026IndexCache: Accelerating Sparse Attention via Cross-Layer Index Reusecited by 27
- 2026Do Value Vectors in Deep Layers Need Context from the Residual Stream?cited by 1
- 2026Attention Projection Mixing with Exogenous Anchorscited by 0
- 2026YOCO++: Enhancing YOCO with KV Residual Connections for Efficient LLM Inferencecited by 0
- 2026Do Value Vectors in Deep Layers Need Context from the Residual Stream?1 Jun 2026
- 2026YOCO++: Enhancing YOCO with KV Residual Connections for Efficient LLM Inference15 Apr 2026
- 2026IndexCache: Accelerating Sparse Attention via Cross-Layer Index Reuse12 Mar 2026
- 2026Attention Projection Mixing with Exogenous Anchors13 Jan 2026
Similar, not cited
Close in wording, no citation either way
- 2025KV-CAR: KV Cache Compression using Autoencoders and KV Reuse in Large Language Models
- 2025TaDA: Training-free recipe for Decoding with Adaptive KV Cache Compression and Mean-centering
- 2026Training Transformers for KV Cache Compressibility
- 2024FiRST: Finetuning Router-Selective Transformers for Input-Adaptive Latency Reduction
- 2025LLMCache: Layer-Wise Caching Strategies for Accelerated Reuse in Transformer Inference
- 2025Compress, Gather, and Recompute: REFORMing Long-Context Processing in Transformers
- 2024A Method for Building Large Language Models with Predefined KV Cache Capacity
- 2026Codec-Gauge: Learning Compression-Friendly Gauges for Transformer KV Caches