Shallow Binary Features Enhance the Robustness of Deep Convolutional Neural Networks

Jin Ding,Jie-Chao Zhao,Yong-Zhi Sun,Ping Tan,Ji-En Ma
DOI: https://doi.org/10.1109/cac59555.2023.10451035
2023-01-01
Abstract:It is known that small noise can mislead the deep convolutional neural networks and have them make misclassifications. The researchers have poured much effort to study the effective and efficient approaches to enhance the robustness of deep networks, among which, adversarial training is a popular one. However, adversarial training is time-consuming and resource-consuming properties, which limit its applicability. It is natural to ask: can we enhance the robustness of deep models without adversarial training? In this paper, motivated by the binary edge features of clean examples and adversarial examples, we propose a binary feature branch module (BFBM for short) for deep convolutional networks. This module consists two kinds of layers, one is called Sobel layer, and the other is called threshold layer. The effect of Sobel layer is to learn four parallel edge feature maps using horizontal edge detector, vertical edge detector, positive edge detector, and negative edge detector, respectively. The goal of threshold layer is to convert the edge features to the binary features. After finishing this, the binary features along with the conventional texture features are taken as the inputs for the fully connected layers to output logits. The module is lightweight, and can work with any existing backbone. The experiments on CIFAR-10 and MNIST show that the BFBM-VGG16 model and BFBM-ResNet34 model can achieve averagely 35% higher accuracy than the original ones under FGSM and PGD attacks.
What problem does this paper attempt to address?