Momentum benefits non-iid federated learning simply and provably Z Cheng, X Huang, K Yuan International Conference on Learning Representations (ICLR 2024), 2023 | 8 | 2023 |
Particle-based Variational Inference with Generalized Wasserstein Gradient Flow Z Cheng, S Zhang, L Yu, C Zhang Advances in Neural Information Processing Systems (NeurIPS 2023), 2023 | 3 | 2023 |
The Limits and Potentials of Local SGD for Distributed Heterogeneous Learning with Intermittent Communication KK Patel, M Glasgow, A Zindari, L Wang, SU Stich, Z Cheng, N Joshi, ... The 37th Annual Conference on Learning Theory (COLT 2024), 2024 | 2 | 2024 |
Kernel Semi-Implicit Variational Inference Z Cheng, L Yu, T Xie, S Zhang, C Zhang International Conference on Machine Learning (ICML 2024), 2024 | | 2024 |
Reflected Flow Matching T Xie, Y Zhu, L Yu, T Yang, Z Cheng, S Zhang, X Zhang, C Zhang International Conference on Machine Learning (ICML 2024), 2024 | | 2024 |
Joint Graph Learning and Model Fitting in Laplacian Regularized Stratified Models Z Cheng, J Zhang, A Agrawal, S Boyd arXiv preprint arXiv:2305.02573, 2023 | | 2023 |