论文

Subspace Tuning:限制更新方向以缓解微调中的灾难性遗忘

Towards Continual Learning: Mitigating Forgetting With Subspace Tuning

精选理由

微调总把模型原有能力练没了?这篇提出只在低奇异值子空间更新权重,KL 漂移少 2-4 倍,做微调的可以看看。

arXiv 论文提出 Subspace Tuning(SST),将每次权重更新限制在奇异值最低的子空间内,避免改动主要奇异方向。SFT 类方法更新集中在主方向上,容易造成灾难性遗忘,而 SST 与优化器无关,可与 SFT、自蒸馏等微调方案组合。实验显示保护 99% 的权重子空间能大幅提升旧能力保留率,KL 漂移比基线方案低 2-4 倍,且模型和数据规模越大,遗忘越接近消除。

原文 · arXiv cs.LG

Towards Continual Learning: Mitigating Forgetting With Subspace Tuning

Large language models (LLMs) acquire new skills through fine-tuning, but it can destroy capabilities they already have: catastrophic forgetting remains a central obstacle to continual learning in LLMs. Recent works connect forgetting to how far, and along which directions, fine-tuning moves the weights. On-policy methods such as reinforcement learning and on-policy distillation avoid updating the principal singular directions of weight matrices and thus forget less. In contrast, SFT concentrates updates along these directions and is susceptible to catastrophic forgetting. In this work, we introduce Subspace Tuning (SST), which restricts each weight update to the bottom singular subspace, eliminating changes along the principal directions. SST is optimizer-agnostic and composes well with fine-tuning recipes including SFT and self-distillation. SST combines tight subspace protection, dense gradient estimation, and a simple refresh mechanism, enabling stronger learning than prior orthogonal fine-tuning methods. With these mechanisms, protecting as much as 99\% of the weight subspace substantially improves retention while preserving strong target-task performance. This behavior strengthens with model and data size and nearly eliminates forgetting in the large-model regime. SST also keeps the model close to its base distribution, inducing $2$--$4\times$ less KL drift than the baseline recipes. Overall, SST advances the learning--forgetting Pareto frontier, making it a practical default for fine-tuning pre-trained language models.