mlsanityData to 14 Oct 2026

Learning Orthogonal Multi-Index Models Beyond Small Initialization: Incremental Learning, Competitive Dynamics and Symmetry

arXiv 2609.10879 ↗First posted 9 Sep 2026cs.LG stat.MLcited by 0

Not on the map. Why?

Abstract (arXiv)

Recent work has identified incremental learning in shallow networks trained on single-index and multi-index models. However, existing analyses often rely on simplifying settings, such as small initialization, correlation loss, or layer-wise training. These choices reduce neuron interactions and leave some feature learning dynamics under standard initialization unexplored. We study training dynamics for polynomial-width two-layer networks learning orthogonal multi-index targets under standard initialization using polynomially many samples. We first prove that incremental learning still occurs: the loss decreases sequentially according to the Hermite expansion of the target, with lower-order components learned before higher-order components recover the individual target directions. In this standard initialization regime, training also shows a competitive reallocation of parameter mass: after the total mass fits the target mean and stabilizes, mass shifts into the target subspace and then concentrates on aligned neurons. Our theoretical analysis uses slightly modified gradient flow, while vanilla gradient descent empirically exhibits the same qualitative dynamics. Technically, we introduce a symmetry-based finite-width approximation via symmetrized networks, rather than comparing directly with an infinite-width limit. This yields better control of approximation errors and may be of independent interest.

Key facts. No citing papers in the map yet (data through 14 Oct 2026). Builds on 22 of its 62 references that are in the map. First posted on arXiv 9 Sep 2026.

Builds on

22 of 62 references.

References · 22

  1. 2016Gaussian Error Linear Units (GELUs)cited by 3,502
  2. 2013Exact solutions to the nonlinear dynamics of learning in deep linear neural networkscited by 763
  3. 2020Towards Understanding Ensemble, Knowledge Distillation and Self-Distillation in Deep Learningcited by 297
  4. 2017Searching for Activation Functionscited by 169
  5. 2023Explaining grokking through circuit efficiencycited by 111
  6. 2021Saddle-to-Saddle Dynamics in Deep Linear Networks: Small Initialization Training, Symmetry, and Sparsitycited by 85
  7. 2017Learning One-hidden-layer Neural Networks with Landscape Designcited by 81
  8. 2023Dichotomy of Early and Late Phase Implicit Biases Can Provably Induce Grokkingcited by 80
  9. 2022Neural Networks Efficiently Learn Low-Dimensional Representations with SGDcited by 63
  10. 2023On Learning Gaussian Multi-index Models with Gradient Flowcited by 57
  11. 2023SGD Finds then Tunes Features in Two-Layer Neural Networks with near-Optimal Sample Complexity: A Case Study in the XOR problemcited by 39
  12. 2025Emergence and scaling laws in SGD learning of shallow neural networkscited by 32
10 more
  1. 2025Dynamical Decoupling of Generalization and Overfitting in Large Two-Layer Networkscited by 25
  2. 2024Fundamental computational limits of weak learnability in high-dimensional multi-index modelscited by 19
  3. 2024Learning Multi-Index Models with Neural Networks via Mean-Field Langevin Dynamicscited by 15
  4. 2024Mean-field Analysis on Two-layer Neural Networks from a Kernel Perspectivecited by 13
  5. 2026Phase Transitions for Feature Learning in Neural Networkscited by 11
  6. 2023Depth Separation with Multilayer Mean-Field Networkscited by 9
  7. 2024Learning Gaussian Multi-Index Models with Gradient Flow: Time Complexity and Directional Convergencecited by 9
  8. 2024Learning Orthogonal Multi-Index Models: A Fine-Grained Information Exponent Analysiscited by 6
  9. 2025Neural Networks Learn Generic Multi-Index Models Near Information-Theoretic Limitcited by 6
  10. 2025Propagation of Chaos in One-hidden-layer Neural Networks beyond Logarithmic Timecited by 4

Built on by

Not cited yet.