arXiv Machine Learning By Antonin Chodron de Courcel

Gradient descent at the Edge of Stability: free energy model and kinetic description of the two-layer network

Read the original on arXiv Machine Learning →

arXiv:2606. 05326v1 Announce Type: cross Abstract: 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 loss and the sharpness.

Summary generated by The Flow from the publisher's feed. The full article lives at arXiv Machine Learning.