Federated Learning on Riemannian Manifolds: A Gradient-Free Projection-Based Approach

Fuente: arXiv
Saved in:
Bibliographic Details
Main Authors: Wang, Hongye, Pan, Zhaoye, He, Chang, Li, Jiaxiang, Jiang, Bo
Format: Preprint
Published: 2025
Subjects:
Online Access:
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866911113171435520
author Wang, Hongye
Pan, Zhaoye
He, Chang
Li, Jiaxiang
Jiang, Bo
author_facet Wang, Hongye
Pan, Zhaoye
He, Chang
Li, Jiaxiang
Jiang, Bo
contents Federated learning (FL) has emerged as a powerful paradigm for collaborative model training across distributed clients while preserving data privacy. However, existing FL algorithms predominantly focus on unconstrained optimization problems with exact gradient information, limiting its applicability in scenarios where only noisy function evaluations are accessible or where model parameters are constrained. To address these challenges, we propose a novel zeroth-order projection-based algorithm on Riemannian manifolds for FL. By leveraging the projection operator, we introduce a computationally efficient zeroth-order Riemannian gradient estimator. Unlike existing estimators, ours requires only a simple Euclidean random perturbation, eliminating the need to sample random vectors in the tangent space, thus reducing computational cost. Theoretically, we first prove the approximation properties of the estimator and then establish the sublinear convergence of the proposed algorithm, matching the rate of its first-order counterpart. Numerically, we first assess the efficiency of our estimator using kernel principal component analysis. Furthermore, we apply the proposed algorithm to two real-world scenarios: zeroth-order attacks on deep neural networks and low-rank neural network training to validate the theoretical findings.
format Preprint
id arxiv_https___arxiv_org_abs_2507_22855
institution arXiv
publishDate 2025
record_format arxiv
spellingShingle Federated Learning on Riemannian Manifolds: A Gradient-Free Projection-Based Approach
Wang, Hongye
Pan, Zhaoye
He, Chang
Li, Jiaxiang
Jiang, Bo
Optimization and Control
Machine Learning
Federated learning (FL) has emerged as a powerful paradigm for collaborative model training across distributed clients while preserving data privacy. However, existing FL algorithms predominantly focus on unconstrained optimization problems with exact gradient information, limiting its applicability in scenarios where only noisy function evaluations are accessible or where model parameters are constrained. To address these challenges, we propose a novel zeroth-order projection-based algorithm on Riemannian manifolds for FL. By leveraging the projection operator, we introduce a computationally efficient zeroth-order Riemannian gradient estimator. Unlike existing estimators, ours requires only a simple Euclidean random perturbation, eliminating the need to sample random vectors in the tangent space, thus reducing computational cost. Theoretically, we first prove the approximation properties of the estimator and then establish the sublinear convergence of the proposed algorithm, matching the rate of its first-order counterpart. Numerically, we first assess the efficiency of our estimator using kernel principal component analysis. Furthermore, we apply the proposed algorithm to two real-world scenarios: zeroth-order attacks on deep neural networks and low-rank neural network training to validate the theoretical findings.
title Federated Learning on Riemannian Manifolds: A Gradient-Free Projection-Based Approach
topic Optimization and Control
Machine Learning
url https://arxiv.org/abs/2507.22855