Scaling Limits of Constant-Stepsize SGD at Flat Minima
This work proves existence, uniqueness, and geometric convergence to an augmented invariant law in a Wasserstein distance induced by an $\alpha$-dependent metric and analysis of the small-stepsize scaling limit gives corresponding results for coordinate-separable objectives with unequal flatness exponents.