arXiv:2502.14370cs.LGcs.CV2025-02

用强化学习高效破解黑盒模型,还原训练数据

PPO-MI: Efficient Black-Box Model Inversion via Proximal Policy Optimization

  • 将数据重构转化为马尔可夫决策过程,用PPO算法探索生成模型潜空间
  • 仅需模型输出预测,查询效率高,重建精度优于现有方法
  • 无需梯度或白盒访问,适用于多种模型与数据集,适合隐私风险评估

模型逆向攻击会通过训练好的模型重建私有训练数据,构成重大隐私风险。现有方法多依赖梯度估计或需要白盒访问模型参数,限制了其在实际场景中的应用。本文提出PPO-MI,一种基于强化学习的黑盒模型逆向攻击新框架。将逆向任务建模为马尔可夫决策过程,代理通过生成模型的潜空间进行探索,仅利用模型预测结果重构私有样本。采用基于动量的状态转移机制与平衡预测准确性和探索性的奖励函数,结合近端策略优化(PPO),实现高效的潜空间搜索和高查询效率。大量实验表明,PPO-MI在所需攻击知识更少的情况下仍优于现有方法,并对多种模型架构和数据集具有鲁棒性。结果凸显其在真实黑盒场景下的有效性与通用性,揭示了部署机器学习模型的严重隐私漏洞。

原文摘要 · Abstract (English)

Model inversion attacks pose a significant privacy risk by attempting to reconstruct private training data from trained models. Most of the existing methods either depend on gradient estimation or require white-box access to model parameters, which limits their applicability in practical scenarios. In this paper, we propose PPO-MI, a novel reinforcement learning-based framework for black-box model inversion attacks. Our approach formulates the inversion task as a Markov Decision Process, where an agent navigates the latent space of a generative model to reconstruct private training samples using only model predictions. By employing Proximal Policy Optimization (PPO) with a momentum-based state transition mechanism, along with a reward function balancing prediction accuracy and exploration, PPO-MI ensures efficient latent space exploration and high query efficiency. We conduct extensive experiments illustrates that PPO-MI outperforms the existing methods while require less attack knowledge, and it is robust across various model architectures and datasets. These results underline its effectiveness and generalizability in practical black-box scenarios, raising important considerations for the privacy vulnerabilities of deployed machine learning models.

模型逆向强化学习隐私安全黑盒攻击

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。