A high-precision segmentation network for industrial surface defect detection
Source abstract
Abstract
Accurate surface defect detection is essential for improving product quality and reducing manufacturing costs, particularly in high-precision industries. However, existing deep learning methods struggle with multi-scale feature fusion and spatial information preservation. To address these challenges, this paper proposes the Spatial-Positional and Cross-Scale Fusion Network (SPCS-Net), an advanced segmentation network incorporating the Spatial-Positional Attention Module (SPAM) and the Cross-Scale Attention Fusion Module (CSAFM). SPAM integrates multi-scale convolutions, spatial attention mechanisms, and positional encoding to enhance the perception of defects with varying shapes. This design helps mitigate spatial information loss commonly observed in traditional U-Net models. CSAFM optimizes multi-scale feature fusion in the decoding stage by employing asymmetric convolutions and an adaptive feature weighting mechanism, effectively bridging the gap between high-level semantic information and low-level spatial details. Experimental results demonstrate that SPCS-Net outperforms state-of-the-art models on the NEU-Seg, MBP-Seg, and USB-Seg datasets, achieving F1 scores of 0.8854, 0.6543, and 0.7793, respectively, with Mean Intersection over Union scores of 0.7964, 0.5654, and 0.6674. Furthermore, SPCS-Net attains an inference speed of 104 f/s on MBP-Seg, striking a balance between accuracy and efficiency. These results highlight SPCS-Net as a promising solution for surface defect segmentation, with potential applications in automated quality inspection and material science.
Source: crossref · Quality: baseline
Research context
OpenAlex topics: Industrial Vision Systems and Defect Detection · Surface Roughness and Optical Measurements · Optical measurement and interference techniques
SciVX discovery tags are terminology-based. OpenAlex topics are source-provided classifications.
Affiliations
School of Information Engineering, Nantong Institute of Technology 1 , Nantong, Jiangsu,
Division of Information and Communication Convergence Engineering, Mokwon University 2 , Daejeon,
References (44)
Show source-provided references
1. A convolutional neural network-based method for workpiece surface defect detection10.1016/j.measurement.2021.109185 ↗
2. Automated surface defect detection framework using machine vision and convolutional neural networks10.1007/s10845-021-01878-w ↗
3. U-net: Convolutional networks for biomedical image segmentation,
4. Medical image segmentation review: The success of U-Net10.1109/TPAMI.2024.3435571 ↗
5. DSIA U-net: Deep shallow interaction with attention mechanism UNet for remote sensing satellite images10.1038/s41598-024-84134-4 ↗
6. Online detection and evaluation of weld surface defects based on lightweight network VGG16-UNet and laser scanning10.1016/j.jmapro.2024.08.037 ↗
7. O. Oktay , J.Schlemper, L. L.Folgoc, M.Lee, M.Heinrich, K.Misawa, K.Mori, S.McDonagh, N.Y Hammerla, B.Kainz, B.Glocker, and D.Rueckert, “Attention u-net: Learning where to look for the pancreas,” arXiv:1804.03999 (2018).
8. Surface defect detection of CFRP materials based on infrared thermography and attention U-Net algorithm10.1080/10589759.2023.2191954 ↗
9. UNet 3+: A full-scale connected UNet for medical image segmentation
10. Intelligent identification of oceanic eddies in remote sensing data via dual-pyramid UNet10.1016/j.aosl.2023.100335 ↗
11. Enhanced damage segmentation in RC components using pyramid Haar wavelet downsampling and attention U-net10.1016/j.autcon.2024.105746 ↗
12. Automated surface defect detection in metals: A comparative review of object detection and semantic segmentation using deep learning10.1109/TIA.2022.3151560 ↗
13. A review of deep learning-based approaches for defect detection in smart manufacturing10.1007/s12596-023-01340-5 ↗
14. A systematic review on deep learning with CNNs applied to surface defect detection10.3390/jimaging9100193 ↗
15. Automatic defect detection and segmentation of tunnel surface using modified Mask R-CNN10.1016/j.measurement.2021.109316 ↗
16. A systematic review of deep learning approaches for surface defect detection in industrial applications10.1016/j.engappai.2023.107717 ↗
17. Research of U-Net-based CNN architectures for metal surface defect detection10.3390/machines10050327 ↗
18. Research on segmentation of steel surface defect images based on improved Res-UNet network10.11999/JEIT211350 ↗
19. Fabric defect detection based on separate convolutional UNet10.1007/s11042-022-13568-7 ↗
20. Detection of scratch defects on metal surfaces based on MSDD-UNet10.3390/electronics13163241 ↗
21. AnomalySeg: Deep learning-based fast anomaly segmentation approach for surface defect detection10.3390/electronics13020284 ↗
22. Soldering defect segmentation method for PCB on improved UNet10.3390/app14167370 ↗
23. RHNet: A real-time high-resolution network for surface defect detection
24. Deep learning-based automated steel surface defect segmentation: A comparative experimental study10.1007/s11042-023-15307-y ↗
25. Resformer-Unet: A U-shaped framework combining ResNet and transformer for segmentation of strip steel surface defects10.2355/isijinternational.ISIJINT-2023-222 ↗
26. Rail surface defect detection using a transformer-based network10.1016/j.jii.2024.100584 ↗
27. LGGFormer: A dual-branch local-guided global self-attention network for surface defect segmentation10.1016/j.aei.2024.103099 ↗
28. Cross-supervised contrastive learning domain adaptation network for steel defect segmentation10.1016/j.aei.2024.102964 ↗
29. J. Chen , Y.Lu, Q.Yu, X.Luo, E.Adeli, Y.Wang, L.Lu, A. L.Yuille, and Y.Zhou, “TransUNet: Transformers make strong encoders for medical image segmentation,” arXiv:2102.04306 (2021).
30. J. Ruan and S.Xiang, “VM-UNet: Vision mamba UNet for medical image segmentation,” arXiv:2402.02491 (2024).
31. RS3Mamba: Visual state space model for remote sensing image semantic segmentation10.1109/LGRS.2024.3414293 ↗
32. DiffDD: A surface defect detection framework with diffusion probabilistic model10.1016/j.aei.2024.102637 ↗
33. MA-Net: A multi-scale attention network for liver and tumor segmentation10.1109/ACCESS.2020.3025372 ↗
34. A Unet-inspired spatial-attention transformer model for segmenting gear tooth surface defects10.1016/j.aei.2024.102933 ↗
35. Pyramid cross attention network for pixel-wise surface defect detection10.1016/j.ndteint.2024.103053 ↗
36. An adaptive image segmentation network for surface defect detection10.1109/TNNLS.2022.3230426 ↗
37. A novel multi-scale cross-patch attention with dilated convolution (MCPAD-UNET) for metallic surface defect detection10.1007/s11760-023-02745-2 ↗
38. A deep learning-based surface defect inspection system using multiscale and channel-compressed features10.1109/TIM.2020.2986875 ↗
39. Segmentation-assisted classification model with convolutional neural network for weld defect detection10.1016/j.advengsoft.2024.103788 ↗
40. GAF-Net: A new automated segmentation method based on multiscale feature fusion and feedback module10.1016/j.patrec.2024.11.025 ↗
41. A sub-region Unet for weak defects segmentation with global information and mask-aware loss10.1016/j.engappai.2023.106011 ↗
42. An adaptive feature refinement network for pixel-level segmentation of surface defect10.1088/1361-6501/ad9cb0 ↗
43. WFF-Net: Trainable weight feature fusion convolutional neural networks for surface defect detection10.1016/j.aei.2024.103073 ↗
44. PGA-Net: Pyramid feature fusion and global context attention network for automated surface defect detection10.1109/TII.2019.2958826 ↗