On the Training Instability of Shuffling SGD with Batch Normalization. - David Xing Wu; Chulhee Yun; Suvrit Sra (ICML, 2023)

Lower Bounds for Multiclass Classification with Overparameterized Linear Models. - David Xing Wu; Anant Sahai (ISIT, 2023)

Precise asymptotic generalization for multiclass classification with overparameterized linear models. - David Xing Wu; Anant Sahai (NeurIPS, 2023)

Provable weak-to-strong generalization via benign overfitting. - David Xing Wu; Anant Sahai (ICLR, 2025)