Momentum Benefits Non-iid Federated Learning Simply and Provably - Cheng, Ziheng; Huang, Xinmeng; Wu, Pengfei; Yuan, Kun (International Conference on Learning Representations, 2024)