欢迎访问《图学学报》 分享到:

图学学报 ›› 2023, Vol. 44 ›› Issue (3): 560-569.DOI: 10.11996/JG.j.2095-302X.2023030560

• 计算机图形学与虚拟现实 • 上一篇    下一篇

结合对抗样本检测和重构的三维点云防御框架

赵玉琨(), 任爽(), 张鑫云   

  1. 北京交通大学计算机与信息技术学院,北京 100044
  • 收稿日期:2022-10-27 接受日期:2022-12-11 出版日期:2023-06-30 发布日期:2023-06-30
  • 通讯作者: 任爽(1981-),男,副教授,博士。主要研究方向为机器学习、计算机视觉和虚拟现实技术等。E-mail:sren@bjtu.edu.cn
  • 作者简介:

    赵玉琨(1999-),女,硕士研究生。主要研究方向为三维点云对抗攻击与防御。E-mail:yukun0125@bjtu.edu.cn

  • 基金资助:
    国家自然科学基金项目(62072025)

A 3D point cloud defense framework combined with adversarial examples detection and reconstruction

ZHAO Yu-kun(), REN Shuang(), ZHANG Xin-yun   

  1. School of Computer and Information Technology, Beijing Jiaotong University, Beijing 100044, China
  • Received:2022-10-27 Accepted:2022-12-11 Online:2023-06-30 Published:2023-06-30
  • Contact: REN Shuang (1981-), associate professor, Ph.D. His main research interests cover machine learning, computer vision, virtual reality technology, etc. E-mail:sren@bjtu.edu.cn
  • About author:

    ZHAO Yu-kun (1999-), master student. Her main research interests cover 3D point cloud adversarial attack and defense. E-mail:yukun0125@bjtu.edu.cn

  • Supported by:
    National Natural Science Foundation of China(62072025)

摘要:

近年来,三维点云深度神经网络已应用于许多高安全性的任务中。然而,对抗样本可以很容易地使正常训练的深度学习模型做出错误的预测,所以需要提高深度神经网络输入数据的鲁棒性。现存的三维点云防御网络效率低,也无法很好地恢复点云的曲面变形和点分布。针对现存问题,提出了一种将对抗样本检测与重构相结合的三维点云对抗防御网络框架。对于一个输入样本,首先由基于重构误差的检测器对其进行检测。若是对抗样本,则由一个基于变分自编码器的重构器对其进行重构后再输入分类网络中。变分自编码器的结构可以更好地学习隐空间上的数值空洞,对于点云形状的恢复效果也更好。实验部分,在ModelNet40数据集上对多种经典的分类模型进行了攻击,并测试了检测器-重构器防御框架对这些攻击的防御效果。实验表明,该防御方法在PointNet上的分类准确率均优于其他防御方法,特别是在防御基于显著图和对抗生成网络的攻击中表现出色。该防御网络框架可以将删除200点的攻击的准确率从47.65%提高到75.02%。通过消融实验和可视化重构结果分别证明了检测器和重构器的有效提升对整体分类准确率的效果。

关键词: 对抗防御, 对抗攻击, 对抗样本检测, 点云重构, 点云分类

Abstract:

The development of 3D point cloud deep neural networks has enabled their application in many high-security tasks. However, adversarial examples could easily lead the normally trained deep learning models to make incorrect predictions, making it essential to improve the robustness of input data to deep neural networks. The existing 3D point cloud defense networks are inefficient and fail to recover the surface deformation of the point cloud and point distribution adequately. To address these issues, a 3D point cloud adversarial defense network framework combining adversarial example detection and reconstruction was proposed. The input sample was first detected by an error-based detector before and after reconstruction. If it was an adversarial example, it was then reconstructed by a variational autoencoder-based reformer before being fed into the classification network. The variational autoencoder’s structure enhanced the learning of numerical voids on the hidden space, and the same number of points before and after reconstruction ensured efficient subsequent networks and better recovery of the point cloud shape. For the experiments, a variety of classical classification models were attacked on the ModelNet40 dataset, and the effectiveness of the detector-reformer defense framework against these attacks was tested. The experiments demonstrated that the defense method outperformed all other defense methods in terms of classification accuracy on PointNet and especially performed well in the attack based on the saliency map and the adversarial generation network. The detector-reformer defense network framework could improve the accuracy from 47.65% to 75.02% on the dropping attacks with 200 points lost. The effectiveness of the detector and reformer on the overall classification accuracy was demonstrated by ablation experiments and visual reconstruction results.

Key words: adversarial defense, adversarial attack, detection of adversarial examples, point cloud reconstruction, point cloud classification

中图分类号: