arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2304.04625cs.LGcs.CRcs.CV

基于强化学习的黑盒模型反演攻击

Reinforcement Learning-Based Black-Box Model Inversion Attacks

KAIST(韩国科学技术院)

Gyojin Han, Jaehyun Choi, Haeil Lee, Junmo Kim

AI总结:

针对现有黑盒模型反演攻击无法保证在预定查询次数内完成且性能不及白盒攻击的局限,提出一种基于强化学习的方法,将潜在空间搜索形式化为MDP问题,利用生成图像的置信度作为奖励,成功实现了最先进的攻击性能并恢复目标模型私有信息。

摘要:

模型反演攻击是一种隐私攻击,仅通过访问模型即可重构用于训练机器学习模型的私有数据。近年来,利用生成对抗网络(GAN)从公共数据集中蒸馏知识的白盒模型反演攻击因其出色的攻击性能而备受关注。另一方面,当前利用GAN的黑盒模型反演攻击存在一些问题,例如无法保证在预定的查询访问次数内完成攻击过程,或无法达到与白盒攻击相同的性能水平。为了克服这些局限性,我们提出了一种基于强化学习的黑盒模型反演攻击。我们将潜在空间搜索形式化为马尔可夫决策过程(MDP)问题,并用强化学习加以解决。我们的方法利用生成图像的置信度分数为智能体提供奖励。最后,可以使用在MDP中训练的智能体找到的潜在向量来重构私有数据。在多个数据集和模型上的实验结果表明,我们的攻击通过取得最先进的攻击性能,成功恢复了目标模型的私有信息。我们通过提出一种更先进的黑盒模型反演攻击,强调了隐私保护机器学习研究的重要性。

英文摘要:

Model inversion attacks are a type of privacy attack that reconstructs private data used to train a machine learning model, solely by accessing the model. Recently, white-box model inversion attacks leveraging Generative Adversarial Networks (GANs) to distill knowledge from public datasets have been receiving great attention because of their excellent attack performance. On the other hand, current black-box model inversion attacks that utilize GANs suffer from issues such as being unable to guarantee the completion of the attack process within a predetermined number of query accesses or achieve the same level of performance as white-box attacks. To overcome these limitations, we propose a reinforcement learning-based black-box model inversion attack. We formulate the latent space search as a Markov Decision Process (MDP) problem and solve it with reinforcement learning. Our method utilizes the confidence scores of the generated images to provide rewards to an agent. Finally, the private data can be reconstructed using the latent vectors found by the agent trained in the MDP. The experiment results on various datasets and models demonstrate that our attack successfully recovers the private information of the target model by achieving state-of-the-art attack performance. We emphasize the importance of studies on privacy-preserving machine learning by proposing a more advanced black-box model inversion attack.

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

↑