Goto

Collaborating Authors

 msign


Supplementary Material

Neural Information Processing Systems

Then each deterministic NN in {ฯ€w,b | (w,b) Wฯ€}is safe if and only if the system of constraints ฮฆ(ฯ€,X0,Xu,) is not satisfiable. We prove the equivalent claim that there exists a weight vector (w,b) Wฯ€ for which ฯ€w,b is unsafe if and only if ฮฆ(ฯ€,X0,Xu,) is satisfiable. First, suppose that there exists a weight vector (w,b) Wฯ€ for which ฯ€w,b is unsafe and we want to show that ฮฆ(ฯ€,X0,Xu,) is satisfiable. This direction of the proof is straightforward since values of the network's neurons on the unsafe input give rise to a solution of ฮฆ(ฯ€,X0,Xu,). Indeed, by assumption there exists a vector of input neuron values x0 X0 for which the corresponding vector of output neuron values xl = ฯ€w,b(x0) is unsafe, i.e. xl Xu.


Preconditioning Benefits of Spectral Orthogonalization in Muon

arXiv.org Machine Learning

The Muon optimizer, a matrix-structured algorithm that leverages spectral orthogonalization of gradients, is a milestone in the pretraining of large language models. However, the underlying mechanisms of Muon -- particularly the role of gradient orthogonalization -- remain poorly understood, with very few works providing end-to-end analyses that rigorously explain its advantages in concrete applications. We take a step by studying the effectiveness of a simplified variant of Muon through two case studies: matrix factorization, and in-context learning of linear transformers. For both problems, we prove that simplified Muon converges linearly with iteration complexities independent of the relevant condition number, provably outperforming gradient descent and Adam. Our analysis reveals that the Muon dynamics decouple into a collection of independent scalar sequences in the spectral domain, each exhibiting similar convergence behavior. Our theory formalizes the preconditioning effect induced by spectral orthogonalization, offering insight into Muon's effectiveness in these matrix optimization problems and potentially beyond.


Multiscale Invertible Generative Networks for High-Dimensional Bayesian Inference

arXiv.org Machine Learning

We propose a Multiscale Invertible Generative Network (MsIGN) and associated training algorithm that leverages multiscale structure to solve high-dimensional Bayesian inference. To address the curse of dimensionality, MsIGN exploits the low-dimensional nature of the posterior, and generates samples from coarse to fine scale (low to high dimension) by iteratively upsampling and refining samples. MsIGN is trained in a multi-stage manner to minimize the Jeffreys divergence, which avoids mode dropping in high-dimensional cases. On two high-dimensional Bayesian inverse problems, we show superior performance of MsIGN over previous approaches in posterior approximation and multiple mode capture. On the natural image synthesis task, MsIGN achieves superior performance in bits-per-dimension over baseline models and yields great interpret-ability of its neurons in intermediate layers.