MWYOLO: A Mamba-Enhanced Lightweight YOLO Framework with Multi-Frequency Attention for Industrial Surface Defect Detection
Industrial surface defect detection constitutes a fundamental component in automated quality inspection but remains challenging due to complex textures, diverse defect scales, and stringent real-time constraints. To address these issues, we present MWYOLO, an enhanced YOLO11-based detection framework tailored for accurate and efficient industrial inspection. First, a C3k2-Mamba Spatial Fusion Block (C3k2-MSFB) that integrates global contextual information with local structural cues via state-space modeling, enabling more discriminative representations of fine-grained texture variations. Second, a multi-scale wavelet attention (MWA) module is embedded into the backbone, leveraging wavelet-domain feature decomposition and dual attention to capture multi-frequency patterns, thereby improving sensitivity to fine-grained and subtle defect patterns. Third, an Inner-CIoU loss is developed to emphasize interior geometric alignment during bounding-box regression, offering more stable optimization for ambiguous or low-contrast targets. Extensive experiments conducted on three representative industrial datasets—NEU-DET, HRIPCB, and a self-constructed GSD dataset—demonstrate the effectiveness of MWYOLO. The model achieves mAP50 scores of 81.4%, 98.1%, and 67.7%, respectively, while maintaining a lightweight design with only 3.2M parameters and 7.3 GFLOPs. The results validate MWYOLO as a robust and computationally efficient solution, offering a favorable balance between accuracy, interpretability, and deployability for real-world industrial defect detection tasks.
Authors
- Haidong Xiao
- Junjian Chen (ORCID: https://orcid.org/0009-0000-6358-4937)
- Zongze Wu
- Zhigang Ren (ORCID: https://orcid.org/0000-0002-0913-0130)
Institutions
- Guangdong University of Technology (CN)
- Shenzhen University (CN)
- Guangdong Institute of Intelligent Manufacturing (CN)
Publication Details
- Journal
- AI for Engineering
- Published
- 2026-05-11
- DOI
- https://doi.org/10.3390/aieng1010003
- Primary Topic
- Industrial Vision Systems and Defect Detection
- Type
- article
- Field-Weighted Citation Impact
- 0.00
Funders
- Basic and Applied Basic Research Foundation of Guangdong Province