arXiv:2501.01090cs.CRcs.CV2025-01AAAI被引 7

用陷阱模型反制模型窃取,让盗版者训练出的模型自动失效。

HoneypotNet: Backdoor Attacks Against Model Extraction

  • 在目标模型中植入陷阱层,通过双层优化使输出带毒
  • 在四个数据集上验证,能以高成功率注入后门
  • 适合想保护模型版权或防御窃取的开发者使用

模型提取攻击是一种推理阶段攻击,攻击者通过向黑盒目标模型发起若干查询,并利用其预测结果训练一个替代模型来近似原模型的功能与性能。这类攻击对生产环境中的模型和机器学习即服务(MLaaS)平台构成严重威胁,可能导致重大经济损失。现有防御方法包括主动防御(修改输出或增加查询开销)和被动防御(检测恶意查询或使用水印)。本文提出一种新防御范式:以攻为防,将模型输出改造成有毒数据,使任何试图用其训练替代模型的攻击者都将被污染。为此,我们提出轻量级后门攻击方法 HoneypotNet:用蜜罐层替换目标模型分类层,并通过双层优化,以影子模型模拟提取过程,微调蜜罐层使其输出具有毒性,同时保持原模型性能。我们在四个常用基准数据集上实证表明,HoneypotNet 能以高成功率将后门注入替代模型。该后门不仅可用于所有权验证,还能破坏替代模型功能,有效威慑模型提取攻击。

原文摘要 · Abstract (English)

Model extraction attacks are one type of inference-time attacks that approximate the functionality and performance of a black-box victim model by launching a certain number of queries to the model and then leveraging the model's predictions to train a substitute model. These attacks pose severe security threats to production models and MLaaS platforms and could cause significant monetary losses to the model owners. A body of work has proposed to defend machine learning models against model extraction attacks, including both active defense methods that modify the model's outputs or increase the query overhead to avoid extraction and passive defense methods that detect malicious queries or leverage watermarks to perform post-verification. In this work, we introduce a new defense paradigm called attack as defense which modifies the model's output to be poisonous such that any malicious users that attempt to use the output to train a substitute model will be poisoned. To this end, we propose a novel lightweight backdoor attack method dubbed HoneypotNet that replaces the classification layer of the victim model with a honeypot layer and then fine-tunes the honeypot layer with a shadow model (to simulate model extraction) via bi-level optimization to modify its output to be poisonous while remaining the original performance. We empirically demonstrate on four commonly used benchmark datasets that HoneypotNet can inject backdoors into substitute models with a high success rate. The injected backdoor not only facilitates ownership verification but also disrupts the functionality of substitute models, serving as a significant deterrent to model extraction attacks.

模型安全后门攻击防御机制

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