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

Read full text
Authors: Yunwei Ren, Yatin Dandi, Florent Krząkała, Jason D. Lee
Cited by: 0

Abstract

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.

Fields of Study & Subjects

Neural Networks and Applications Stochastic Gradient Optimization Techniques Machine Learning and Algorithms

Publication details

Journal / source
arXiv (Cornell University)
Publication date
27 January 2026
Language
EN
Document type
Preprint