Differentially Private Adaptation of Diffusion Models via Noisy Aggregated Embeddings

Read full text
Authors: Pura Peetathawatchai, Weining Chen, Berivan Isik, Oluwasanmi O. Koyejo, Albert No
Cited by: 0
References: 32

Abstract

Personalizing large-scale diffusion models poses serious privacy risks, especially when adapting to small, sensitive datasets. A common approach is to fine-tune the model using differentially private stochastic gradient descent (DP-SGD), but this suffers from severe utility degradation due to the high noise needed for privacy, particularly in the small data regime. We propose an alternative that leverages Textual Inversion (TI), which learns an embedding vector for an image or set of images, to enable adaptation under differential privacy (DP) constraints. Our approach, Differentially Private Aggregation via Textual Inversion (DPAgg-TI), adds calibrated noise to the aggregation of per-image embeddings to ensure formal DP guarantees while preserving high output fidelity. We show that DPAgg-TI outperforms DP-SGD finetuning in both utility and robustness under the same privacy budget, achieving results closely matching the non-private baseline on style adaptation tasks using private artwork from a single artist and Paris 2024 Olympic pictograms. In contrast, DP-SGD fails to generate meaningful outputs in this setting.

Fields of Study & Subjects

Privacy-Preserving Technologies in Data

Publication details

Journal / source
2026 IEEE Conference on Secure and Trustworthy Machine Learning (SaTML)
Publisher
IEEE
Pages
446–461
Publication date
1 March 2026
Language
EN