A Lightweight Group Multiscale Bidirectional Interactive Network for Real-Time Steel Surface Defect Detection

TL;DR

This paper introduces GMBINet, a lightweight, multiscale bidirectional network that achieves real-time steel surface defect detection with high accuracy and low computational cost, suitable for resource-limited industrial environments.

Contribution

The paper proposes GMBINet, a novel group multiscale bidirectional interactive framework that enhances multiscale feature extraction and interaction while maintaining low computational complexity.

Findings

Achieves 1048 FPS on GPU and 16.53 FPS on CPU at 512 resolution.

Uses only 0.19 million parameters, demonstrating efficiency.

Shows strong generalization on multiple defect datasets.

Abstract

Real-time surface defect detection is critical for maintaining product quality and production efficiency in the steel manufacturing industry. Despite promising accuracy, existing deep learning methods often suffer from high computational complexity and slow inference speeds, which limit their deployment in resource-constrained industrial environments. Recent lightweight approaches adopt multibranch architectures based on depthwise separable convolution (DSConv) to capture multiscale contextual information. However, these methods often suffer from increased computational overhead and lack effective cross-scale feature interaction, limiting their ability to fully leverage multiscale representations. To address these challenges, we propose GMBINet, a lightweight framework that enhances multiscale feature extraction and interaction through novel Group Multiscale Bidirectional Interactive (GMBI) modules. The GMBI adopts a group-wise strategy for multiscale feature extraction, ensuring scale-agnostic computational complexity. It further integrates a Bidirectional Progressive Feature Interactor (BPFI) and a parameter-free Element-Wise Multiplication-Summation (EWMS) operation to enhance cross-scale interaction without introducing additional computational overhead. Experiments on SD-Saliency-900 and NRSD-MN datasets demonstrate that GMBINet delivers competitive accuracy with real-time speeds of 1048 FPS on GPU and 16.53 FPS on CPU at 512 resolution, using only 0.19 M parameters. Additional evaluations on the NEU-CLS defect classification dataset further confirm the strong generalization ability of our method, demonstrating its potential for broader industrial vision applications beyond surface defect detection. The dataset and code are publicly available at: https://github.com/zhangyongcode/GMBINet.