ConvNeXt-Based Fine-Grained Image Classification and Bilinear Attention Mechanism Model

被引:14
|
作者
Li, Zhiheng [1 ,2 ]
Gu, Tongcheng [2 ,3 ]
Li, Bing [2 ,3 ]
Xu, Wubin [2 ,3 ]
He, Xin [2 ,3 ]
Hui, Xiangyu [2 ,3 ]
机构
[1] Guangxi Liugong Machinery Co Ltd, Liuzhou 545006, Peoples R China
[2] Guangxi Sci & Technol Univ, Guangxi Earthmoving Machinery Collaborat Innovat, Liuzhou 545006, Peoples R China
[3] Guangxi Sci & Technol Univ, Coll Mech & Automot Engn, Liuzhou 545006, Peoples R China
来源
APPLIED SCIENCES-BASEL | 2022年 / 12卷 / 18期
关键词
deep learning; convolutional neural network; image classification; fine grained; attention mechanism;
D O I
10.3390/app12189016
中图分类号
O6 [化学];
学科分类号
0703 ;
摘要
Featured Application This paper studies attention-related optimizations and innovations for the ConvNeXt network proposed in January 2022, providing a reference for subsequent researchers to optimize this network . Thus far, few studies have been conducted on fine-grained classification tasks for the latest convolutional neural network ConvNeXt, and no effective optimization method has been made available. To achieve more accurate fine-grained classification, this paper proposes two attention embedding methods based on ConvNeXt network and designs a new bilinear CBAM; simultaneously, a multiscale, multi-perspective and all-around attention framework is proposed, which is then applied in ConvNeXt. Experimental verification shows that the accuracy rate of the improved ConvNeXt for fine-grained image classification reaches 87.8%, 91.2%, and 93.2% on fine-grained classification datasets CUB-200-2011, Stanford Cars, and FGVC Aircraft, respectively, showing increases of 2.7%, 0.3% and 0.4%, respectively, compared to those of the original network without optimization, and increases of 3.7%, 8.0% and 2.0%, respectively, compared to those of the traditional BCNN. In addition, ablation experiments are set up to verify the effectiveness of the proposed attention framework.
引用
收藏
页数:18
相关论文
共 50 条
  • [31] An attention-based bilinear feature extraction mechanism for fine-grained laser welding molten pool/keyhole defect recognition
    Liu, Tianyuan
    Zheng, Pai
    Chen, Huabin
    Zhang, Lingling
    JOURNAL OF MANUFACTURING PROCESSES, 2023, 87 : 150 - 159
  • [32] Application of Image Classification for Fine-Grained Nudity Detection
    Ion, Cristian
    Minea, Cristian
    ADVANCES IN VISUAL COMPUTING, ISVC 2019, PT I, 2020, 11844 : 3 - 15
  • [33] Fine-Grained Radio Frequency Fingerprint Recognition Network Based on Attention Mechanism
    Zhang, Yulan
    Hu, Jun
    Jiang, Rundong
    Lin, Zengrong
    Chen, Zengping
    ENTROPY, 2024, 26 (01)
  • [34] Press-Plate State Recognition Based on Improved Bilinear Fine-Grained Model
    Yang Qianwen
    Zhou Ke
    LASER & OPTOELECTRONICS PROGRESS, 2021, 58 (20)
  • [35] A fine-grained image classification method based on information interaction
    Zhu, Shuo
    Zhang, Xukang
    Wang, Yu
    Wang, Zongyang
    Sun, Jiahao
    IET IMAGE PROCESSING, 2024, 18 (14) : 4852 - 4861
  • [36] Research on the Fine-grained Plant Image Classification
    Hu, Zhifeng
    Zhang, Yin
    Tan, Liang
    PROCEEDINGS OF THE 2016 4TH INTERNATIONAL CONFERENCE ON MACHINERY, MATERIALS AND INFORMATION TECHNOLOGY APPLICATIONS, 2016, 71 : 1307 - 1311
  • [37] Research and Application of Fine-Grained Image Classification Based on Small Collar Dataset
    Huang Chengcheng
    Yuan Jian
    Qin Xiao
    FRONTIERS IN COMPUTATIONAL NEUROSCIENCE, 2022, 15
  • [38] Image Classification With Tailored Fine-Grained Dictionaries
    Shu, Xiangbo
    Tang, Jinhui
    Qi, Guo-Jun
    Li, Zechao
    Jiang, Yu-Gang
    Yan, Shuicheng
    IEEE TRANSACTIONS ON CIRCUITS AND SYSTEMS FOR VIDEO TECHNOLOGY, 2018, 28 (02) : 454 - 467
  • [39] Fine-grained bird image classification based on counterfactual method of vision transformer model
    Tianhua Chen
    Yanyue Li
    Qinghua Qiao
    The Journal of Supercomputing, 2024, 80 : 6221 - 6239
  • [40] A Broad Sparse Fine-Grained Image Classification Model Based on Dictionary Selection Strategy
    Zheng, Jianjie
    Liang, Pengpeng
    Zhao, Huimin
    Deng, Wu
    IEEE TRANSACTIONS ON RELIABILITY, 2024, 73 (01) : 576 - 588