Linear Convergence of Adaptive Stochastic Gradient Descent. - Yuege Xie; Xiaoxia Wu; Rachel A. Ward (AISTATS, 2020)

Weighted Optimization: better generalization by smoother interpolation. - Yuege Xie; Rachel A. Ward; Holger Rauhut; Hung-Hsu Chou (CoRR, 2020)

SHRIMP: Sparser Random Feature Models via Iterative Magnitude Pruning. - Yuege Xie; Bobby Shi; Hayden Schaeffer; Rachel A. Ward (CoRR, 2021)

AdaLoss: A Computationally-Efficient and Provably Convergent Adaptive Gradient Method. - Xiaoxia Wu; Yuege Xie; Simon Shaolei Du; Rachel A. Ward (AAAI, 2022)

SHRIMP: Sparser Random Feature Models via Iterative Magnitude Pruning. - Yuege Xie; Robert Shi; Hayden Schaeffer; Rachel A. Ward (MSML, 2022)

AdaWAC: Adaptively Weighted Augmentation Consistency Regularization for Volumetric Medical Image Segmentation. - Yijun Dong; Yuege Xie; Rachel A. Ward (CoRR, 2022)

Overparameterization and Generalization Error: Weighted Trigonometric Interpolation. - Yuege Xie; Hung-Hsu Chou; Holger Rauhut; Rachel A. Ward (SIAM J. Math. Data Sci., 2022)

Adaptively Weighted Data Augmentation Consistency Regularization for Robust Optimization under Concept Shift. - Yijun Dong; Yuege Xie; Rachel A. Ward (ICML, 2023)