Journal Article

Two-Stream Network One-Class Classification Model for Defect Inspections

Luo, Chenglong, Lee, Seunghun, Lee, Sungkwan, Jung, Hoeryong

2023

Overview

Defect inspection is important to ensure consistent quality and efficiency in industrial manufacturing. Recently, machine vision systems integrating artificial intelligence (AI)-based inspection algorithms have exhibited promising performance in various applications, but practically, they often suffer from data imbalance. This paper proposes a defect inspection method using a one-class classification (OCC) model to deal with imbalanced datasets. A two-stream network architecture consisting of global and local feature extractor networks is presented, which can alleviate the representation collapse problem of OCC. By combining an object-oriented invariant feature vector with a training-data-oriented local feature vector, the proposed two-stream network model prevents the decision boundary from collapsing to the training dataset and obtains an appropriate decision boundary. The performance of the proposed model is demonstrated in the practical application of automotive-airbag bracket-welding defect inspection. The effects of the classification layer and two-stream network architecture on the overall inspection accuracy were clarified by using image samples collected in a controlled laboratory environment and from a production site. The results are compared with those of a previous classification model, demonstrating that the proposed model can improve the accuracy, precision, and F1 score by up to 8.19%, 10.74%, and 4.02%, respectively.

• Publisher: MDPI AG,MDPI
• Subject: Accuracy / Air bags / Algorithms / Analysis / Artificial Intelligence / Classification / Datasets / Deep learning / defect inspection / Defective products / Defects / Human error / Machine vision / Manufacturing / Neural networks / one-class classification / Rivers / Steel pipes / Support vector machines / two-stream network / Vision systems