Accurate detection of defects on steel surfaces is crucial for maintaining quality standards in steel production. This paper addresses the challenge of classifying steel sheets into distinct defect categories by presenting a robust method that leverages deep learning and advanced optimization techniques. We propose a novel approach that utilizes the ResNet101 model to extract deep features, which are then classified using a support vector machine (SVM). To enhance the SVM's performance, Bayesian optimization is employed for hyperparameter tuning. Our method is validated using the "Severstal: Steel Defect Detection" dataset from Kaggle, achieving a validation accuracy of 89.1% and a test accuracy of 90.6%, with a classification error of 0.10934. Additionally, the area under the curve (AUC) for each class exceeds 0.95 in both the validation and test sets, demonstrating excellent discriminatory power. Further evaluation on the DAGM dataset achieved flawless results, with an accuracy of 100%, AUC of 1, sensitivity of 100%, specificity of 100%, precision of 100%, MCC of 100%, F1 score of 1, and kappa of 100%. On the NEU dataset, our method achieved an accuracy of 97.92%, sensitivity of 97.92%, specificity of 99.58%, precision of 98.06%, F1 score of 0.9791, MCC of 97.55%, and kappa of 92.50%. These results demonstrate the robustness and adaptability of the proposed method, offering an efficient and reliable solution for automating steel defect detection and surface defect classification in industrial applications.
Read full abstract