Abstract:Multi-index models - functions which only depend on the covariates through a non-linear transformation of their projection on a subspace - are a useful benchmark for investigating feature learning with neural nets. This paper examines the theoretical boundaries of efficient learnability in this hypothesis class, focusing on the minimum sample complexity required for weakly recovering their low-dimensional structure with first-order iterative algorithms, in the high-dimensional regime where the number of samples $n\!=\!\alpha d$ is proportional to the covariate dimension $d$. Our findings unfold in three parts: (i) we identify under which conditions a trivial subspace can be learned with a single step of a first-order algorithm for any $\alpha\!>\!0$; (ii) if the trivial subspace is empty, we provide necessary and sufficient conditions for the existence of an easy subspace where directions that can be learned only above a certain sample complexity $\alpha\!>\!\alpha_c$, where $\alpha_{c}$ marks a computational phase transition. In a limited but interesting set of really hard directions -- akin to the parity problem -- $\alpha_c$ is found to diverge. Finally, (iii) we show that interactions between different directions can result in an intricate hierarchical learning phenomenon, where directions can be learned sequentially when coupled to easier ones. We discuss in detail the grand staircase picture associated to these functions (and contrast it with the original staircase one). Our theory builds on the optimality of approximate message-passing among first-order iterative methods, delineating the fundamental learnability limit across a broad spectrum of algorithms, including neural networks trained with gradient descent, which we discuss in this context.

Noise misleads rotation invariant algorithms on sparse targets

Low Rank Matrix Recovery with Adversarial Sparse Noise

On Ridge Estimation in High-dimensional Rotationally Sparse Linear Regression

When Do Neural Networks Outperform Kernel Methods?

Pruning is Optimal for Learning Sparse Features in High-Dimensions

Sparsity in Continuous-Depth Neural Networks

Optimality of Approximate Message Passing Algorithms for Spiked Matrix Models with Rotationally Invariant Noise

Mind the spikes: Benign overfitting of kernels and neural networks in fixed dimension

Fundamental computational limits of weak learnability in high-dimensional multi-index models

Adversarial Examples Exist in Two-Layer ReLU Networks for Low Dimensional Linear Subspaces

Computational-Statistical Gaps for Improper Learning in Sparse Linear Regression

Optimally rotated coordinate systems for adaptive least-squares regression on sparse grids

Rotational Invariant Dimensionality Reduction Algorithms.

Does SGD really happen in tiny subspaces?

Bayes-optimal learning of an extensive-width neural network from quadratically many samples

Outliers with Opposing Signals Have an Outsized Effect on Neural Network Optimization

Gradient Descent Fails to Learn High-frequency Functions and Modular Arithmetic

Weak Correlations as the Underlying Principle for Linearization of Gradient-Based Learning Systems

A Variational Inequality Model for Learning Neural Networks

Nonlinear Stochastic Gradient Descent and Heavy-tailed Noise: A Unified Framework and High-probability Guarantees