A lightweight algorithm for steel surface defect detection using improved YOLOv8

Abstract

In response to the issues of low precision, a large number of parameters and high model complexity in steel surface defect detection, a lightweight algorithm using improved YOLOv8 is proposed. Firstly, GhostNet is utilized as the backbone network in order to reduce the number of model parameters and computational complexity. Secondly, the MPCA (MultiPath Coordinate Attention) attention mechanism is integrated to enhance feature extraction capabilities. Finally, the SIoU (Simplified IoU ) is used to replace the traditional CIoU loss function, which can make the anchor frame more fast and accurate in the regression process, to improve the stability and the robustness of detection. The experimental results indicate that these enhancements have led to a reduction of 37% in calculation amount for the improved YOLOv8n algorithm, a decrease of 32% in parameter count, and an increase in average detection accuracy ( mAP ) by 1.2%. This model achieves a balance between lightweighting and detection accuracy while providing a viable solution for deployment in computationally resource-constrained edge computing environments such as embedded systems and mobile devices.