I am currently a fourth year PhD student at the Department of Computer Sciences, UW-Madison. I am fortunate to be advised by Jelena Diakonikolas. Earlier, I received my BS in Mathematics from Zhiyuan Honors College at Shanghai Jiao Tong University.

My research interests are in optimization, with applications in machine learning and generative AI.

Please reach out via xcai74 [at] wisc [dot] edu.


Recent Papers

(chronologically, * denotes equal contribution)

Variance Reduced Halpern Iteration for Finite-Sum Monotone Inclusions.
Xufeng Cai*, Ahmet Alacaoglu*, Jelena Diakonikolas.
In Proc. ICLR'24, 2024.
abstract / arXiv

Empirical Risk Minimization with Shuffled SGD: A Primal-Dual Perspective and Improved Bounds.
Xufeng Cai, Cheuk Yin Lin, Jelena Diakonikolas.
arXiv preprint, arXiv:2306.12498, 2023.
abstract / arXiv

Cyclic Block Coordinate Descent With Variance Reduction for Composite Nonconvex Optimization.
Xufeng Cai, Chaobing Song, Stephen J. Wright, Jelena Diakonikolas.
In Proc. ICML'23, 2023.
abstract / arXiv

Stochastic Halpern Iteration with Variance Reduction for Stochastic Monotone Inclusions.
Xufeng Cai, Chaobing Song, Cristóbal Guzmán, Jelena Diakonikolas.
In Proc. NeurIPS'22, 2022.
abstract / arXiv

Talks

Stochastic Halpern Iteration with Variance Reduction for Stochastic Monotone Inclusions.
The seventh International Conference on Continuous Optimization (ICCOPT), Bethlehem, PA, USA. 2022.

Teaching

CS639, Foundations of Data Science, UW-Madison, Spring 2024 (TA).
CS639, Foundations of Data Science, UW-Madison, Spring 2022 (TA).
CS760, Machine Learning, UW-Madison, Spring 2021 (TA).
CS760, Machine Learning, UW-Madison, Fall 2020 (TA).

Miscellaneous

In my free time, I enjoy reading (not these days), photography (cameras and gear), and music (vinyls).