Second-Order Multi-Level Variance Correction for Modality Competition in Multimodal Models
Not on the map. Why?
Abstract (arXiv)
Autoregressive next-token training offers a unified formulation for image generation and text understanding, but it also creates strong modality competition that destabilizes optimization and limits large-batch scaling. We show that first-order optimizers such as AdamW are vulnerable to cross-modality gradient heterogeneity, while second-order preconditioning, particularly SOAP, provides a more stable basis for multimodal alignment. Building on this insight, we propose ML-FOP-SOAP, a second-order optimization framework with Multi-Level Variance Correction. Our Fisher-Orthogonal Projection suppresses variance-induced modality conflicts, reducing the trade-off between visual generation and textual understanding. To make this practical under large gradient accumulation, we introduce a hierarchical folding strategy that captures fine-grained variance with low micro-step overhead. Experiments on Janus and Emu3 show consistent gains across both modalities and stable training at batch size 8192. Compared with AdamW, our method improves sample efficiency by up to 1.4× and accelerates wall-clock training by up to 1.5×, offering a robust optimizer for scaling multimodal foundation models.
Key facts. Cited by 2 papers in the map (data through 14 Oct 2026). Builds on 22 of its 26 references that are in the map. First posted on arXiv 15 May 2026.
Builds on
22 of 26 references.
References · 22
- 2023Visual Instruction Tuningcited by 7,869
- 2017Neural Discrete Representation Learningcited by 4,712
- 2023Sigmoid Loss for Language Image Pre-Trainingcited by 3,109
- 2022Training Compute-Optimal Large Language Modelscited by 3,019
- 2023InstructBLIP: Towards General-purpose Vision-Language Models with Instruction Tuningcited by 2,876
- 2023Qwen-VL: A Versatile Vision-Language Model for Understanding, Localization, Text Reading, and Beyondcited by 2,224
- 2020Gradient Surgery for Multi-Task Learningcited by 1,256
- 2024Chameleon: Mixed-Modal Early-Fusion Foundation Modelscited by 927
- 2017GradNorm: Gradient Normalization for Adaptive Loss Balancing in Deep Multitask Networkscited by 834
- 2015Optimizing Neural Networks with Kronecker-factored Approximate Curvaturecited by 796
- 2022Mind the Gap: Understanding the Modality Gap in Multi-modal Contrastive Representation Learningcited by 561
- 2024Janus: Decoupling Visual Encoding for Unified Multimodal Understanding and Generationcited by 457
10 more
- 2022Balanced Multimodal Learning via On-the-fly Gradient Modulationcited by 264
- 2017The Marginal Value of Adaptive Gradient Methods in Machine Learningcited by 250
- 2023LLaVA-Plus: Learning to Use Tools for Creating Multimodal Agentscited by 193
- 2024SOAP: Improving and Stabilizing Shampoo using Adamcited by 192
- 2013Revisiting Natural Gradient for Deep Networkscited by 159
- 2023A Distributed Data-Parallel PyTorch Implementation of the Distributed Shampoo Optimizer for Training Neural Networks At-Scalecited by 77
- 2024Improving Multimodal Learning with Multi-Loss Gradient Modulationcited by 18
- 2025The Potential of Second-Order Optimization for LLMs: A Study with Full Gauss-Newtoncited by 17
- 2025DynCIM: Dynamic Curriculum for Imbalanced Multimodal Learningcited by 16
- 2025Beyond the Mean: Fisher-Orthogonal Projection for Natural Gradient Descent in Large Batch Trainingcited by 5
Built on by
Citing papers per year: 2026: 2.
2 papers.