arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2402.03124cs.CRcs.CVcs.LG
收藏

致力于消除梯度反转攻击中的硬标签约束

Towards Eliminating Hard Label Constraints in Gradient Inversion Attacks

School of Artificial Intelligence, University of Chinese Academy of Sciences (UCAS)(中国科学院大学人工智能学院);Institute of Automation, Chinese Academy of Sciences (CASIA)(中国科学院自动化研究所)

Yanbo Wang, Jian Liang, Ran He

AI总结:

本研究聚焦梯度反转攻击的硬标签约束问题,提出首个可从单输入梯度同时恢复增强软标签与全连接层输入特征的算法,给出分析类标签恢复的必要条件,实验验证其标签恢复准确率及对图像重建的增益。

摘要:

梯度反转攻击旨在从联邦学习框架中暴露的中间梯度重建本地训练数据。尽管攻击已取得成功,但此前所有方法——从重建单个数据点起步,随后将单图像限制放宽到批次级别——都仅在硬标签约束下进行测试。即便对于单图像重建,我们仍缺乏一种基于分析的算法来恢复经增强的软标签。在本研究中,我们将研究重点从扩大批次规模转向探究硬标签约束,考虑了训练过程中使用标签平滑和mixup技术的更贴近现实的场景。具体而言,我们首次提出了一种全新算法,能够从单输入梯度中同时恢复真实增强标签和最后一层全连接层的输入特征,并给出了任何基于分析的标签恢复方法都需满足的必要条件。大量实验验证了该方法的标签恢复准确率,以及其对后续图像重建的增益效果。我们认为,分类任务中的软标签在梯度反转攻击领域值得进一步关注。

英文摘要:

Gradient inversion attacks aim to reconstruct local training data from intermediate gradients exposed in the federated learning framework. Despite successful attacks, all previous methods, starting from reconstructing a single data point and then relaxing the single-image limit to batch level, are only tested under hard label constraints. Even for single-image reconstruction, we still lack an analysis-based algorithm to recover augmented soft labels. In this work, we change the focus from enlarging batchsize to investigating the hard label constraints, considering a more realistic circumstance where label smoothing and mixup techniques are used in the training process. In particular, we are the first to initiate a novel algorithm to simultaneously recover the ground-truth augmented label and the input feature of the last fully-connected layer from single-input gradients, and provide a necessary condition for any analytical-based label recovery methods. Extensive experiments testify to the label recovery accuracy, as well as the benefits to the following image reconstruction. We believe soft labels in classification tasks are worth further attention in gradient inversion attacks.

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

↑