Riemannian SVRG Using Barzilai–Borwein Method as Second-Order Approximation for Federated Learning
In this paper, we propose a modified RFedSVRG method by incorporating the Barzilai–Borwein (BB) method to approximate second-order information on the manifold for Federated Learning (FL). Moreover, we use the BB strategy to obtain self-adjustment of step size. We show the convergence of our methods under some assumptions. The numerical experiments on both synthetic and real datasets demonstrate that the proposed methods outperform some used methods in FL in some test problems.
Authors
- Xiao He (ORCID: https://orcid.org/0000-0003-3998-7060)
- Kai You Wang (ORCID: https://orcid.org/0000-0002-6170-4744)
- Tao Yan (ORCID: https://orcid.org/0000-0003-3511-8562)
Institutions
- Nanjing University of Science and Technology (CN)
Publication Details
- Journal
- Symmetry
- Published
- 2024-08-23
- DOI
- https://doi.org/10.3390/sym16091101
- Primary Topic
- Stochastic Gradient Optimization Techniques
- Type
- article
- Field-Weighted Citation Impact
- 0.00
Funders
- National Natural Science Foundation of China