Laser & Optoelectronics Progress, Volume. 60, Issue 6, 0610008(2023)
Improved ResNet Image Classification Model Based on Tensor Synthesis Attention
An improved ResNet-101 network model that fuses tensor synthesis attention (RTSA Net-101) is proposed to solve insufficient feature extraction and the indiscriminate contribution of the extracted features when processing image classification tasks using a convolutional neural network. First, the image features are extracted using a Resnet-101 backbone network and the tensor synthesis attention module is embedded after the convolution structure of the residual network. The features are calculated using a three-tensor product to obtain the attention feature matrix. Next, the Softmax function is used to normalize the attention feature matrix to assign weights to features and distinguish the contribution of features. Finally, the weighted sum of the weights and critical values are calculated as the final features in our proposed method to improve the image classification performance. Comparative experiments are conducted on natural image datasets, CIFAR-10 and CIFAR-100, and street brand dataset, SVHN. The classification accuracy values of the models are 96.12%, 81.60%, and 96.67%, respectively, and the average test running time of the images are 0.0258 s, 0.0260 s, and 0.0262 s, respectively. The experimental results show that compared with the other seven advanced image classification models, the RTSA Net-101 model can achieve higher classification accuracy and shorter test run time, and it can effectively enhance the feature learning ability of the network, thereby render the proposed model innovative and efficient.
Get Citation
Copy Citation Text
Yunfei Qiu, Jiaxin Zhang, Hai Lan, Jiaxu Zong. Improved ResNet Image Classification Model Based on Tensor Synthesis Attention[J]. Laser & Optoelectronics Progress, 2023, 60(6): 0610008
Category: Image Processing
Received: Oct. 29, 2021
Accepted: Jan. 17, 2022
Published Online: Mar. 7, 2023
The Author Email: Zhang Jiaxin (1595775491@qq.com)