\k{appa}-LoRA: Condition Numbers Reveal Which LoRA Matrices Worth Updating

Generative AI & LLMs
Published: arXiv: 2607.22489v1
Authors

Jianghui Wang Silong Yong Francesco Orabona Marco Canini Katia P. Sycara Yaqi Xie

Abstract

Low-Rank Adaptation (LoRA) has become a widely adopted technique for efficient neural network fine-tuning, decomposing model updates into low-rank matrices. However, LoRA remains computationally costly because it updates all matrices uniformly, regardless of their actual contribution to adaptation. This cost is especially prohibitive for large-scale models with billions of parameters and for resource-constrained settings such as edge deployment and on-device fine-tuning. We show for the first time that not all LoRA matrices are equally worth tuning: matrices with smaller condition numbers (the ratio of largest to smallest singular value) are already well-balanced across directions and contribute only marginally to adaptation, whereas matrices with larger condition numbers contain underdeveloped directions that span richer subspaces and drive most of the performance gains. This observation itself is a key contribution of our work, and it motivates a more selective approach to fine-tuning. Building on this insight, we propose \k{appa}-LoRA, a method that optimizes LoRA by focusing updates on the matrices with the largest condition numbers, which capture the most informative directions of change. By restricting LoRA updates to the top 50% of weight matrices ranked by condition number, \k{appa}-LoRA halves the trainable parameter count and correspondingly reduces compute and memory cost. Extensive experiments across multiple benchmarks show that this design cuts fine-tuning time by 16.2% on average while matching the accuracy of standard LoRA and reducing memory cost by 4.5%. Further analysis reveals that the condition numbers of the selected matrices consistently decrease over training, suggesting that \k{appa}-LoRA's effectiveness stems from targeted spectral rebalancing rather than parameter selection alone.

Paper Summary

Problem
Fine-tuning large pre-trained models has become a crucial step in achieving state-of-the-art performance in various domains. However, as model sizes grow, the computational resources required for fine-tuning increase rapidly, creating significant challenges in both training time and memory usage. Traditional fine-tuning approaches update all parameters of the model, incurring high computational and memory costs.
Key Innovation
The researchers propose a new method called κ-LoRA, which optimizes Low-Rank Adaptation (LoRA) by focusing updates on the matrices with the largest condition numbers. Condition numbers measure the ratio of the largest to smallest singular value of a matrix, indicating how well-balanced the matrix is across directions. The researchers show that matrices with smaller condition numbers contribute marginally to adaptation, while matrices with larger condition numbers contain underdeveloped directions that drive most of the performance gains.
Practical Impact
κ-LoRA can be applied in various real-world scenarios, such as: * Reducing fine-tuning time: By restricting LoRA updates to the top 50% of weight matrices ranked by condition number, κ-LoRA halves the trainable parameter count and correspondingly reduces compute and memory cost. * Improving model performance: κ-LoRA's targeted spectral rebalancing leads to better model performance, matching the accuracy of standard LoRA while reducing memory cost by 4.5%. * Enabling edge deployment and on-device fine-tuning: κ-LoRA's reduced computational cost and memory usage make it suitable for resource-constrained settings.
Analogy / Intuitive Explanation
Think of a piano with multiple strings, each representing a matrix in the model. Some strings are tightly wound, while others are loose. The condition number measures how tightly wound each string is. In κ-LoRA, the researchers focus on tightening the loose strings (high-κ matrices) to improve the model's performance, rather than wasting effort on tightening the tightly wound strings (low-κ matrices). This targeted approach leads to better model performance and reduced computational cost.
Paper Information
Categories:
cs.LG cs.AI
Published Date:

arXiv ID:

2607.22489v1

Quick Actions