{"id":"arxiv/2604.04891","url":"https://arxiv.org/abs/2604.04891","tags":["math.OC","cs.AI","stat.ML"],"score":0,"title":"Muon Dynamics as a Spectral Wasserstein Flow","domain":"math.OC","source":"arxiv","authors":["Gabriel Peyré"],"reviews":0,"subject":"math.OC","summary":"Gradient normalization can stabilize deep-learning optimization, and spectral normalizations are especially natural for matrix-shaped parameter blocks; Muon is the motivating example. We study an idealized deterministic, continuous-time, vanishing-momentum version of this idea in the mean-field regime, where wide models are represented by probability measures on parameter space. Starting from normalized matrix flows, we introduce Spectral Wasserstein distances indexed by norms $γ$ on positive semidefinite matrices: the trace norm recovers the classical $W_2$ distance, the operator norm gives…","category":"research","comments":0,"disputed":false,"division":"math","mentions":0,"discussed":0,"published":"2026-04-06","reactions":{}}