Freeze the Decoder, Heal the Encoder: Parameter-Efficient Adaptation for SVD-Based KV-Cache Compression
cs.LG, cs.AI
Submitted: 2026-09-25
Updated: 2026-09-25
Comments: Preprint, Under Review
License: http://creativecommons.org/licenses/by/4.0/
The gist: Comparing parameter-efficient fine-tuning recipes under a single, shared learning rate is a common but flawed practice: when the arms being compared have very different trainable-parameter counts, a
Terminology
Abstract
Comparing parameter-efficient fine-tuning recipes under a single, shared learning rate is a common but flawed practice: when the arms being compared have very different trainable-parameter counts, a shared rate can simultaneously depress the larger arms' means and inflate their variance, manufacturing a large, seemingly multi-seed-significant advantage for the smallest arm that is not a real effect. We document this confound in a concrete setting: post-hoc SVD-based KV-cache compression, where an already-pretrained model is converted to a low-rank (multi-head-latent-attention-style) cache by factorizing its key/value weights into a down-projection ("encoder") and an up-projection ("decoder"), after which a short fine-tune ("healing") recovers the accuracy lost to truncation. Under a shared learning rate, freezing the decoder and healing only the encoder looks like a clear win over healing the decoder or both factors; once every arm is given its own tuned learning rate, that apparent advantage disappears, and encoder-only healing instead reaches parity with the alternatives, at a real, measured saving of 3x fewer trainable parameters and 3x less optimizer-state memory. We verify this parity with per-arm learning-rate tuning and three seeds per configuration on a vision-language model (Qwen2.5-VL-3B-Instruct), at the one compression ratio this protocol covers, and replicate it on a text-only testbed across two backbones. Encoder-only healing is therefore a lower-memory drop-in recipe for retrofitting low-rank KV-cache compression at training time, and the shared-learning-rate pitfall we document and correct is a cautionary result for comparing any fine-tuning recipes whose arms differ in trainable-parameter count.
Sources
- A3 : an Analytical Low-Rank Approximation Framework for Attention
- Zero Sum SVD: Balancing Loss Sensitivity for Low Rank LLM Compression
- GQA: Training Generalized Multi-Query Transformer Models from Multi-Head Checkpoints
- Grow, Don't Overwrite: Fine-tuning Without Forgetting
- Don't be so Stief! Learning KV Cache low-rank approximation over the Stiefel manifold
- Palu: Compressing KV-Cache with Low-Rank Projection
- xKV: Cross-Layer KV-Cache Compression via Aligned Singular Vector Extraction
- KV-CoRE: Benchmarking Data-Dependent Low-Rank Compressibility of KV-Caches in LLMs
- DHA: Learning Decoupled-Head Attention from Transformer Checkpoints via Adaptive Heads Fusion
- Optimised Grouped-Query Attention Mechanism for Transformers
- DeepSeek-V2: A Strong, Economical, and Efficient Mixture-of-Experts Language Model
- Towards Economical Inference: Enabling DeepSeek's Multi-Head Latent Attention in Any Transformer-based LLMs
- Align Attention Heads Before Merging Them: An Effective Way for Converting MHA to GQA
- Beyond Uniform Query Distribution: Key-Driven Grouped Query Attention
- KQ-SVD: Compressing the KV Cache with Provable Guarantees on Attention Fidelity
- Multi-Head Low-Rank Attention
- GQLA: Group-Query Latent Attention for Hardware-Adaptive Decoding
- TransMLA: Multi-Head Latent Attention Is All You Need
- MosaicKV: Serving Long-Context LLM with Dynamic Two-D KV Cache Compression
- Eigen Attention: Attention in Low-Rank Space for KV Cache Compression
Related papers
- Polynomial-Augmented Neural Networks (PANNs) with Weak Orthogonality Constraints for Enhanced Function and PDE Approximation
- AIRL-S: Unifying Reinforcement Learning and Search-Based Test-Time Scaling via Adversarial Inverse Reinforcement Learning
- Transformers as Bayesian In-Context Experimenters: Smoothness-Adaptive Efficient ATE Estimation
- Convergence issues in Relational Concept Analysis based on AOC-posets
- Beliefs Beyond Posteriors: The Predictive Cost of ELBO Binding
- Understanding Diffusion Models via Ratio-Based Function Approximation with SignReLU Networks