Chinese Journal of Liquid Crystals and Displays, Volume. 39, Issue 9, 1223(2024)

Hourglass attention and progressive hybrid Transformer for image classification

Yanfei PENG, Yun CUI*, Kun CHEN, and Yongxin LI
Author Affiliations
  • School of Electronic and Information Engineering,Liaoning Technical University,Huludao 125105,China
  • show less

    Transformer has a wide range of applications in image classification tasks, but in small dataset classification tasks, Transformer is affected by factors such as small amount of data and excessive amount of model parameters, which leads to low classification accuracy and slow convergence speed. Therefore, a progressive hybrid transformer model with hourglass attention is proposed. Firstly, the global feature relationships are modeled by the hourglass self-attention with down-up sampling, and up-sampling is used to supplement the information lost by the down-sampling operation, while the learning temperature parameters and negative diagonal mask are used to sharpen the fractional distribution of the attention to avoid excessive smoothing due to the excessive number of layers. Secondly, progressive down-sampling modules are designed to obtain the fine-grained multi-scale feature maps, which can effectively capture the low-dimensional feature information. Finally, a hybrid architecture is used, where the designed hourglass attention is used in the top stage, the pooling layer is used in the bottom stage instead of the attention module, and layer normalization with deep convolution is introduced to increase network locality. The proposed method is experimented on T-ImageNet, CIFAR10, CIFAR100, and SVHN datasets, the classification accuracy can reach 97.42%, and the computation and parameters are 3.41G and 25M. The experimental results show that compared with the comparison algorithms, the classification accuracy of the proposed method is significantly improved, with a significant reduction in computation and parameters, which improves the performance of Transformer model on small datasets.

    Keywords
    Tools

    Get Citation

    Copy Citation Text

    Yanfei PENG, Yun CUI, Kun CHEN, Yongxin LI. Hourglass attention and progressive hybrid Transformer for image classification[J]. Chinese Journal of Liquid Crystals and Displays, 2024, 39(9): 1223

    Download Citation

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

    Category:

    Received: Oct. 25, 2023

    Accepted: --

    Published Online: Nov. 13, 2024

    The Author Email: Yun CUI (1727015916@qq.com)

    DOI:10.37188/CJLCD.2023-0338

    Topics