Differentially Private Gradient-Tracking-Based Distributed Stochastic Optimization over Directed Graphs

Fuente: arXiv
Guardado en:
Detalles Bibliográficos
Autores principales: Chen, Jialong, Wang, Jimin, Zhang, Ji-Feng
Formato: Preprint
Publicado: 2025
Materias:
Acceso en línea:
Etiquetas: Agregar Etiqueta
Sin Etiquetas, Sea el primero en etiquetar este registro!
_version_ 1866913027681419264
author Chen, Jialong
Wang, Jimin
Zhang, Ji-Feng
author_facet Chen, Jialong
Wang, Jimin
Zhang, Ji-Feng
contents This paper proposes a differentially private gradient-tracking-based distributed stochastic optimization algorithm over directed graphs. In particular, privacy noises are incorporated into each agent's state and tracking variable to mitigate information leakage, after which the perturbed states and tracking variables are transmitted to neighbors. We design two novel schemes for the step-sizes and the sampling number within the algorithm. The sampling parameter-controlled subsampling method employed by both schemes enhances the differential privacy level, and ensures a finite cumulative privacy budget even over infinite iterations. The algorithm achieves both almost sure and mean square convergence for nonconvex objectives. Furthermore, when nonconvex objectives satisfy the Polyak-Lojasiewicz condition, Scheme (S1) achieves a polynomial mean square convergence rate, and Scheme (S2) achieves an exponential mean square convergence rate. The trade-off between privacy and convergence is presented. The effectiveness of the algorithm and its superior performance compared to existing works are illustrated through numerical examples of distributed training on the benchmark datasets "MNIST" and "CIFAR-10".
format Preprint
id arxiv_https___arxiv_org_abs_2501_06793
institution arXiv
publishDate 2025
record_format arxiv
spellingShingle Differentially Private Gradient-Tracking-Based Distributed Stochastic Optimization over Directed Graphs
Chen, Jialong
Wang, Jimin
Zhang, Ji-Feng
Systems and Control
This paper proposes a differentially private gradient-tracking-based distributed stochastic optimization algorithm over directed graphs. In particular, privacy noises are incorporated into each agent's state and tracking variable to mitigate information leakage, after which the perturbed states and tracking variables are transmitted to neighbors. We design two novel schemes for the step-sizes and the sampling number within the algorithm. The sampling parameter-controlled subsampling method employed by both schemes enhances the differential privacy level, and ensures a finite cumulative privacy budget even over infinite iterations. The algorithm achieves both almost sure and mean square convergence for nonconvex objectives. Furthermore, when nonconvex objectives satisfy the Polyak-Lojasiewicz condition, Scheme (S1) achieves a polynomial mean square convergence rate, and Scheme (S2) achieves an exponential mean square convergence rate. The trade-off between privacy and convergence is presented. The effectiveness of the algorithm and its superior performance compared to existing works are illustrated through numerical examples of distributed training on the benchmark datasets "MNIST" and "CIFAR-10".
title Differentially Private Gradient-Tracking-Based Distributed Stochastic Optimization over Directed Graphs
topic Systems and Control
url https://arxiv.org/abs/2501.06793