MLCRLGDec 13, 2023

Differentially Private Gradient Flow based on the Sliced Wasserstein Distance

arXiv:2312.08227v34 citationsh-index: 9Trans. Mach. Learn. Res.
Originality Incremental advance
AI Analysis

This addresses privacy concerns in generative modeling for sensitive data, offering a novel approach with improved performance, though it appears incremental as it builds on existing differential privacy and gradient flow methods.

The paper tackles the problem of safeguarding privacy in sensitive training data for generative modeling by introducing a differentially private gradient flow based on the Sliced Wasserstein Distance, achieving higher-fidelity data generation at a low privacy budget compared to existing generator-based models.

Safeguarding privacy in sensitive training data is paramount, particularly in the context of generative modeling. This can be achieved through either differentially private stochastic gradient descent or a differentially private metric for training models or generators. In this paper, we introduce a novel differentially private generative modeling approach based on a gradient flow in the space of probability measures. To this end, we define the gradient flow of the Gaussian-smoothed Sliced Wasserstein Distance, including the associated stochastic differential equation (SDE). By discretizing and defining a numerical scheme for solving this SDE, we demonstrate the link between smoothing and differential privacy based on a Gaussian mechanism, due to a specific form of the SDE's drift term. We then analyze the differential privacy guarantee of our gradient flow, which accounts for both the smoothing and the Wiener process introduced by the SDE itself. Experiments show that our proposed model can generate higher-fidelity data at a low privacy budget compared to a generator-based model, offering a promising alternative.

Code Implementations1 repo
Foundations

The foundational work for this paper's niche, ranked by how specifically the neighbourhood builds on it — not by global fame.

Your Notes