基于域自适应的鲁棒图像分类算法
作者:
作者单位:

作者简介:

通讯作者:

中图分类号:

TP391.41

基金项目:

山西省基础研究计划项目 (202303021212222);山西省重点研发计划项目(202202010101005);中国高校产学研创新基金(2021ZYA11005)


Robust Image Classification Algorithm Based on Domain Adaptation
Author:
Affiliation:

Fund Project:

  • 摘要
  • |
  • 图/表
  • |
  • 访问统计
  • |
  • 参考文献
  • |
  • 相似文献
  • |
  • 引证文献
  • |
  • 资源附件
  • |
  • 文章评论
    摘要:

    对抗样本的存在容易误导深度学习模型输出错误的预测,严重影响深度学习模型的稳健性。为了提高模型的鲁棒性,抵御对样样本的干扰,本文以深度图像分类网络为研究对象,提出了基于域自适的鲁棒图像分类方法。首先,通过分析干净图像和对抗样本的数据分布特点,使用域自适应学习方法对齐干净图像和对抗样本的特征空间。其次,将干净图像和对抗样本分别作为源域和目标域,构建生成对抗分类网络,最后,构造对抗学习的线性损失函数优化网络。采用l∞范数和l2范数约束的对抗样本验证所提算法,相较于标准训练和对抗训练算法,在MNIST-M数据集上,所提算法的标准准确率提高了4.3%和1.23%。 在CIFAR-10数据集上提高了1.23%和20.45%。同时,在3种对抗样本上的鲁棒准确率均提高超过10%。在遥感场景分类SIRI-WHU数据集上鲁棒准确率达到了79.6%。实验结果表明,所提算法能有效提高图像分类模型的标准准确率和鲁棒准确率,使模型在面对对抗样本扰动时表现出更强的鲁棒性。

    Abstract:

    The existence of adversarial examples can easily mislead deep learning models into making incorrect predictions, severely affecting the robustness of these models. To enhance the robustness of models and resist interference from adversarial samples, a domain adaptive robust image classification method is proposed. Firstly, by analyzing the distribution characteristics of clean images and adversarial examples, domain adaptation learning methods are used to align the feature spaces of clean images and adversarial examples. Secondly, clean images and adversarial examples are treated as the source domain and target domain respectively, to construct a generative adversarial classification network. Finally, an adversarial learning linear loss function is constructed to optimize the network. The proposed algorithm, validated with adversarial samples constrained by l∞ and l2 norms, demonstrates an improvement of 4.3% and 1.23% in standard accuracy compared to standard training and adversarial training algorithms on the MNIST-M dataset, and an improvement of 1.23% and 20.45% on the CIFAR-10 dataset. Meanwhile, the robust accuracy on three types of adversarial samples is increased by more than 10%. The robust accuracy on the remote sensing scene classification SIRI-WHU dataset reaches 79.6%. Experimental results indicate that the proposed algorithm effectively enhances the standard accuracy and robust accuracy of image classification models, demonstrating stronger robustness when facing adversarial sample perturbations.

    参考文献
    相似文献
    引证文献
引用本文

李林娟,张兆祥,贺赟. 基于域自适应的鲁棒图像分类算法[J]. 科学技术与工程, 2024, 24(36): 15558-15566.
Li Linjuan, Zhang Zhaoxiang, He Yun. Robust Image Classification Algorithm Based on Domain Adaptation[J]. Science Technology and Engineering,2024,24(36):15558-15566.

复制
文章指标
  • 点击次数:
  • 下载次数:
  • HTML阅读次数:
  • 引用次数:
历史
  • 收稿日期:2024-02-21
  • 最后修改日期:2024-10-18
  • 录用日期:2024-05-02
  • 在线发布日期: 2025-01-02
  • 出版日期:
×
2026年会通知 | “技术经济学驱动智能经济生态构建与治理变革”——中国技术经济学会第三十三届学术年会(2026)会议通知暨征文启事(第一轮)
亟待确认版面费归属稿件,敬请作者关注