FOLDS seminar: Dynamics of large-step gradient descent near flat minima
October 15 at 12:00 PM - 1:00 PM
Details
Venue
Zoom link: https://upenn.zoom.us/j/98220304722
Classically, convergence guarantees for gradient descent (GD) require that the step size be sufficiently small compared to the local curvature (“sharpness”) of the objective. This “small-step” condition is routinely violated in the practical training of deep learning models, wherein the favoured “large-step” GD is observed to converge with intriguing oscillatory dynamics and an implicit bias toward solutions of low curvature known as “flat minima”. In this talk, I will present a new geometric perspective on these phenomena. In particular, I will indicate how the loss geometry of certain multilayer models enables the proof of a new normal form for large-step GD near the global minima manifold. In this normal form, large-step GD is revealed to act as Riemannian GD on the objective sharpness along the solution manifold coupled to a flip bifurcation in the directions orthogonal to the solution manifold, enabling new local convergence theorems for large-step GD near flat minima.
Speaker

Lachlan MacDonald
Postdoctoral Researcher in the Vidal Lab at the University of Pennsylvania
Lachlan MacDonald is a postdoc in the Vidal Lab at the University of Pennsylvania studying the mathematical foundations of deep learning. He received his PhD in pure mathematics from the University of Wollongong in 2019 and held postdoc positions in pure mathematics at the Australian National University and the University of Adelaide. He shifted his focus to the theory of deep learning in 2021, holding postdoc positions at the University of Adelaide and Johns Hopkins University before joining UPenn in 2024. His research leverages differential geometry and dynamical systems theory to understand the nonlinear dynamics and implicit bias of neural network training.

