用隐私保护的聚类生成图像嵌入,实现高精度模型训练。
Private Training & Data Generation by Clustering Embeddings
- 通过差分隐私聚类拟合高斯混合模型生成嵌入数据。
- 在标准数据集上分类准确率达当前最优水平。
- 可生成真实感图像,适合需隐私保护的视觉任务。
深度神经网络通常依赖大规模高质量数据集以在诸多机器学习任务中取得优异表现。当训练涉及敏感数据时,模型可能无意中记忆并泄露个人信息,甚至重构整个训练样本。差分隐私(DP)提供了保护个体数据的稳健框架,一种新方法是先用私有方式生成合成数据集,再进行后续训练。本文提出一种基于差分隐私聚类的新型原理性方法,用于生成合成图像嵌入,其核心是在合适的嵌入空间中拟合高斯混合模型(GMM)。该方法在分离条件下可证明地学习到一个GMM。实验证明,仅用两层神经网络在合成嵌入上训练,即可在标准基准数据集上达到状态最先进水平的分类准确率。此外,我们展示了该方法能生成逼真的合成图像,其下游分类性能媲美当前最优方法。该方法具有高度通用性,编码器与解码器模块可自由替换以适应不同任务;同时具备高度可扩展性,仅由线性复杂度子程序构成,可在分布式系统中高效实现。
原文摘要 · Abstract (English)
Deep neural networks often use large, high-quality datasets to achieve high performance on many machine learning tasks. When training involves potentially sensitive data, this process can raise privacy concerns, as large models have been shown to unintentionally memorize and reveal sensitive information, including reconstructing entire training samples. Differential privacy (DP) provides a robust framework for protecting individual data and in particular, a new approach to privately training deep neural networks is to approximate the input dataset with a privately generated synthetic dataset, before any subsequent training algorithm. We introduce a novel principled method for DP synthetic image embedding generation, based on fitting a Gaussian Mixture Model (GMM) in an appropriate embedding space using DP clustering. Our method provably learns a GMM under separation conditions. Empirically, a simple two-layer neural network trained on synthetically generated embeddings achieves state-of-the-art (SOTA) classification accuracy on standard benchmark datasets. Additionally, we demonstrate that our method can generate realistic synthetic images that achieve downstream classification accuracy comparable to SOTA methods. Our method is quite general, as the encoder and decoder modules can be freely substituted to suit different tasks. It is also highly scalable, consisting only of subroutines that scale linearly with the number of samples and/or can be implemented efficiently in distributed systems.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。