English

Optimizer-Induced Mode Connectivity: From AdamW to Muon

Artificial Intelligence 2026-05-12 v1 Machine Learning Optimization and Control

Abstract

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\mathcal{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.

Cite

@article{arxiv.2605.09991,
  title  = {Optimizer-Induced Mode Connectivity: From AdamW to Muon},
  author = {Fangzhao Zhang and Sungyoon Kim and Erica Zhang and Yiqi Jiang and Mert Pilanci},
  journal= {arXiv preprint arXiv:2605.09991},
  year   = {2026}
}
R2 v1 2026-07-22T07:03:13.652Z