Surface Defect Detection: An Approach Utilising Feature-Based Transfer Learning

Junqing Yang, Chengzhangzheng Wu, Taimingwang Liu, Muhammad Ateeq, Hadyan Hafizh, Ahmad Fakhri Ab. Nasir, Anwar P.P. Abdul Majeed*

*Corresponding author for this work

Research output: Chapter in Book or Report/Conference proceedingConference Proceedingpeer-review

Abstract

Surface defects in manufacturing processes pose significant challenges, affecting product quality and safety. Traditional labour-based inspection is deemed to be ineffective and has led the shift to computer vision-based solutions and to a certain extent, the employment of artificial intelligence. In the present study, we leverage the capability of a pre-trained convolutional neural networks model, i.e. VGG19, in extracting the features from a set of surface defect dataset that comprises six unique defect categories. The ability of different machine learning models, namely Logistic Regression (LR), Random Forest (RF), k-Nearest Neighbour (kNN) and Support Vector Machine (SVM), to classify the defects was investigated. It was demonstrated from the study that the VGG-19 + LR combination is the optimal pipeline. This study suggests that the feature-based transfer learning approach is an attractive approach to be employed for surface defect detection.

Original languageEnglish
Title of host publicationRobot Intelligence Technology and Applications 8 - Results from the 11th International Conference on Robot Intelligence Technology and Applications
EditorsAnwar P. P. Abdul Majeed, Eng Hwa Yap, Pengcheng Liu, Xiaowei Huang, Anh Nguyen, Wei Chen, Ue-Hwan Kim
PublisherSpringer Science and Business Media Deutschland GmbH
Pages58-65
Number of pages8
ISBN (Print)9783031706868
DOIs
Publication statusPublished - 2024
Event11th International Conference on Robot Intelligence Technology and Applications, RiTA 2023 - Taicang, China
Duration: 6 Dec 20238 Dec 2023

Publication series

NameLecture Notes in Networks and Systems
Volume1133 LNNS
ISSN (Print)2367-3370
ISSN (Electronic)2367-3389

Conference

Conference11th International Conference on Robot Intelligence Technology and Applications, RiTA 2023
Country/TerritoryChina
CityTaicang
Period6/12/238/12/23

Keywords

  • Deep learning
  • Feature extraction
  • Surface defect detection
  • Transfer learning

Fingerprint

Dive into the research topics of 'Surface Defect Detection: An Approach Utilising Feature-Based Transfer Learning'. Together they form a unique fingerprint.

Cite this