Interface of PDEs and Machine Learning
FedCBO: Reaching Group Consensus in Clustered Federated Learning through Consensus-based Optimization. [pdf]
Jose A. Carrillo, Nicolas Garcia Trillos, Sixu Li, Yuhua Zhu*.
Why Resampling Outperforms Reweighting for Correcting Sampling Bias with Stochastic Gradients. [pdf]
Jing An, Lexing Ying, Yuhua Zhu*.
International Conference on Learning Representations (ICLR), 2021.
A Sharp Convergence Rate for a Model Equation of the Asynchronous Stochastic Gradient Descent. [pdf]
Yuhua Zhu, Lexing Ying.
Communications in Mathematical Sciences, 19(3), 851-863, 2020.
On Large Batch Training and Sharp Minima: A Fokker-Planck perspective. [pdf]
Xiaowu Dai and Yuhua Zhu*.
Journal of Statistical Theory and Practice (JSTP), special issue on "Advances in Deep Learning", 2020.
A Consensus-Based Global Optimization Method for High Dimensional Machine Learning Problems. [pdf]
Jose Carrillo, Shi Jin, Lei Li and Yuhua Zhu*
ESAIM: Control, Optimisation and Calculus of Variations 27, S5, 2020.
Borrowing From the Future: An Attempt to Address Double Sampling. [pdf]
Yuhua Zhu and Lexing Ying.
Mathematical and Scientific Machine Learning, PMLR 107:246-268, 2020.
*: Alphabetical authorship.