The Computational Advantage of Depth in Learning High-Dimensional Hierarchical Targets
Understanding the advantages of deep neural networks trained by gradient descent (GD) compared to shallow models remains an open theoretical challenge. In this paper, we introduce a class of target functions (single and multi-index Gaussian hierarchical targets) that incorporate a hierarchy of latent subspace dimensionalities. This framework enables us to analytically study the learning dynamics and generalization performance of deep networks compared to shallow ones in the high-dimensional limit. Specifically, our main theorem shows that feature learning with GD successively reduces the effective dimensionality, transforming a high-dimensional problem into a sequence of lower-dimensional ones. This enables learning the target function with drastically less samples than with shallow networks. While the results are proven in a controlled training setting, we also discuss more common training procedures and argue that they learn through the same mechanisms.
Cite this item
Dandi, Y., Zdeborová, L., Pesce, L., & Krzakala, F. (2025, September 18). The Computational Advantage of Depth in Learning High-Dimensional Hierarchical Targets. The Thirty-Ninth Annual Conference on Neural Information Processing Systems, San Diego, CA, US, December 2, 2025. https://infoscience.epfl.ch/handle/20.500.14299/263045
Click Copy to copy the citation, or select the text manually.
12192_The_Computational_Advant.pdf
Main Document
openaccess
CC BY
2.36 MB
Adobe PDF
13f774763073e8dc09abb848b3344971