mlsanityData to 7 Oct 2026

Provable Learning of Random Hierarchy Models and Hierarchical Shallow-to-Deep Chaining

arXiv 2601.19756 ↗First posted 27 Jan 2026cs.LG stat.MLcited by 5

Not on the map. Why?

Abstract (arXiv)

The empirical success of deep learning is often attributed to deep networks' ability to exploit hierarchical structure in data, constructing increasingly complex features across layers. Yet despite substantial progress in deep learning theory, most optimization results sill focus on networks with only two or three layers, leaving the theoretical understanding of hierarchical learning in genuinely deep models limited. This leads to a natural question: can we prove that deep networks, trained with gradient-based methods and standard input-label pairs, can efficiently exploit hierarchical structure? In this work, we consider Random Hierarchy Models -- a hierarchical context-free grammar introduced by arXiv:2307.02129 and conjectured to separate deep and shallow networks. We prove that, under mild conditions, a deep convolutional network can be efficiently trained to learn this function class. Our proof builds on a general observation: if intermediate layers can receive clean signal from the labels and the relevant features are weakly identifiable, then layerwise training each individual layer suffices to hierarchically learn the target function.

Key facts. Cited by 5 papers in the map (data through 7 Oct 2026). Builds on 40 of its 58 references that are in the map. First posted on arXiv 27 Jan 2026.

Builds on

40 of 58 references.

References · 40

  1. 2015Deep Residual Learning for Image Recognitioncited by 51,809
  2. 2018Learning and Generalization in Overparameterized Neural Networks, Going Beyond Two Layerscited by 340
  3. 2016Benefits of depth in neural networkscited by 248
  4. 2015The Power of Depth for Feedforward Neural Networkscited by 181
  5. 2020Deep Network Approximation for Smooth Functionscited by 163
  6. 2022Neural Networks can Learn Representations with Gradient Descentcited by 161
  7. 2023SGD learning on neural networks: leap complexity and saddle-to-saddle dynamicscited by 142
  8. 2020Online stochastic gradient descent on non-convex losses from high-dimensional inferencecited by 119
  9. 2022Learning Single-Index Models with Shallow Neural Networkscited by 92
  10. 2019What Can ResNet Learn Efficiently, Going Beyond Kernels?cited by 91
  11. 2019The phase diagram of approximation rates for deep neural networkscited by 86
  12. 2023Physics of Language Models: Part 1, Learning Hierarchical Language Structurescited by 83
28 more
  1. 2023How Deep Neural Networks Learn Compositional Data: The Random Hierarchy Modelcited by 70
  2. 2020A Rigorous Framework for the Mean Field Limit of Multilayer Neural Networkscited by 68
  3. 2023How Two-Layer Neural Networks Learn, One (Giant) Step at a Timecited by 67
  4. 2016Depth-Width Tradeoffs in Approximating Natural Functions with Neural Networkscited by 58
  5. 2024Transformers Provably Solve Parity Efficiently with Chain of Thoughtcited by 49
  6. 2023Do Transformers Parse while Predicting the Masked Word?cited by 48
  7. 2017Depth Separation for Neural Networkscited by 46
  8. 2024Towards a theory of how the structure of language is acquired by deep neural networkscited by 38
  9. 2025Learning Compositional Functions with Transformers from Easy-to-Hard Datacited by 34
  10. 2020On the Global Convergence of Training Deep Linear ResNetscited by 29
  11. 2025Emergence and scaling laws in SGD learning of shallow neural networkscited by 29
  12. 2019Depth Separations in Neural Networks: What is Actually Being Separated?cited by 27
  13. 2024Probing the Latent Hierarchical Structure of Data via Diffusion Modelscited by 25
  14. 2019Is Deeper Better only when Shallow is Good?cited by 24
  15. 2016Deep Learning and Hierarchal Generative Modelscited by 23
  16. 2021The Connection Between Approximation, Depth Separation and Learnability in Neural Networkscited by 19
  17. 2020Backward Feature Correction: How Deep Learning Performs Deep (Hierarchical) Learningcited by 18
  18. 2021Depth separation beyond radial functionscited by 18
  19. 2023Provable Guarantees for Nonlinear Feature Learning in Three-Layer Neural Networkscited by 18
  20. 2021Optimization-Based Separations for Neural Networkscited by 17
  21. 2024U-Nets as Belief Propagation: Efficient Classification, Denoising, and Diffusion in Generative Hierarchical Modelscited by 17
  22. 2025Learning curves theory for hierarchically compositional data with power-law distributed featurescited by 17
  23. 2023Depth Separation with Multilayer Mean-Field Networkscited by 8
  24. 2023Learning Hierarchical Polynomials with Three-Layer Neural Networkscited by 8
  25. 2025Scaling Laws and Representation Learning in Simple Hierarchical Languages: Transformers vs. Convolutional Architecturescited by 5
  26. 2024Depth Separations in Neural Networks: Separating the Dimension from the Accuracycited by 4
  27. 2025Noise Sensitivity and Learning Lower Bounds for Hierarchical Functionscited by 2
  28. 2026Deep Networks Learn Deep Hierarchical Modelscited by 2

Built on by

Cited by year (2026 to 7 Oct)

Citing papers per year: 2026: 5.

5 papers.