Steel surface defect classification approach using an All-optical Neuron-based SNN with attention mechanism

Liang Gong,Hang Dong,Xin Cheng,Zhenghui Ge,Liangchao Guo
DOI: https://doi.org/10.1108/ijicc-02-2023-0034
2023-06-15
International Journal of Intelligent Computing and Cybernetics
Abstract:Purpose The purpose of this study is to propose a new method for the end-to-end classification of steel surface defects. Design/methodology/approach This study proposes an AM-AoN-SNN algorithm, which combines an attention mechanism (AM) with an All-optical Neuron-based spiking neural network (AoN-SNN). The AM enhances network learning and extracts defective features, while the AoN-SNN predicts both the labels of the defects and the final labels of the images. Compared to the conventional Leaky-Integrated and Fire SNN, the AoN-SNN has improved the activation of neurons. Findings The experimental findings on Northeast University (NEU)-CLS demonstrate that the proposed neural network detection approach outperforms other methods. Furthermore, the network's effectiveness was tested, and the results indicate that the proposed method can achieve high detection accuracy and strong anti-interference capabilities while maintaining a basic structure. Originality/value This study introduces a novel approach to classifying steel surface defects using a combination of a shallow AoN-SNN and a hybrid AM with different network architectures. The proposed method is the first study of SNN networks applied to this task.
What problem does this paper attempt to address?