Laser & Optoelectronics Progress, Volume. 60, Issue 6, 0610008(2023)

Improved ResNet Image Classification Model Based on Tensor Synthesis Attention

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

    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.

    Tools

    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

    Download Citation

    EndNote(RIS)BibTexPlain Text
    Save article for my favorites
    Paper Information

    Category: Image Processing

    Received: Oct. 29, 2021

    Accepted: Jan. 17, 2022

    Published Online: Mar. 7, 2023

    The Author Email: Zhang Jiaxin (1595775491@qq.com)

    DOI:10.3788/LOP212836

    Topics