arXiv:2411.04811cs.LGcs.AI2024-11被引 3

首个针对图像回归模型的后门攻击防御方法,有效识别隐藏在连续输出中的恶意行为。

Defending Deep Regression Models against Backdoor Attacks

  • 基于回归模型输出与特征空间特性设计逆向优化机制
  • 在四个数据集上对多种攻击均实现高精度检测与防御
  • 适合关注安全关键场景下回归模型可靠性的研究人员

深度回归模型广泛应用于各类安全敏感任务,但易受后门攻击。尽管已有大量分类模型的防御方法,但它们无法适配回归模型——因回归输出为连续值而非离散标签,攻击目标无限多;且后门触发依赖特征空间所有神经元激活值,难以被现有方法检测。为此,本文提出 DRMGuard,首个针对图像域深度回归模型的后门检测与防御框架。该方法基于回归模型输出空间与特征空间的独特性质,构建逆向优化问题以恢复潜在后门模式。我们在两个回归任务和四个数据集上进行了全面评估,结果表明,DRMGuard能持续抵御多种后门攻击。此外,我们还将四种主流分类模型防御方法泛化至回归场景,对比显示,DRMGuard显著优于所有基线方法。

原文摘要 · Abstract (English)

Deep regression models are used in a wide variety of safety-critical applications, but are vulnerable to backdoor attacks. Although many defenses have been proposed for classification models, they are ineffective as they do not consider the uniqueness of regression models. First, the outputs of regression models are continuous values instead of discretized labels. Thus, the potential infected target of a backdoored regression model has infinite possibilities, which makes it impossible to be determined by existing defenses. Second, the backdoor behavior of backdoored deep regression models is triggered by the activation values of all the neurons in the feature space, which makes it difficult to be detected and mitigated using existing defenses. To resolve these problems, we propose DRMGuard, the first defense to identify if a deep regression model in the image domain is backdoored or not. DRMGuard formulates the optimization problem for reverse engineering based on the unique output-space and feature-space characteristics of backdoored deep regression models. We conduct extensive evaluations on two regression tasks and four datasets. The results show that DRMGuard can consistently defend against various backdoor attacks. We also generalize four state-of-the-art defenses designed for classifiers to regression models, and compare DRMGuard with them. The results show that DRMGuard significantly outperforms all those defenses.

后门攻击回归模型模型安全防御

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