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

DP-2Stage:将语言模型适配为差分隐私表格数据生成器

DP-2Stage: Adapting Language Models as Differentially Private Tabular Data Generators

CISPA Helmholtz Center for Information Security(CISPA 欧洲信息安全研究中心)

Tejumade Afonja, Hui-Po Wang, Raouf Kerkouche, Mario Fritz

AI总结:

本文提出DP-2Stage两阶段微调框架,通过先在伪数据集上非隐私微调再在私有数据集上差分隐私微调,有效解决了LLMs在差分隐私下生成表格数据时隐私预算分配低效的问题,显著提升了各指标性能。

摘要:

在差分隐私(DP)保护下生成表格数据确保了理论上的隐私保证,但对训练机器学习模型提出了挑战,主要原因在于需要在含噪监督信号下捕捉复杂结构。最近,预训练的大型语言模型(LLMs)——甚至是GPT-2规模的模型——在合成表格数据方面展现出了巨大潜力。然而,它们在DP约束下的应用在很大程度上仍未被探索。在这项工作中,我们通过将DP技术应用于合成表格数据的生成来填补这一空白。我们的发现表明,当使用DP进行微调时,LLMs在生成连贯文本方面面临困难,因为隐私预算被低效地分配给表格结构等非隐私元素。为了克服这一点,我们提出了DP-2Stage,一个用于差分隐私表格数据生成的两阶段微调框架。第一阶段在伪数据集上进行非隐私微调,随后在私有数据集上进行DP微调。我们的实证结果表明,与DP上下文中直接微调的LLMs相比,该方法在各种设置和指标上均提高了性能。我们在https://github.com/tejuafonja/DP-2Stage发布了代码和设置。

英文摘要:

Generating tabular data under differential privacy (DP) protection ensures theoretical privacy guarantees but poses challenges for training machine learning models, primarily due to the need to capture complex structures under noisy supervision signals. Recently, pre-trained Large Language Models (LLMs) -- even those at the scale of GPT-2 -- have demonstrated great potential in synthesizing tabular data. However, their applications under DP constraints remain largely unexplored. In this work, we address this gap by applying DP techniques to the generation of synthetic tabular data. Our findings shows that LLMs face difficulties in generating coherent text when fine-tuned with DP, as privacy budgets are inefficiently allocated to non-private elements like table structures. To overcome this, we propose DP-2Stage, a two-stage fine-tuning framework for differentially private tabular data generation. The first stage involves non-private fine-tuning on a pseudo dataset, followed by DP fine-tuning on a private dataset. Our empirical results show that this approach improves performance across various settings and metrics compared to directly fine-tuned LLMs in DP contexts. We release our code and setup at https://github.com/tejuafonja/DP-2Stage.

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

↑