← Search

Ermin Wei

8 accepted papers

2025

Debiasing Federated Learning with Correlated Client Participation

ICLR 2025poster

In cross-device federated learning (FL) with millions of mobile clients, only a small subset of clients participate in training in every communication round, and Federated Averaging (FedAvg) is the most popular algorithm in practice. Existing analyses of FedAvg usually assume the participating clie…

Cited by 0SourcePDFScholar
2025

Improved Lower Bounds for First-order Stochastic Non-convex Optimization under Markov Sampling

ICML 2025poster

Unlike its vanilla counterpart with i.i.d. samples, stochastic optimization with Markovian sampling allows the sampling scheme following a Markov chain. This problem encompasses various applications that range from asynchronous distributed optimization to reinforcement learning. In this work, we low…

Cited by 0SourcePDFScholar
2024

Understanding Generalization of Federated Learning via Stability: Heterogeneity Matters

AISTATS 2024poster

Generalization performance is a key metric in evaluating machine learning models when applied to real-world applications. Good generalization indicates the model can predict unseen data correctly when trained under a limited number of data. Federated learning (FL), which has emerged as a popular dis…

Cited by 28SourcePDFScholar
2022

A Communication-efficient Algorithm with Linear Convergence for Federated Minimax Learning

NeurIPS 2022accept

In this paper, we study a large-scale multi-agent minimax optimization problem, which models many interesting applications in statistical learning and game theory, including Generative Adversarial Networks (GANs). The overall objective is a sum of agents' private local objective functions. We focus…

Cited by 18SourcePDFScholar