Research on the Detection of Steel Plate Defects Based on SimAM and Twin-NMF Transfer

Yongqiang Zou, Guanghui Zhang () and Yugang Fan
Additional contact information
Yongqiang Zou: Faculty of Information Engineering and Automation, Kunming University of Science and Technology, Kunming 650500, China
Guanghui Zhang: Faculty of Information Engineering and Automation, Kunming University of Science and Technology, Kunming 650500, China
Yugang Fan: Faculty of Information Engineering and Automation, Kunming University of Science and Technology, Kunming 650500, China

Mathematics, 2024, vol. 12, issue 17, 1-18

Abstract: Pulsed eddy current thermography can detect surface or subsurface defects in steel, but in the process of combining deep learning, it is expensive and inefficient to build a complete sample of defects due to the complexity of the actual industrial environment. Consequently, this study proposes a transfer learning method based on Twin-NMF and combines it with the SimAM attention mechanism to enhance the detection accuracy of the target domain task. First, to address the domain differences between the target domain task and the source domain samples, this study introduces a Twin-NMF transfer method. This approach reconstructs the feature space of both the source and target domains using twin non-negative matrix factorization and employs cosine similarity to measure the correlation between the features of these two domains. Secondly, this study integrates a parameter-free SimAM into the neck of the YOLOv8 model to enhance its capabilities in extracting and classifying steel surface defects, as well as to alleviate the precision collapse phenomenon associated with multi-scale defect recognition. The experimental results show that the proposed Twin-NMF model with SimAM improves the detection accuracy of steel surface defects. Taking NEU-DET and GC10-DET as source domains, respectively, in the ECTI dataset, mAP@0.5 reaches 99.3% and 99.2%, and the detection accuracy reaches 98% and 98.5%.

Keywords: defect detection; NMF; transfer learning; eddy current thermal imaging; YOLOv8 (search for similar items in EconPapers)
JEL-codes: C (search for similar items in EconPapers)
Date: 2024
References: View complete reference list from CitEc
Citations:

Downloads: (external link)
https://www.mdpi.com/2227-7390/12/17/2782/pdf (application/pdf)
https://www.mdpi.com/2227-7390/12/17/2782/ (text/html)

Related works:
This item may be available elsewhere in EconPapers: Search for items with the same title.

Export reference: BibTeX RIS (EndNote, ProCite, RefMan) HTML/Text

Persistent link: https://EconPapers.repec.org/RePEc:gam:jmathe:v:12:y:2024:i:17:p:2782-:d:1473942

Access Statistics for this article

Mathematics is currently edited by Ms. Emma He

More articles in Mathematics from MDPI
Bibliographic data for series maintained by MDPI Indexing Manager ().