Skip to yearly menu bar Skip to main content


Poster

How Gradient descent balances features: A dynamical analysis for two-layer neural networks

Zhenyu Zhu · Fanghui Liu · Volkan Cevher

Hall 3 + Hall 2B #447
[ ]
Fri 25 Apr midnight PDT — 2:30 a.m. PDT

Abstract: This paper investigates the fundamental regression task of learning kk neurons (\emph{a.k.a.} teachers) from Gaussian input, using two-layer ReLU neural networks with width m (\emph{a.k.a.} students) and m,k=O(1), trained via gradient descent under proper initialization and a small step-size. Our analysis follows a three-phase structure: \emph{alignment} after weak recovery, \emph{tangential growth}, and \emph{local convergence}, providing deeper insights into the learning dynamics of gradient descent (GD). We prove the global convergence at the rate of O(T3) for the zero loss of excess risk. Additionally, our results show that GD automatically groups and balances student neurons, revealing an implicit bias toward achieving the minimum balanced'' 2-norm in the solution. Our work extends beyond previous studies in exact-parameterization setting (m=k=1, (Yehudai and Ohad, 2020)) and single-neuron setting (mk=1, (Xu and Du, 2023)). The key technical challenge lies in handling the interactions between multiple teachers and students during training, which we address by refining the alignment analysis in Phase 1 and introducing a new dynamic system analysis for tangential components in Phase 2. Our results pave the way for further research on optimizing neural network training dynamics and understanding implicit biases in more complex architectures.

Live content is unavailable. Log in and register to view live content