Muon-OGD: Muon-based Spectral Orthogonal Gradient Projection for LLM Continual Learning
Not on the map. Why?
Abstract (arXiv)
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.
Key facts. Cited by 5 papers in the map (data through 14 Oct 2026). Builds on 31 of its 40 references that are in the map. First posted on arXiv 9 May 2026.
Builds on
31 of 40 references.
References · 31
- 2021Training Verifiers to Solve Math Word Problemscited by 10,109
- 2018GLUE: A Multi-Task Benchmark and Analysis Platform for Natural Language Understandingcited by 4,899
- 2016Overcoming catastrophic forgetting in neural networkscited by 4,787
- 2016Learning without Forgettingcited by 2,496
- 2015Character-level Convolutional Networks for Text Classificationcited by 2,330
- 2017Gradient Episodic Memory for Continual Learningcited by 1,898
- 2019SuperGLUE: A Stickier Benchmark for General-Purpose Language Understanding Systemscited by 1,729
- 2018Continual Lifelong Learning with Neural Networks: A Reviewcited by 1,330
- 2017Continual Learning with Deep Generative Replaycited by 1,158
- 2017Memory Aware Synapses: Learning what (not) to forgetcited by 1,105
- 2018Efficient Lifelong Learning with A-GEMcited by 1,058
- 2021Learning to Prompt for Continual Learningcited by 944
19 more
- 2023A Comprehensive Survey of Continual Learning: Theory, Method and Applicationcited by 872
- 2013An Empirical Investigation of Catastrophic Forgetting in Gradient-Based Neural Networkscited by 722
- 2017Lifelong Learning with Dynamically Expandable Networkscited by 694
- 2022DualPrompt: Complementary Prompting for Rehearsal-free Continual Learningcited by 642
- 2024BigCodeBench: Benchmarking Code Generation with Diverse Function Calls and Complex Instructionscited by 546
- 2025Muon is Scalable for LLM Trainingcited by 428
- 2019Orthogonal Gradient Descent for Continual Learningcited by 369
- 2023Orthogonal Subspace Learning for Language Model Continual Learningcited by 312
- 2023Progressive Prompts: Continual Learning for Language Modelscited by 197
- 2021LFPT5: A Unified Framework for Lifelong Few-shot Language Learning Based on Prompt Tuning of T5cited by 104
- 2025Muon Optimizes Under Spectral Norm Constraintscited by 86
- 2024Modular Duality in Deep Learningcited by 77
- 2023TRACE: A Comprehensive Benchmark for Continual Learning in Large Language Modelscited by 66
- 2025Isotropic Curvature Model for Understanding Deep Learning Optimization: Is Gradient Orthogonalization Optimal?cited by 37
- 2025When do spectral gradient updates help in deep learning?cited by 31
- 2025OPLoRA: Orthogonal Projection LoRA Prevents Catastrophic Forgetting during Parameter-Efficient Fine-Tuningcited by 21
- 2025Sculpting Subspaces: Constrained Full Fine-Tuning in LLMs for Continual Learningcited by 15
- 2025Dynamic Orthogonal Continual Fine-tuning for Mitigating Catastrophic Forgettingscited by 2
- 2026GEM-Style Constraints for PEFT with Dual Gradient Projection in LoRAcited by 1
Built on by
Citing papers per year: 2026: 5.
5 papers.
- 2026Memory for Large Language Modelscited by 4
- 2026fPINN-DeepONet: A Physics-Informed Operator Learning Framework for Multi-term Time-fractional Mixed Diffusion-wave Equationscited by 0
- 2026Omega-S: A Functional Resilience Index for LLM Fine-Tuningcited by 0
- 2026CONTRAMEM: Learning Self-Evolving Procedural Memory from Contrasting Multi-Model Trajectoriescited by 0
- 2026When Muon Meets Task Interference: A Spectral Perspective on Continual Learning and Model Mergingcited by 0
- 2026When Muon Meets Task Interference: A Spectral Perspective on Continual Learning and Model Merging27 Aug 2026
- 2026CONTRAMEM: Learning Self-Evolving Procedural Memory from Contrasting Multi-Model Trajectories23 Aug 2026
- 2026Omega-S: A Functional Resilience Index for LLM Fine-Tuning4 Aug 2026
- 2026Memory for Large Language Models28 Jul 2026
- 2026fPINN-DeepONet: A Physics-Informed Operator Learning Framework for Multi-term Time-fractional Mixed Diffusion-wave Equations15 May 2026