WaveMamba-YOLO: Combining frequency awareness and state-space modeling for defect localization

Aping Ge, Yang Lv and Jun Huang

PLOS ONE, 2026, vol. 21, issue 3, 1-24

Abstract: Steel surface defect detection is critical for ensuring the reliability and safety of automotive manufacturing. However, existing methods often suffer from high computational cost, weak sensitivity to fine textures, and limited adaptability to diverse defect scales. To address these challenges, we propose WaveMamba-YOLO, a real-time detection framework that integrates frequency-domain enhancement with efficient state-space modeling. The architecture introduces three key modules: (1) CHDWT, which combines Haar wavelet decomposition and residual learning to preserve structural details during downsampling; (2) GLaM, a global-local-aware Mamba module that couples large-kernel convolution with state-space modeling to capture long-range dependencies at linear complexity; and (3) LWGA, a lightweight group attention mechanism that adaptively attends to micro-, regular-, medium-, and large-scale defects. Experiments on the Severstal Steel Defect and NEU-DET datasets demonstrate that WaveMamba-YOLO achieves superior performance, reaching 51.70% mAP@0.5 and 58.60% precision on Severstal and 77.70% mAP@0.5 on NEU-DET, consistently surpassing mainstream lightweight detectors. These results confirm the effectiveness of WaveMamba-YOLO in balancing detection accuracy and efficiency, highlighting its potential for real-time industrial inspection.

Date: 2026
References: Add references at CitEc
Citations:

Downloads: (external link)
https://journals.plos.org/plosone/article?id=10.1371/journal.pone.0344940 (text/html)
https://journals.plos.org/plosone/article/file?id= ... 44940&type=printable (application/pdf)

Related works:
This item may be available elsewhere in EconPapers: Search for items with the same title.

Export reference: BibTeX RIS (EndNote, ProCite, RefMan) HTML/Text

Persistent link: https://EconPapers.repec.org/RePEc:plo:pone00:0344940

DOI: 10.1371/journal.pone.0344940

Access Statistics for this article

More articles in PLOS ONE from Public Library of Science
Bibliographic data for series maintained by plosone ().