mlsanityData to 14 Oct 2026

Muon-OGD: Muon-based Spectral Orthogonal Gradient Projection for LLM Continual Learning

arXiv 2605.08949 ↗First posted 9 May 2026cs.LGcited by 5

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

  1. 2021Training Verifiers to Solve Math Word Problemscited by 10,109
  2. 2018GLUE: A Multi-Task Benchmark and Analysis Platform for Natural Language Understandingcited by 4,899
  3. 2016Overcoming catastrophic forgetting in neural networkscited by 4,787
  4. 2016Learning without Forgettingcited by 2,496
  5. 2015Character-level Convolutional Networks for Text Classificationcited by 2,330
  6. 2017Gradient Episodic Memory for Continual Learningcited by 1,898
  7. 2019SuperGLUE: A Stickier Benchmark for General-Purpose Language Understanding Systemscited by 1,729
  8. 2018Continual Lifelong Learning with Neural Networks: A Reviewcited by 1,330
  9. 2017Continual Learning with Deep Generative Replaycited by 1,158
  10. 2017Memory Aware Synapses: Learning what (not) to forgetcited by 1,105
  11. 2018Efficient Lifelong Learning with A-GEMcited by 1,058
  12. 2021Learning to Prompt for Continual Learningcited by 944
19 more
  1. 2023A Comprehensive Survey of Continual Learning: Theory, Method and Applicationcited by 872
  2. 2013An Empirical Investigation of Catastrophic Forgetting in Gradient-Based Neural Networkscited by 722
  3. 2017Lifelong Learning with Dynamically Expandable Networkscited by 694
  4. 2022DualPrompt: Complementary Prompting for Rehearsal-free Continual Learningcited by 642
  5. 2024BigCodeBench: Benchmarking Code Generation with Diverse Function Calls and Complex Instructionscited by 546
  6. 2025Muon is Scalable for LLM Trainingcited by 428
  7. 2019Orthogonal Gradient Descent for Continual Learningcited by 369
  8. 2023Orthogonal Subspace Learning for Language Model Continual Learningcited by 312
  9. 2023Progressive Prompts: Continual Learning for Language Modelscited by 197
  10. 2021LFPT5: A Unified Framework for Lifelong Few-shot Language Learning Based on Prompt Tuning of T5cited by 104
  11. 2025Muon Optimizes Under Spectral Norm Constraintscited by 86
  12. 2024Modular Duality in Deep Learningcited by 77
  13. 2023TRACE: A Comprehensive Benchmark for Continual Learning in Large Language Modelscited by 66
  14. 2025Isotropic Curvature Model for Understanding Deep Learning Optimization: Is Gradient Orthogonalization Optimal?cited by 37
  15. 2025When do spectral gradient updates help in deep learning?cited by 31
  16. 2025OPLoRA: Orthogonal Projection LoRA Prevents Catastrophic Forgetting during Parameter-Efficient Fine-Tuningcited by 21
  17. 2025Sculpting Subspaces: Constrained Full Fine-Tuning in LLMs for Continual Learningcited by 15
  18. 2025Dynamic Orthogonal Continual Fine-tuning for Mitigating Catastrophic Forgettingscited by 2
  19. 2026GEM-Style Constraints for PEFT with Dual Gradient Projection in LoRAcited by 1

Built on by

Cited by year (2026 to 14 Oct)

Citing papers per year: 2026: 5.

5 papers.