Semiconductor Optoelectronics, Volume. 46, Issue 1, 90(2025)

Classification of Steel Surface Defects Based on Multi-Attention Mechanism

YING Hejie1,2, LAI Lianfeng1,2, REN Xuehang1, XIONG Lingling3, and XUE Zhangqi1
Author Affiliations
  • 1College of Mechanical and Electrical Engineering Ningde Normal University, Ningde 352100, CHN
  • 2The Collaborative Innovation Center of Ningde Normal University Ningde Normal University, Ningde 352100, CHN
  • 3College of Information Engineering Ningde Normal University, Ningde 352100, CHN
  • show less

    In view of the characteristics of steel defects and the current classification methods' dependence on complex models and non-heuristic attention mechanisms, this paper proposes a multiattention-based classification method (MACM) for steel surface defect classification based on the CBAM (Convolutional Block Attention Module) hybrid attention architecture. Firstly, heuristic channel attention is achieved through gray correlation analysis to enhance the interpretability of the algorithm and overcome the impact of interlayer information loss on channel attention determination. Secondly, compact bilinear pooling is used to achieve spatial feature fusion, enhancing the capture of nonlinear complex interaction information between abstract features. Finally, the combination of channel and spatial attention is achieved based on the CBAM architecture. The experiments show that the MACM method not only performs well but is also lighter than existing state-of-the-art methods, validating the effectiveness of the MACM algorithm in improving classification accuracy and reducing model complexity.

    Tools

    Get Citation

    Copy Citation Text

    YING Hejie, LAI Lianfeng, REN Xuehang, XIONG Lingling, XUE Zhangqi. Classification of Steel Surface Defects Based on Multi-Attention Mechanism[J]. Semiconductor Optoelectronics, 2025, 46(1): 90

    Download Citation

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

    Category:

    Received: Jul. 17, 2024

    Accepted: Sep. 18, 2025

    Published Online: Sep. 18, 2025

    The Author Email:

    DOI:10.16818/j.issn1001-5868.20240717001

    Topics