Signal-Plus-Noise Decomposition of Nonlinear Spiked Random Matrix Models
Behrad Moniri, Hamed Hassani
TL;DR
This work extends spiked random matrix theory to nonlinear, element-wise transformations by analyzing $\mathbf{Y}=\frac{1}{\sqrt{n}}f(\mathbf{W}+\lambda\sqrt{n}\boldsymbol{x}\boldsymbol{x}^\top)$ with general, non-Gaussian noise $\mathbf{W}$. It proves a signal-plus-noise decomposition: $\mathbf{Y}=\frac{1}{\sqrt{n}}f(\mathbf{W})+\sum_{k=1}^{\ell}\mathbf{H}_k+o_{\mathbb{P}}(1)$, where $\mathbf{H}_k$ are rank-$\leq$ rank$(\mathbb{E}f^{(k)}(\mathbf{W}))$ components, and shows phase transitions as the signal exponent $\alpha$ grows, yielding up to $\ell$ outliers. The decomposition enables precise analysis of two problems: signed signal recovery, where sign identifiability depends on odd/even derivatives of $f$ and can require larger $\lambda$, and transformed stochastic block models, where community detection exhibits regime-dependent outliers and thresholds. Numerical experiments corroborate the theory, illustrating emergent outliers and alignment patterns that generalize spectral methods to nonlinear observations. The results offer a broad, non-Gaussian, non-identically distributed framework with potential implications for understanding feature learning in shallow neural networks.
Abstract
In this paper, we study a nonlinear spiked random matrix model where a nonlinear function is applied element-wise to a noise matrix perturbed by a rank-one signal. We establish a signal-plus-noise decomposition for this model and identify precise phase transitions in the structure of the signal components at critical thresholds of signal strength. To demonstrate the applicability of this decomposition, we then utilize it to study new phenomena in the problems of signed signal recovery in nonlinear models and community detection in transformed stochastic block models. Finally, we validate our results through a series of numerical simulations.
