DP-MicroAdam: Private and Frugal Algorithm for Training and Fine-tuning

Fuente: arXiv
Enregistré dans:
Détails bibliographiques
Auteurs principaux: Hudişteanu, Mihaela, Kalinin, Nikita P., Cyffers, Edwige
Format: Preprint
Publié: 2025
Sujets:
Accès en ligne:
Tags: Ajouter un tag
Pas de tags, Soyez le premier à ajouter un tag!
_version_ 1866915641888342016
author Hudişteanu, Mihaela
Kalinin, Nikita P.
Cyffers, Edwige
author_facet Hudişteanu, Mihaela
Kalinin, Nikita P.
Cyffers, Edwige
contents Adaptive optimizers are the de facto standard in non-private training as they often enable faster convergence and improved performance. In contrast, differentially private (DP) training is still predominantly performed with DP-SGD, typically requiring extensive compute and hyperparameter tuning. We propose DP-MicroAdam, a memory-efficient and sparsity-aware adaptive DP optimizer. We prove that DP-MicroAdam converges in stochastic non-convex optimization at the optimal $\mathcal{O}(1/\sqrt{T})$ rate, up to privacy-dependent constants. Empirically, DP-MicroAdam outperforms existing adaptive DP optimizers and achieves competitive or superior accuracy compared to DP-SGD across a range of benchmarks, including CIFAR-10, large-scale ImageNet training, and private fine-tuning of pretrained transformers. These results demonstrate that adaptive optimization can improve both performance and stability under differential privacy.
format Preprint
id arxiv_https___arxiv_org_abs_2511_20509
institution arXiv
publishDate 2025
record_format arxiv
spellingShingle DP-MicroAdam: Private and Frugal Algorithm for Training and Fine-tuning
Hudişteanu, Mihaela
Kalinin, Nikita P.
Cyffers, Edwige
Machine Learning
Adaptive optimizers are the de facto standard in non-private training as they often enable faster convergence and improved performance. In contrast, differentially private (DP) training is still predominantly performed with DP-SGD, typically requiring extensive compute and hyperparameter tuning. We propose DP-MicroAdam, a memory-efficient and sparsity-aware adaptive DP optimizer. We prove that DP-MicroAdam converges in stochastic non-convex optimization at the optimal $\mathcal{O}(1/\sqrt{T})$ rate, up to privacy-dependent constants. Empirically, DP-MicroAdam outperforms existing adaptive DP optimizers and achieves competitive or superior accuracy compared to DP-SGD across a range of benchmarks, including CIFAR-10, large-scale ImageNet training, and private fine-tuning of pretrained transformers. These results demonstrate that adaptive optimization can improve both performance and stability under differential privacy.
title DP-MicroAdam: Private and Frugal Algorithm for Training and Fine-tuning
topic Machine Learning
url https://arxiv.org/abs/2511.20509