理论解释为何 LoRA 低秩就够:瓶颈秩决定最优解秩
How Finite-Rank Bottlenecks Shape the Low-Rank Adaptation Landscape
搞微调的都该看看这篇:它不用 NTK 线性化,用瓶颈秩从理论上说清了为什么 LoRA 低秩就够用。
arXiv 论文研究 LoRA 为何低秩更新就足够,突破了以往依赖 Neural Tangent Kernel(NTK)线性化假设的分析方式。作者发现骨干冻结后,损失函数只能通过冻结激活诱导的有限秩线性映射观察到适配权重矩阵,该映射的秩被称为 bottleneck rank(瓶颈秩)。trace-norm 正则化会使全局最优解的秩由瓶颈秩控制,而非样本量。对 multi-head attention,利用 softmax 对行加常数不敏感等不变性,作者得到了更紧的秩上界。论文还证明带 weight decay 的因式分解 LoRA 目标在任意小的 PSD 扰动后,当 LoRA 秩超过瓶颈秩阈值时一阶驻点均为秩亏的。
How Finite-Rank Bottlenecks Shape the Low-Rank Adaptation Landscape
Low-rank adaptation (LoRA) has become the standard method for parameter-efficient fine-tuning of large pretrained models, yet theoretical explanations for why low-rank updates suffice remain incomplete. Most existing analyses rely on the Neural Tangent Kernel (NTK) regime, which linearizes the network around the pretrained weights. In this work, we take a different approach that requires no linearization: our key observation is that, once the backbone is frozen, the loss sees each adapted weight matrix only through a finite-rank linear map induced by the frozen activations. Trace-norm regularization then forces the optimizer to pick a low-rank matrix among all matrices that produce the same observed output. This yields global minimizers whose ranks are controlled by the rank of this map, which we call the \emph{bottleneck rank}, rather than the sample size. For multi-head attention, we obtain tighter rank upper bounds by exploiting two additional invariances: row-wise softmax is insensitive to additive row constants in the score matrix, and value updates are observable only after attention weighting and frozen output projection. For the factorized LoRA objective with weight decay, we prove a complementary result on the nonconvex side: after an arbitrarily small generic positive semidefinite perturbation, every first-order stationary point is rank-deficient once the LoRA rank exceeds a threshold determined by the bottleneck rank. Finally, we show that global minimizers of the perturbed objective are near-optimal for the original trace-norm problem, and we provide uniform generalization bounds governed by the spectral structure of the frozen features.