提出新型梯度反演攻击Grab,可高效恢复语言模型训练数据
Uncovering Gradient Inversion Risks in Practical Language Model Training
- 设计双优化机制,兼顾层间丢弃掩码与离散词序优化
- 在真实设置下实现最高92.9%的文本恢复率,提升超48%
- 揭示语言模型联邦训练中的严重隐私风险,适合安全研究者关注
梯度反演攻击被证实是联邦学习(FL)在视觉领域的重要隐私威胁,但在语言模型中常被认为效果有限或依赖不切实际的训练条件,主要因文本数据的离散性带来的挑战。本文提出一种面向语言模型的专用梯度反演攻击方法Grab(gradient inversion with hybrid optimization),采用交替优化策略:在层间同步优化丢弃掩码以提升词元恢复效果,并通过离散优化有效建模词序。在基准设置下,该方法相比使用辅助模型的离散优化策略,文本恢复率提升最高达28.9%;在真实训练场景中,提升幅度更达48.5%。实验表明,其最大恢复率可达92.9%,显著暴露了语言模型联邦训练模式下的隐私漏洞。
原文摘要 · Abstract (English)
The gradient inversion attack has been demonstrated as a significant privacy threat to federated learning (FL), particularly in continuous domains such as vision models. In contrast, it is often considered less effective or highly dependent on impractical training settings when applied to language models, due to the challenges posed by the discrete nature of tokens in text data. As a result, its potential privacy threats remain largely underestimated, despite FL being an emerging training method for language models. In this work, we propose a domain-specific gradient inversion attack named Grab (gradient inversion with hybrid optimization). Grab features two alternating optimization processes to address the challenges caused by practical training settings, including a simultaneous optimization on dropout masks between layers for improved token recovery and a discrete optimization for effective token sequencing. Grab can recover a significant portion (up to 92.9% recovery rate) of the private training data, outperforming the attack strategy of utilizing discrete optimization with an auxiliary model by notable improvements of up to 28.9% recovery rate in benchmark settings and 48.5% recovery rate in practical settings. Grab provides a valuable step forward in understanding this privacy threat in the emerging FL training mode of language models.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。