mlsanityData to 7 Oct 2026

Improving Model Representation and Reducing KV Cache via Skip Connections with First Value Heads

arXiv 2510.16807 ↗First posted 19 Oct 2025cs.LG cs.AIcited by 4

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

  1. 2024KVSharer: Efficient Inference via Layer-Wise Dissimilar KV Cache Sharing24 Oct
  2. 2024Lossless KV Cache Compression to 2%20 Oct
  3. 2024A Systematic Study of Cross-Layer KV Sharing for Efficient LLM Inference18 Oct
  4. 2024Reducing Transformer Key-Value Cache Size with Cross-Layer Attention21 May
  5. 2024Layer-Condensed KV Cache for Efficient Inference of Large Language Models17 May
  6. 2024GEAR: An Efficient KV Cache Compression Recipe for Near-Lossless Generative Inference of LLM8 Mar
  7. 2023H2O: Heavy-Hitter Oracle for Efficient Generative Inference of Large Language Models24 Jun
  8. 2023Scissorhands: Exploiting the Persistence of Importance Hypothesis for LLM KV Cache Compression at Test Time26 May

Other references · 47

  1. 2017Attention Is All You Needcited by 59,932
  2. 2020Language Models are Few-Shot Learnerscited by 32,199
  3. 2020An Image is Worth 16x16 Words: Transformers for Image Recognition at Scalecited by 29,915
  4. 2015U-Net: Convolutional Networks for Biomedical Image Segmentationcited by 19,604
  5. 2019Exploring the Limits of Transfer Learning with a Unified Text-to-Text Transformercited by 14,491
  6. 2023LLaMA: Open and Efficient Foundation Language Modelscited by 13,958
  7. 2016Densely Connected Convolutional Networkscited by 6,691
  8. 2021RoFormer: Enhanced Transformer with Rotary Position Embeddingcited by 4,581
  9. 2022FlashAttention: Fast and Memory-Efficient Exact Attention with IO-Awarenesscited by 3,611
  10. 2020The Pile: An 800GB Dataset of Diverse Text for Language Modelingcited by 2,346
  11. 2023Efficient Streaming Language Models with Attention Sinkscited by 2,115
  12. 2020Transformers are RNNs: Fast Autoregressive Transformers with Linear Attentioncited by 1,979
35 more
  1. 2019Generating Long Sequences with Sparse Transformerscited by 1,523
  2. 2023GQA: Training Generalized Multi-Query Transformer Models from Multi-Head Checkpointscited by 1,311
  3. 2016Understanding intermediate layers using linear classifier probescited by 1,279
  4. 2024DeepSeek-V2: A Strong, Economical, and Efficient Mixture-of-Experts Language Modelcited by 1,093
  5. 2022In-context Learning and Induction Headscited by 930
  6. 2022Transformers learn in-context by gradient descentcited by 714
  7. 2019Fast Transformer Decoding: One Write-Head is All You Needcited by 700
  8. 2024KVQuant: Towards 10 Million Context Length LLM Inference with KV Cache Quantizationcited by 571
  9. 2023FlexGen: High-Throughput Generative Inference of Large Language Models with a Single GPUcited by 567
  10. 2022Efficiently Scaling Transformer Inferencecited by 529
  11. 2024GaLore: Memory-Efficient LLM Training by Gradient Low-Rank Projectioncited by 447
  12. 2021Attention is Not All You Need: Pure Attention Loses Rank Doubly Exponentially with Depthcited by 411
  13. 2025Native Sparse Attention: Hardware-Aligned and Natively Trainable Sparse Attentioncited by 391
  14. 2023Trained Transformers Learn Linear Models In-Contextcited by 331
  15. 2023Transformers learn to implement preconditioned gradient descent for in-context learningcited by 274
  16. 2023A Survey on Model Compression for Large Language Modelscited by 263
  17. 2022TinyViT: Fast Pretraining Distillation for Small Vision Transformerscited by 232
  18. 2020SpAtten: Efficient Sparse Attention Architecture with Cascade Token and Head Pruningcited by 228
  19. 2023One Step of Gradient Descent is Provably the Optimal In-Context Learner with One Layer of Linear Self-Attentioncited by 168
  20. 2022Anti-Oversmoothing in Deep Vision Transformers via the Fourier Domain Analysis: From Theory to Practicecited by 136
  21. 2024You Only Cache Once: Decoder-Decoder Architectures for Language Modelscited by 135
  22. 2023Uncovering mesa-optimization algorithms in Transformerscited by 94
  23. 2024Hyper-Connectionscited by 81
  24. 2023How Do Transformers Learn In-Context Beyond Simple Functions? A Case Study on Learning with Representationscited by 72
  25. 2020RealFormer: Transformer Likes Residual Attentioncited by 62
  26. 2024DenseFormer: Enhancing Information Flow in Transformers via Depth Weighted Averagingcited by 48
  27. 2023Transformers Learn to Achieve Second-Order Convergence Rates for In-Context Linear Regressioncited by 47
  28. 2023Mitigating Over-smoothing in Transformers via Regularized Nonlocal Functionalscited by 31
  29. 2023ResiDual: Transformer with Dual Residual Connectionscited by 30
  30. 2023Skip-Attention: Improving Vision Transformers by Paying Less Attentioncited by 23
  31. 2021Rethinking Skip Connection with Layer Normalization in Transformers and ResNetscited by 22
  32. 2024MLKV: Multi-Layer Key-Value Heads for Memory Efficient Transformer Decodingcited by 20
  33. 2024How Transformers Utilize Multi-Head Attention in In-Context Learning? A Case Study on Sparse Linear Regressioncited by 13
  34. 2023A Meta-Learning Perspective on Transformers for Causal Language Modelingcited by 5
  35. 2024Skip-Layer Attention: Bridging Abstract and Detailed Dependencies in Transformerscited by 4

Built on by

Cited by year (2026 to 7 Oct)

Citing papers per year: 2026: 4.

4 papers.

Similar, not cited

Close in wording, no citation either way

  1. 2025KV-CAR: KV Cache Compression using Autoencoders and KV Reuse in Large Language Models
  2. 2025TaDA: Training-free recipe for Decoding with Adaptive KV Cache Compression and Mean-centering
  3. 2026Training Transformers for KV Cache Compressibility
  4. 2024FiRST: Finetuning Router-Selective Transformers for Input-Adaptive Latency Reduction
  5. 2025LLMCache: Layer-Wise Caching Strategies for Accelerated Reuse in Transformer Inference
  6. 2025Compress, Gather, and Recompute: REFORMing Long-Context Processing in Transformers
  7. 2024A Method for Building Large Language Models with Predefined KV Cache Capacity
  8. 2026Codec-Gauge: Learning Compression-Friendly Gauges for Transformer KV Caches