Power-Law Spectrum of the Random Feature Model
Organizations: Department of Mathematics and Statistics, McGill University · Department of Mathematics, University of Southern California
Abstract
Scaling laws for neural networks, in which the loss decays as a power-law in the number of parameters, data, and compute, depend fundamentally on the spectral structure of the data covariance, with power-law eigenvalue decay appearing ubiquitously in vision and language tasks. A central question is whether this spectral structure is preserved or destroyed when data passes through the basic building block of a neural network: a random linear projection followed by a nonlinear activation. We study this question for the random feature model: given data where has -power-law spectrum (, ), a Gaussian sketch matrix , and an entrywise monomial , we characterize the eigenvalues of the population random-feature covariance . We prove matching upper and lower bounds: for all , the -th eigenvalue is of order . For , the -th eigenvalue is of order up to a polylog factor. That is, the power-law exponent is inherited exactly from the input covariance, modified only by a logarithmic correction that depends on the monomial degree . The proof combines a dyadic head-tail decomposition with Wick chaos expansions for higher-order monomials and random matrix concentration inequalities.
Figures & tables
Appendix figures & tables9 assets
Supplementary material from the paper’s appendix.
Appendix
| Architecture | Layer | Dim | Slope | |
|---|---|---|---|---|
| Input (CIFAR-10) | 3072 | 0.998 | ||
| MLP, ReLU (random) | L1 / L2 / L3 / L4 | 1024 | / / / | |
| MLP, ReLU (trained) | L1 / L2 / L3 / L4 | 1024 | / / / | |
| MLP, (random) | L1 / L2 / L3 / L4 | 1024 | / / / | |
| MLP, (random) | L1 / L2 / L3 / L4 | 1024 | / / / | |
| MLP, ReLU+RMSNorm (random) | L1 / L2 / L3 / L4 | 1024 | / / / | |
Random Feature Methods
Random feature methods approximate kernels and neural networks with finite randomized embeddings to reduce computation while retaining useful accuracy.
See all 46 papers