About the Application of Autoencoders For Visual Defect Detection

Richárd Rádli,László Czúni
DOI: https://doi.org/10.24132/csrn.2021.3002.20
2021-01-01
Abstract:Visual defect detection is a key technology in modern industrial manufacturing systems. There are many possibleappearances of product defects, including distortions in color, shape, contamination, missing or superfluous parts.For the detection of those, besides traditional image processing techniques, convolutional neural networks basedmethods have also appeared to avoid the usage of hand-crafted features and to build more efficient detectionmechanisms. In our article we deal with autoencoder convolutional networks (AEs) which do not require examplesof defects for training. Unfortunately, the manual and/or trial-and-error design of AEs is still required to achievegood performance, since there are many unknown parameters of AEs which can greatly influence the detectionabilities. For our study we have chosen a well performing AE known as structural similarity AE (SSIM-AE),where the loss function and the comparison of the output with the input is implemented via the SSIM instead ofthe often used L1 or L2 norms. Investigating the performance of SSIM-AE on different data-sets, we found that itsperformance can be improved with modified convolutional structures without modifying the size of latent space.We also show that finding a model with low reconstruction error during training does not mean good detectionabilities and denoising AEs can increase efficiency.
What problem does this paper attempt to address?