Postdoc, National University of Singapore
1 paper at NeurIPS 2025
We propose FedWMSAM, a novel method that combines personalized momentum and dynamic SAM to achieve faster convergence and flatter minima in federated learning, effectively addressing client drift and data heterogeneity.
