A novel ResNet50-based attention mechanism for image classification

Abstract

Image classification tasks often compress the neural network model to reduce the number of parameters, which leads to a decrease in classification accuracy. herefore, we propose a novel ResNet50-based attention mechanism for image classification. ResNet50 network is used to extract image features and input the features into the graph neural network as node features. Then, packet convolution and depth-separable convolution are used to compress the residual network. The attention mechanism is introduced into the network backbone to make it focus on the important part of the neighborhood and help the branch network to extract key information. The accuracy of 5-way 1-shot task classification on three publicly available datasets reaches 86.32%, 92.21% and 92.19%, respectively. The proposed method has achieved remarkable results in image classification tasks.

Keywords

• image classification
• resnet50
• attention mechanism
• depth-separable convolution
• packet convolution