← Search

Xufeng Cai

5 accepted papers

2024

Tighter Convergence Bounds for Shuffled SGD via Primal-Dual Perspective

NeurIPS 2024poster

Stochastic gradient descent (SGD) is perhaps the most prevalent optimization method in modern machine learning. Contrary to the empirical practice of sampling from the datasets \emph{without replacement} and with (possible) reshuffling at each epoch, the theoretical counterpart of SGD usually relies…

Cited by 1SourcePDFScholar
2024

Variance Reduced Halpern Iteration for Finite-Sum Monotone Inclusions

ICLR 2024poster

Machine learning approaches relying on such criteria as adversarial robustness or multi-agent settings have raised the need for solving game-theoretic equilibrium problems. Of particular relevance to these applications are methods targeting finite-sum structure, which generically arises in empirical…

Cited by 12SourcePDFScholar
2023

Cyclic Block Coordinate Descent With Variance Reduction for Composite Nonconvex Optimization

ICML 2023poster

Nonconvex optimization is central in solving many machine learning problems, in which block-wise structure is commonly encountered. In this work, we propose cyclic block coordinate methods for nonconvex optimization problems with non-asymptotic gradient norm guarantees. Our convergence analysis is b…

Cited by 21SourcePDFScholar
2022

Stochastic Halpern Iteration with Variance Reduction for Stochastic Monotone Inclusions

NeurIPS 2022accept

We study stochastic monotone inclusion problems, which widely appear in machine learning applications, including robust regression and adversarial learning. We propose novel variants of stochastic Halpern iteration with recursive variance reduction. In the cocoercive---and more generally Lipschitz-m…