Optimizer-Induced Mode Connectivity: From AdamW to Muon
Not on the map. Why?
Abstract (arXiv)
Mode connectivity has been widely studied, yet the role of the optimizer remains underexplored. We revisit it through optimizer-induced implicit regularization, asking how connectivity behaves when restricted to solutions constrained by a given optimizer. For two-layer ReLU networks, we show that solutions from a single optimizer -- AdamW, Muon, or others in the Lion-K family -- form a connected set at sufficiently large width, a result not implied by prior work. We then characterize how optimizer-induced regions interact: at large width two different regions can be disjoint or overlap depending on regularization, while in our small-width example AdamW and Muon converge to disconnected zero-loss components separated by a provable loss barrier. Empirically, in GPT-2 pretraining, we observe same-optimizer paths preserve each model's spectrum while cross-optimizer paths traverse a smooth transition. Our results reveal optimizer-dependent structure beyond classical mode connectivity literature.
Key facts. No citing papers in the map yet (data through 14 Oct 2026). Builds on 38 of its 50 references that are in the map. First posted on arXiv 11 May 2026.
Builds on
38 of 50 references.
References · 38
- 2016Pointer Sentinel Mixture Modelscited by 3,096
- 2015Aligning Books and Movies: Towards Story-like Visual Explanations by Watching Movies and Reading Bookscited by 1,043
- 2019Linear Mode Connectivity and the Lottery Ticket Hypothesiscited by 637
- 2018Loss Surfaces, Mode Connectivity, and Fast Ensembling of DNNscited by 588
- 2018signSGD: Compressed Optimisation for Non-Convex Problemscited by 571
- 2018Essentially No Barriers in Neural Network Energy Landscapecited by 330
- 2013One Billion Word Benchmark for Measuring Progress in Statistical Language Modelingcited by 308
- 2021The Role of Permutation Invariance in Linear Mode Connectivity of Neural Networkscited by 291
- 2017The Marginal Value of Adaptive Gradient Methods in Machine Learningcited by 250
- 2018A Simple Method for Commonsense Reasoningcited by 223
- 2024SOAP: Improving and Stabilizing Shampoo using Adamcited by 192
- 2023ZipIt! Merging Models from Different Tasks without Trainingcited by 179
26 more
- 2020Momentum Improves Normalized SGDcited by 139
- 2022REPAIR: REnormalizing Permuted Activations for Interpolation Repaircited by 120
- 2016Topology and Geometry of Half-Rectified Network Optimizationcited by 102
- 2025Muon Optimizes Under Spectral Norm Constraintscited by 86
- 2019Explaining Landscape Connectivity of Low-cost Solutions for Multilayer Netscited by 63
- 2022Re-basin via implicit Sinkhorn differentiationcited by 62
- 2024Implicit Bias of AdamW: ℓ_∞ Norm Constrained Optimizationcited by 58
- 2023Lion Secretly Solves Constrained Optimization: As Lyapunov Predictscited by 51
- 2020The Implicit Bias for Adaptive Optimization Algorithms on Homogeneous Neural Networkscited by 46
- 2025Muon Outperforms Adam in Tail-End Associative Memory Learningcited by 46
- 2025Implicit Bias of Spectral Descent and Muon on Multiclass Separable Datacited by 41
- 2023Transformer Fusion with Optimal Transportcited by 36
- 2023On the Implicit Bias of Adamcited by 27
- 2023Proving Linear Mode Connectivity of Neural Networks via Optimal Transportcited by 25
- 2023Improving Convergence and Generalization Using Parameter Symmetriescited by 23
- 2024C2M3: Cycle-Consistent Multi-Model Mergingcited by 20
- 2024Merging Text Transformer Models from Different Initializationscited by 16
- 2025Generalized Linear Mode Connectivity for Transformerscited by 16
- 2024Exploring the loss landscape of regularized neural networks via convex dualitycited by 12
- 2021When Are Solutions Connected in Deep Networks?cited by 11
- 2023Optimal Sets and Solution Paths of ReLU Networkscited by 10
- 2025Understanding Mode Connectivity via Parameter Space Symmetrycited by 10
- 2021A Note on Connectivity of Sublevel Sets in Deep Learningcited by 8
- 2025On Linear Mode Connectivity of Mixture-of-Experts Architecturescited by 8
- 2025Analyzing the Role of Permutation Invariance in Linear Mode Connectivitycited by 2
- 2025Do We Really Need Permutations? Impact of Model Width on Linear Mode Connectivitycited by 2
Built on by
Not cited yet.