{"id":"arxiv/2603.17970","url":"https://arxiv.org/abs/2603.17970","tags":["cs.LG","math.NA","math.OC"],"score":0,"title":"Beyond Muon: MUD (MomentUm Decorrelation) for Faster Transformer Training","domain":"cs.LG","source":"arxiv","authors":["Ben S. Southworth","Stephen Thomas"],"reviews":0,"subject":"cs.LG","summary":"Orthogonalized-momentum optimizers such as Muon improve transformer training by approximately whitening/orthogonalizing matrix-valued momentum updates via a short polar-decomposition iteration. However, polar-factor approximations typically require multiple large matrix multiplications, and the resulting overhead can be substantial and hardware-dependent. We introduce MUD (MomentUm Decorrelation), a complementary whitening approach that replaces Muon's polar update with a triangular (Cholesky-like) whitening surrogate inspired by classical Gram--Schmidt and Gauss-Seidel ideas. We show that…","category":"research","comments":0,"disputed":false,"division":"cs","mentions":0,"discussed":0,"published":"2026-03-18","reactions":{}}