Mean--Fluctuation Dynamics at the Edge of Stability

Antonin Chodron de Courcel

We study the dynamics of gradient descent in the Edge of Stability regime, where the learning rate is large enough to induce persistent oscillations in the trajectory, which has be…