Hybrid CNN–Transformer–GRU framework with improved residual shrinkage network and DCCS optimisation for steel surface defect classification

View at publisher
Authors: Brajesh Kumar, Sunil Kumar Singh, Sujit Kumar, Bam Bahadur Sinha
Cited by: 1
References: 48

Abstract

Automated inspection of steel surface defects is an important task in intelligent manufacturing systems to ensure product quality and reduce manual inspection efforts. This study proposes a hybrid deep learning framework for accurate classification of metal surface defects by integrating convolutional feature extraction, residual shrinkage learning, transformer-based attention modelling, and recurrent sequence analysis. Initially, greyscale defect images from the NEU metal surface defect dataset are processed using a convolutional neural network (CNN) combined with Batch Normalisation and Improved Residual Shrinkage Network (IRSN) blocks to extract noise-resilient and discriminative local features. The IRSN mechanism performs adaptive soft thresholding to suppress noise and enhance relevant defect patterns. Transformer encoder with multi-head attention learns long-range relationships among extracted features. In parallel, a Gated Recurrent Unit (GRU) branch models sequential dependencies in the feature representation. The outputs from both branches are fused and passed through fully connected layers for multi-class defect classification. A Dual-Condition Cost-Sensitive (DCCS) optimisation strategy is applied to automatically tune hyperparameters. The proposed model attains an accuracy of 93.06% and 98.40% on the NEU metal surface defect dataset and Severstal Steel Defect Detection dataset, respectively. The proposed model underscores an improvement of around 0.06% and 1.2% over the recent benchmark model.

Fields of Study & Subjects

Non-Destructive Testing Techniques Advanced Neural Network Applications Ultrasonics and Acoustic Wave Propagation

Publication details

Journal / source
Australian Journal of Mechanical Engineering
Publisher
Informa UK Limited
Pages
1–26
Publication date
20 May 2026
Language
EN