Paper ID | MLR-APPL-IVASR-4.4 | ||
Paper Title | S2D2NET: AN IMPROVED APPROACH FOR ROBUST STEEL SURFACE DEFECTS DIAGNOSIS WITH SMALL SAMPLE LEARNING | ||
Authors | Vikanksh Nath, Chiranjoy Chattopadhyay, Indian Institute of Technology Jodhpur, India | ||
Session | MLR-APPL-IVASR-4: Machine learning for image and video analysis, synthesis, and retrieval 4 | ||
Location | Area B | ||
Session Time: | Tuesday, 21 September, 13:30 - 15:00 | ||
Presentation Time: | Tuesday, 21 September, 13:30 - 15:00 | ||
Presentation | Poster | ||
Topic | Applications of Machine Learning: Machine learning for image & video analysis, synthesis, and retrieval | ||
IEEE Xplore Open Preview | Click here to view in IEEE Xplore | ||
Abstract | Surface defect recognition of products is a necessary process to guarantee the quality of industrial production. This paper proposes a hybrid model, S2D2Net (Steel Surface Defect Diagnosis Network), for an efficient and robust inspection of the steel surface during the manufacturing process. The S2D2Net uses a pre-trained ImageNet model as a feature extractor and learns a Capsule Network over the extracted features. The experimental results on a publicly available steel surface defect dataset (NEU) show that S2D2Net achieved 99.17% accuracy with minimal training data and improved by 9.59% over its closest competitor based on GAN. S2D2Net proved its robustness by achieving 94.7% accuracy on a diversity enhanced dataset, ENEU, and improved by 3.6% over its closest competitor. It has better, robust recognition performance compared to other state-of-the-art DNN-based detectors. |