Muon controls weight updates through the RMS movement of a linear layer's output. This implicitly treats the linear output as a proxy for the full module, including the nonlinearity that follows. This proxy can be misleading for softmax, where the RMS shift in activations does not faithfully measure the change in the o...
Aditya Somasundaram, Charles Guille-Escuret, Alexander Moreno et al.· 0 citations
Muon motivates designing optimizer geometry around the function of each parameter block and uses the spectral norm for hidden linear layers. For the language-model head, the spectral norm is not a faithful measure of functional change. Softmax removes shared logit shifts, whereas the spectral norm can assign arbitraril...
Aditya Somasundaram, Charles Guille-Escuret, Alexander Moreno et al.· 0 citations
Motivated by Muon’s singular-value conditioning, this work proposes maximizing the smallest row separation while constraining this diameter, producing an approximate-equidistance problem when V ≫ d .
Aditya Somasundaram· 0 citations
We use cookies to run the site and, with your consent, for analytics and to show ads.
See our Cookie Policy.