Lightweight CNN Models for Product Defect Detection with Edge Computing in Manufacturing Industries

Abstract

Detecting product defects is one of the manufacturing industry's most essential processes in quality control. Human visual inspection for product defects is the traditional method employed in the industry. Nevertheless, it can be laborious, prone to human mistakes, and unreliable. Deep Learning-based Convolution Neural Networks (CNN) has been extensively used in fully automating product defect detection systems. However, real-time edge devices installed at the manufacturing site generally have limited computing capability and cannot run different CNN models. A lightweight CNN model is adopted in this scenario to find a balance between defect detection, model training time, memory consumption, computing time and efficiency. This work provides lightweight CNN models with transfer learning for product defect detection on fabric, surface, and casting datasets. We deployed the trained model to the NVIDIA Jetson Nano-kit edge device for detection speed with better simulation results in terms of accuracy, sensitivity rate, specificity rate, and F1 measure in the workplace context of the Manufacturing Industries.