用谱范数优化防止大模型学新任务时遗忘旧知识
Muon-OGD: Muon-based Spectral Orthogonal Gradient Projection for LLM Continual Learning

- 基于谱范数设计正交投影更新,避免旧任务方向干扰
- 在多个任务序列上显著降低遗忘率,性能优于传统方法
- 适合需要持续学习的大型语言模型应用,如医疗、编程场景
大语言模型在持续学习中面临灾难性遗忘问题,即学习新任务时会严重损害对旧任务的性能。现有基于投影的方法通过将参数更新限制在与历史任务方向正交的子空间来缓解干扰,但通常基于欧几里得参数几何,以Frobenius范数控制更新幅度。近期成功的Muon优化器采用正交化矩阵更新并具有谱范数解释,提示Frobenius几何未必是矩阵型LLM参数的最佳选择。受此启发,我们提出Muon-OGD,一种融合Muon式算子范数几何与正交投影约束的谱范数感知持续学习框架。该方法将每次更新建模为带线性非干扰约束的谱范数受限优化问题,并通过对偶迭代和Newton--Schulz矩阵符号近似高效求解。通过避免与先前任务相关的受保护方向,Muon-OGD旨在改善序列适应中的稳定性-可塑性权衡。我们在标准持续学习基准TRACE以及编码-数学-医学等特定领域课程上,使用编码器-解码器和仅解码器架构进行了评估。实验证明,Muon-OGD在多个任务序列上始终优于顺序微调和竞争性正交梯度基线,且计算开销可控。结果表明,谱范数感知的更新几何为大模型持续学习提供了一种实用而有效的替代方案。
原文摘要 · Abstract (English)
A central challenge in continual learning for large language models (LLMs) is catastrophic forgetting, where adapting to new tasks can substantially degrade performance on previously learned ones. Existing projection-based methods mitigate such interference by restricting parameter updates to subspaces that are orthogonal to directions associated with past tasks. However, these methods are typically formulated under Euclidean parameter geometry, with update magnitudes and projections governed by the Frobenius norm. The recent empirical success of the Muon optimizer, which applies orthogonalized matrix updates and admits a spectral-norm interpretation, suggests that Frobenius geometry may not be the most effective choice for matrix-valued LLM parameters. Motivated by this observation, we propose Muon-OGD, a spectral-norm-aware continual learning framework that integrates Muon-style operator-norm geometry with orthogonal projection constraints. Our method formulates each update as a spectral-norm-constrained optimization problem with linear non-interference constraints, and solves it efficiently through dual iterations and Newton--Schulz matrix-sign approximations. By applying orthogonalized momentum updates that avoid protected directions associated with prior tasks, Muon-OGD aims to improve the stability--plasticity trade-off in sequential LLM adaptation. We evaluate the proposed method on standard continual learning benchmarks, TRACE, and domain-specific Coding--Math--Medical curricula using both encoder--decoder and decoder-only architectures. Empirically, Muon-OGD consistently improves over sequential fine-tuning and competitive orthogonal-gradient baselines, while remaining computationally scalable. These results suggest that spectral-norm-aware update geometry provides a practical and effective alternative to Frobenius-norm projection for continual learning in LLMs.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。