1. A novel ResNet50-based attention mechanism for image classification
- Author
-
Jingsi Zhang, Xiaosheng Yu, Xiaoliang Lei, and Chengdong Wu
- Subjects
image classification ,resnet50 ,attention mechanism ,depth-separable convolution ,packet convolution ,Engineering (General). Civil engineering (General) ,TA1-2040 ,Chemical engineering ,TP155-156 ,Physics ,QC1-999 - 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.
- Published
- 2024
- Full Text
- View/download PDF