• Laser & Optoelectronics Progress
  • Vol. 60, Issue 6, 0610008 (2023)
Yunfei Qiu1, Jiaxin Zhang1、*, Hai Lan2, and Jiaxu Zong3
Author Affiliations
  • 1College of Software, Liaoning Technical University, Huludao 125105, Liaoning, China
  • 2Quanzhou Institute of Equipment Manufacturing Haixi Institutes, Chinese Academy of Sciences, Quanzhou 362216, Fujian, China
  • 3Yuanqi Industrial Technology Company, Qingdao 266000, Shandong, China
  • show less
    DOI: 10.3788/LOP212836 Cite this Article Set citation alerts
    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 Copy Citation Text show less

    Abstract

    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.
    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
    Download Citation