Surface Defect Detection Algorithm for Workpieces Based on Improved YOLOv8

Read full text
Authors: Da An, Ng Kok Why, Fangfang Chua
Cited by: 1
References: 35

Abstract

Industrial surface defect detection is crucial for quality control in manufacturing, yet remains challenging due to the small scale, low contrast, and texture variability of defects. While YOLOv8n offers high inference speed and efficiency, its accuracy is limited by insufficient feature representation and inadequate data diversity. This paper proposes a detection framework integrating Channel–Spatial Modulation Attention (CASM) and Small-Scale Grid Texture Shuffling Augmentation (SG-TSA) into YOLOv8n to improve detection performance without sacrificing efficiency. CASM introduces a parallel channel–spatial attention structure with adaptive fusion to better capture fine-grained defect features, while SG-TSA increases sample diversity by introducing realistic texture perturbations within defect regions. Experiments on the NEU-DET dataset show that our method improves mAP@0.5:0.95 by 3.01% and mAP@0.5 by 2.84% over baseline YOLOv8n. These results highlight the importance of architecture-specific optimization for lightweight detectors in industrial scenarios.

Fields of Study & Subjects

Advanced Neural Network Applications Industrial Vision Systems and Defect Detection Advanced Data and IoT Technologies

Publication details

Journal / source
Automation
Publisher
MDPI AG
Volume
7
Issue
1
Pages
32–32
Publication date
12 February 2026
Language
EN