Smoothed Normalization for Efficient Distributed Private Optimization

Fuente: arXiv
Saved in:
Bibliographic Details
Main Authors: Shulgin, Egor, Khirirat, Sarit, Richtárik, Peter
Format: Preprint
Published: 2025
Subjects:
Online Access:
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866912238099496960
author Shulgin, Egor
Khirirat, Sarit
Richtárik, Peter
author_facet Shulgin, Egor
Khirirat, Sarit
Richtárik, Peter
contents Federated learning enables training machine learning models while preserving the privacy of participants. Surprisingly, there is no differentially private distributed method for smooth, non-convex optimization problems. The reason is that standard privacy techniques require bounding the participants' contributions, usually enforced via $\textit{clipping}$ of the updates. Existing literature typically ignores the effect of clipping by assuming the boundedness of gradient norms or analyzes distributed algorithms with clipping but ignores DP constraints. In this work, we study an alternative approach via $\textit{smoothed normalization}$ of the updates motivated by its favorable performance in the single-node setting. By integrating smoothed normalization with an error-feedback mechanism, we design a new distributed algorithm $α$-$\sf NormEC$. We prove that our method achieves a superior convergence rate over prior works. By extending $α$-$\sf NormEC$ to the DP setting, we obtain the first differentially private distributed optimization algorithm with provable convergence guarantees. Finally, our empirical results from neural network training indicate robust convergence of $α$-$\sf NormEC$ across different parameter settings.
format Preprint
id arxiv_https___arxiv_org_abs_2502_13482
institution arXiv
publishDate 2025
record_format arxiv
spellingShingle Smoothed Normalization for Efficient Distributed Private Optimization
Shulgin, Egor
Khirirat, Sarit
Richtárik, Peter
Machine Learning
Cryptography and Security
Distributed, Parallel, and Cluster Computing
Optimization and Control
Federated learning enables training machine learning models while preserving the privacy of participants. Surprisingly, there is no differentially private distributed method for smooth, non-convex optimization problems. The reason is that standard privacy techniques require bounding the participants' contributions, usually enforced via $\textit{clipping}$ of the updates. Existing literature typically ignores the effect of clipping by assuming the boundedness of gradient norms or analyzes distributed algorithms with clipping but ignores DP constraints. In this work, we study an alternative approach via $\textit{smoothed normalization}$ of the updates motivated by its favorable performance in the single-node setting. By integrating smoothed normalization with an error-feedback mechanism, we design a new distributed algorithm $α$-$\sf NormEC$. We prove that our method achieves a superior convergence rate over prior works. By extending $α$-$\sf NormEC$ to the DP setting, we obtain the first differentially private distributed optimization algorithm with provable convergence guarantees. Finally, our empirical results from neural network training indicate robust convergence of $α$-$\sf NormEC$ across different parameter settings.
title Smoothed Normalization for Efficient Distributed Private Optimization
topic Machine Learning
Cryptography and Security
Distributed, Parallel, and Cluster Computing
Optimization and Control
url https://arxiv.org/abs/2502.13482