Mean shift based prototypical network for steel surface anomaly recognition


TAŞTİMUR TEMİZ C.

Scientific Reports, cilt.15, sa.1, 2025 (SCI-Expanded, Scopus) identifier identifier

  • Yayın Türü: Makale / Tam Makale
  • Cilt numarası: 15 Sayı: 1
  • Basım Tarihi: 2025
  • Doi Numarası: 10.1038/s41598-025-31529-6
  • Dergi Adı: Scientific Reports
  • Derginin Tarandığı İndeksler: Science Citation Index Expanded (SCI-EXPANDED), Scopus, BIOSIS, Chemical Abstracts Core, MEDLINE, Directory of Open Access Journals
  • Erzincan Binali Yıldırım Üniversitesi Adresli: Evet

Özet

Steel production plays a crucial role in the global economy, and surface defects in hot-rolled steel sheets significantly affect product quality and consumer acceptance. This paper introduces Mean Shift based Prototypical network (MSPro-Net), a novel and effective Prototypical network based on mean drift for classifying surface defects early in the manufacturing process. Key contributions of this study include the introduction of MSPro-Net’s adaptive prototype computation, which uses the mean-shift method to improve defect detection. Unlike the classical Prototypical network (CL-ProNet), MSPro-Net produces a prototype that better represents all examples within a class, leading to significantly higher accuracy. Additionally, MSPro-Net is highly effective in few-shot learning scenarios, achieving excellent results with limited training data. Extensive experimental evaluations demonstrate that MSPro-Net consistently outperforms CL-ProNet in multiple N-way K-shot scenarios. For example, on the 6-way 25-shot NEU dataset, MSPro-Net achieves 98.67% accuracy com-pared to 50.67% for the classical network. Similar improvements are observed on the 7-way 25-shot XSDD and 10-way 25-shot GC10-Det datasets, with MSPro-Net achieving 96.00% and 90.00% accuracy, respectively, far exceeding the classical network’s performance. These results highlight the superior performance of MSPro-Net across both small and large datasets.