用强化学习优化生成私有数据,提升联邦学习隐私保护下的模型性能。
POPri: Private Federated Learning using Preference-Optimized Synthetic Data
- 将客户端私有反馈视为奖励信号,用策略优化生成高质量私有合成数据。
- 在多个基准上使私有与非私有设置的预测准确率差距缩小至58%(前人仅28%)。
- 适用于注重隐私保护且需高精度文本建模的联邦学习场景。
在实际应用中,差分隐私联邦学习(DP-FL)是利用设备端私有数据训练模型的主流方法。近期研究提出,使用差分隐私合成数据的方法可能优于或超越传统DP-FL。现有生成DP合成数据的算法依赖于基于公开信息的精心提示工程和/或迭代的私有客户端反馈。我们的核心洞察是:先前方法收集的私有客户端反馈可被视为强化学习(RL)中的奖励信号。为此,我们提出策略优化私有数据(POPri)算法,利用直接偏好优化(DPO)等策略优化方法,微调大语言模型以生成高质量的差分隐私合成数据。为评估POPri,我们发布了LargeFedBench——一个用于在联邦客户端数据上无污染评估大语言模型的新联邦文本基准。在LargeFedBench数据集及Xie等人(2024)现有基准上,POPri显著提升了DP合成数据的实用性。相较于之前合成数据方法的28%提升,以及当前最先进的DP联邦学习方法仅3%的提升,POPri将全私有与非私有设置下的下一个词预测准确率差距缩小了最多达58%。代码与数据见https://github.com/meiyuw/POPri。
原文摘要 · Abstract (English)
In practical settings, differentially private Federated learning (DP-FL) is the dominant method for training models from private, on-device client data. Recent work has suggested that DP-FL may be enhanced or outperformed by methods that use DP synthetic data (Wu et al., 2024; Hou et al., 2024). The primary algorithms for generating DP synthetic data for FL applications require careful prompt engineering based on public information and/or iterative private client feedback. Our key insight is that the private client feedback collected by prior DP synthetic data methods (Hou et al., 2024; Xie et al., 2024) can be viewed as an RL (reinforcement learning) reward. Our algorithm, Policy Optimization for Private Data (POPri) harnesses client feedback using policy optimization algorithms such as Direct Preference Optimization (DPO) to fine-tune LLMs to generate high-quality DP synthetic data. To evaluate POPri, we release LargeFedBench, a new federated text benchmark for uncontaminated LLM evaluations on federated client data. POPri substantially improves the utility of DP synthetic data relative to prior work on LargeFedBench datasets and an existing benchmark from Xie et al. (2024). POPri closes the gap between next-token prediction accuracy in the fully-private and non-private settings by up to 58%, compared to 28% for prior synthetic data methods, and 3% for state-of-the-art DP federated learning methods. The code and data are available at https://github.com/meiyuw/POPri.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。