EMamba: A MoE-enhanced state space network for robust steel surface defect detection
- 1Technological Institute of Materials & Energy Science, Xijing University, Xi’an, China
Abstract
Steel surface defect detection is a critical component of intelligent manufacturing and industrial visual quality control. However, due to large variations in defect scale, elongated morphologies, strong directional continuity, and high inter-class visual similarity, existing detection methods struggle to achieve a desirable balance among global structural modeling, semantically adaptive representation, and computational efficiency. To address these challenges, this paper proposes an efficient steel surface defect detection framework, termed EMamba. By integrating state space modeling, conditional expert computation, and a hybrid multi-branch architectural design, the proposed framework systematically enhances detection performance in complex industrial scenarios. Specifically, we introduce a two-dimensional state space modeling mechanism and propose a Convolutional State Space Block (CSSBlock), which models long-range spatial dependencies and directional continuity with linear computational complexity. At the high-level semantic stage of the backbone network, an Efficient Sparse Mixture-of-Experts module (ES-MoE) is incorporated. Through a conditional Top-k routing mechanism, ES-MoE enhances the diversity and discriminability of feature representations. Furthermore, we propose a Mixed Star Network (MSN), which selectively configures high-order feature modeling modules in a hierarchy-aware manner within a multi-branch structure, achieving a fine-grained trade-off between representational capacity and computational efficiency. Building upon these components, a decoupled detection head is adopted to construct the unified EMamba detection framework. Extensive experiments on multiple steel surface defect datasets demonstrate that EMamba achieves competitive detection accuracy, structural defect perception capability, and inference efficiency compared with representative existing methods. These results indicate the effectiveness and practicality of the proposed hybrid modeling paradigm for steel surface defect detection.
Related
- YOLO-CBNet: A robust attention-enhanced detection framework for underwater fish recognition in aquaculture environments
- ACO-Enhanced DCNN-BiLSTM framework for intrusion detection in smart Consumer Electronics network
- CuFormer U-Net: a transformer-enhanced multi-task network for cucumber leaf disease detection and segmentation
- DINOCT: robust skin surface localization for robotic noncontact dynamic optical coherence elastography
- Hybrid local–global vision transformer framework for robust plant disease detection in precision agriculture
BibTeX
@article{yang2026emamba,
title = {{EMamba: A MoE-enhanced state space network for robust steel surface defect detection}},
author = {Haiyan Yang and Peng Wang and Yangyang Liu},
journal = {PLOS One},
year = {2026},
month = sep,
doi = {10.1371/journal.pone.0358866},
url = {https://doi.org/10.1371/journal.pone.0358866},
note = {CC BY 4.0},
}
RIS
TY - JOUR
TI - EMamba: A MoE-enhanced state space network for robust steel surface defect detection
AU - Haiyan Yang
AU - Peng Wang
AU - Yangyang Liu
PY - 2026
DA - 2026/09/25
JO - PLOS One
DO - 10.1371/journal.pone.0358866
UR - https://doi.org/10.1371/journal.pone.0358866
AB - Steel surface defect detection is a critical component of intelligent manufacturing and industrial visual quality control. However, due to large variations in defect scale, elongated morphologies, strong directional continuity, and high inter-class visual similarity, existing detection methods struggle to achieve a desirable balance among global structural modeling, semantically adaptive representation, and computational efficiency. To address these challenges, this paper proposes an efficient steel surface defect detection framework, termed EMamba. By integrating state space modeling, conditional expert computation, and a hybrid multi-branch architectural design, the proposed framework systematically enhances detection performance in complex industrial scenarios. Specifically, we introduce a two-dimensional state space modeling mechanism and propose a Convolutional State Space Block (CSSBlock), which models long-range spatial dependencies and directional continuity with linear computational complexity. At the high-level semantic stage of the backbone network, an Efficient Sparse Mixture-of-Experts module (ES-MoE) is incorporated. Through a conditional Top-k routing mechanism, ES-MoE enhances the diversity and discriminability of feature representations. Furthermore, we propose a Mixed Star Network (MSN), which selectively configures high-order feature modeling modules in a hierarchy-aware manner within a multi-branch structure, achieving a fine-grained trade-off between representational capacity and computational efficiency. Building upon these components, a decoupled detection head is adopted to construct the unified EMamba detection framework. Extensive experiments on multiple steel surface defect datasets demonstrate that EMamba achieves competitive detection accuracy, structural defect perception capability, and inference efficiency compared with representative existing methods. These results indicate the effectiveness and practicality of the proposed hybrid modeling paradigm for steel surface defect detection.
SN - 1932-6203
ER -