arXiv · 2206.05668
Federated Learning on Riemannian Manifolds
Abstract
Federated learning (FL) has found many important applications in smart-phone-APP based machine learning applications. Although many algorithms have been studied for FL, to the best of our knowledge, algorithms for FL with nonconvex constraints have not been studied. This paper studies FL over Riemannian manifolds, which finds important applications such as federated PCA and federated kPCA. We propose a Riemannian federated SVRG (RFedSVRG) method to solve federated optimization over Riemannian manifolds. We analyze its convergence rate under different scenarios. Numerical experiments are conducted to compare RFedSVRG with the Riemannian counterparts of FedAvg and FedProx. We observed from the numerical experiments that the advantages of RFedSVRG are significant.
Explore related subjects
Keep this discovery
Jiaxiang Li, Shiqian Ma. 2022-06-12. Federated Learning on Riemannian Manifolds. https://arxiv.org/abs/2206.05668
Cite the original work for its findings. Save a collection to share your selection of sources.