mlsanityData to 14 Oct 2026

FOGO: Forgetting-aware Orthogonalization Optimizer

arXiv 2606.10406 ↗First posted 9 Jun 2026cs.LG cs.AIcited by 0

Not on the map. Why?

Abstract (arXiv)

We argue that forgetting is not confined to continual learning but is a general optimization phenomenon: during standard training, dominant mini-batch gradients suppress rare but useful update directions, causing short-term forgetting at every step. When such knowledge is never revisited, these losses compound into long-term forgetting-the classical failure mode of continual learning. We introduce FOGO, a scalable optimizer that continuously detects and resolves gradient interference across both regimes. FOGO spectrally orthogonalizes momentum updates to prevent dominant directions from monopolizing optimization, then stores representative past directions in a compact codebook memory built on random projection, where pairwise distances are provably preserved in low-dimensional space. At each step, conflicts between the current update and stored directions are resolved via lightweight orthogonal correction and lifted back through a proximal step, with minimal overhead and no data storage. Across class-imbalanced classification, continual visual learning under domain and class shifts, continual fine-tuning of LLaVA-7B, and GPT-2 pretraining, FOGO consistently improves convergence and knowledge retention, outperforming Adam and Muon.

Key facts. No citing papers in the map yet (data through 14 Oct 2026). Builds on 28 of its 40 references that are in the map. First posted on arXiv 9 Jun 2026.

Builds on

28 of 40 references.

References · 28

  1. 2015Deep Residual Learning for Image Recognitioncited by 52,096
  2. 2023LLaMA: Open and Efficient Foundation Language Modelscited by 14,132
  3. 2023Visual Instruction Tuningcited by 7,869
  4. 2016Overcoming catastrophic forgetting in neural networkscited by 4,787
  5. 2021RoFormer: Enhanced Transformer with Rotary Position Embeddingcited by 4,703
  6. 2016Making the V in VQA Matter: Elevating the Role of Image Understanding in Visual Question Answeringcited by 2,516
  7. 2017Gradient Episodic Memory for Continual Learningcited by 1,898
  8. 2016Progressive Neural Networkscited by 1,361
  9. 2018Efficient Lifelong Learning with A-GEMcited by 1,058
  10. 2017Deeper, Broader and Artier Domain Generalizationcited by 986
  11. 2020Dark Experience for General Continual Learning: a Strong, Simple Baselinecited by 911
  12. 2023A Comprehensive Survey of Continual Learning: Theory, Method and Applicationcited by 872
16 more
  1. 2024MiniCPM: Unveiling the Potential of Small Language Models with Scalable Training Strategiescited by 723
  2. 2018Learning to Learn without Forgetting by Maximizing Transfer and Minimizing Interferencecited by 521
  3. 2025Muon is Scalable for LLM Trainingcited by 428
  4. 2018Shampoo: Preconditioned Stochastic Tensor Optimizationcited by 426
  5. 2019Orthogonal Gradient Descent for Continual Learningcited by 369
  6. 2021Gradient Projection Memory for Continual Learningcited by 348
  7. 2023Orthogonal Subspace Learning for Language Model Continual Learningcited by 312
  8. 2022The Vendi Score: A Diversity Evaluation Metric for Machine Learningcited by 311
  9. 2024Old Optimizer, New Norm: An Anthologycited by 193
  10. 2023A Definition of Continual Reinforcement Learningcited by 101
  11. 2025Nested Learning: The Illusion of Deep Learning Architecturescited by 72
  12. 2025AdaMuon: Adaptive Muon Optimizercited by 67
  13. 2024Addressing Loss of Plasticity and Catastrophic Forgetting in Continual Learningcited by 50
  14. 2025Prismatic Synthesis: Gradient-based Data Diversification Boosts Generalization in LLM Reasoningcited by 40
  15. 2025CL-MoE: Enhancing Multimodal Large Language Model with Dual Momentum Mixture-of-Experts for Continual Visual Question Answeringcited by 37
  16. 2026FIRE: Frobenius-Isometry Reinitialization for Balancing the Stability-Plasticity Tradeoffcited by 7

Built on by

Not cited yet.