Gradient Descent on Two-layer Nets: Margin Maximization and Simplicity Bias - Lyu, Kaifeng; Li, Zhiyuan; Wang, Runzhe; Arora, Sanjeev (Advances in Neural Information Processing Systems, 2021)