arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2402.18752cs.LGcs.CR
收藏

使用有限公共数据预训练差分隐私模型

Pre-training Differentially Private Models with Limited Public Data

Amazon(亚马逊);University of Southern California(南加州大学);University of Minnesota(明尼苏达大学)

Zhiqi Bu, Xinwei Zhang, Mingyi Hong, Sheng Zha, George Karypis

AI总结:

针对预训练阶段应用差分隐私(DP)易致性能下降、无法保护多数预训练数据的问题,该研究通过理论分析提出利用有限公共数据的DP持续预训练策略,仅用10%公共数据便取得媲美标准预训练、优于现有DP预训练的效果,并开源了模型。

摘要:

大型基础模型的优异性能依赖于海量高质量数据的使用,这些数据往往包含需要正式保护的敏感、隐私及受版权保护的内容。尽管差分隐私(DP)是衡量模型安全防护程度的重要方法,但由于在预训练阶段应用DP会导致性能下降,其应用通常局限于模型微调阶段。因此,DP目前尚无法为初始预训练过程中使用的大部分数据提供保护。\n在本研究中,我们首先通过分析每轮迭代的损失提升,从理论层面阐释了DP训练的有效性。我们得出一项关键发现:使用有限的公共数据可显著缓解DP优化器的性能下降问题,由此提出了一种新颖的DP持续预训练策略。实验结果表明,仅使用10%的公共数据,我们的策略在ImageNet-21k上(ε=8)可达到41.5%的DP准确率,在下游任务Places365和iNaturalist-2021上的非DP准确率分别为55.7%和60.0%,与当前最优的标准预训练效果相当,且大幅优于现有的DP预训练模型。我们的DP预训练模型已在fastDP库(https://github.com/awslabs/fast-differential-privacy/releases/tag/v2.1)中发布。

英文摘要:

The superior performance of large foundation models relies on the use of massive amounts of high-quality data, which often contain sensitive, private and copyrighted material that requires formal protection. While differential privacy (DP) is a prominent method to gauge the degree of security provided to the models, its application is commonly limited to the model fine-tuning stage, due to the performance degradation when applying DP during the pre-training stage. Consequently, DP is yet not capable of protecting a substantial portion of the data used during the initial pre-training process. In this work, we first provide a theoretical understanding of the efficacy of DP training by analyzing the per-iteration loss improvement. We make a key observation that DP optimizers' performance degradation can be significantly mitigated by the use of limited public data, which leads to a novel DP continual pre-training strategy. Empirically, using only 10\% of public data, our strategy can achieve DP accuracy of 41.5\% on ImageNet-21k (with $ε=8$), as well as non-DP accuracy of 55.7\% and and 60.0\% on downstream tasks Places365 and iNaturalist-2021, respectively, on par with state-of-the-art standard pre-training and substantially outperforming existing DP pre-trained models. Our DP pre-trained models are released in fastDP library (https://github.com/awslabs/fast-differential-privacy/releases/tag/v2.1)

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑