Towards understanding how momentum improves generalization in deep learning. - Jelassi, Samy; Li, Yuanzhi (CoRR, 2022)