Nonconvex Federated Learning on Compact Smooth Submanifolds With Heterogeneous Data - Zhang, Jiaojiao; Hu, Jiang; So, Anthony Man-Cho; Johansson, Mikael (Advances in Neural Information Processing Systems, 2024)