Deep Learning based MURA Defect Detection

Abstract

INTRODUCTION: MURA defects in LED/LCD panels are one of the most challenging defects for Automatic Defect Classification and Localization (ADC) due to their extremely low contrast when compared with the background. Manual detection is subjective, error prone, very tedious and time consuming. Even when the type of MURA defects can be ascertained manually, the exact bounding box for defect is hard to determine. Various heuristic based image processing techniques have been applied giving sub-optimal accuracy over generic datasets. OBJECTIVES: The primary objective of this paper is to check whether the state of the art DL (Deep Learning) network for general object classification and localization (MSCOCO PASCAL VOC etc.) can be applied successfully for MURA Defect Classification and Localization. METHODS: In this paper we present a single DL pipeline for classification and localization which for the first time is applied for MURA defects. Naive DL network - Single Shot multi-box Detector (SSD, pre-trained on ImageNet) was not sufficient to give a good F1 score because of the nature of the defect. Accuracy improved a little after applying various DL specific optimization methods such as loss function optimization, network optimization etc. Utilizing the knowledge from MURA domain for data augmentation, like filtering based on image capture wavelength etc. improved the results significantly. RESULTS: Using optimization techniques that are from both DL domain as well as specific to MURA domain, we show improvement in the accuracy of the base DL pipeline from ~30% to ~80%. Minimum heuristics were used to define the pipeline so that it can easily adapt to any new MURA dataset. The paper shows the importance of domain specific preprocessing steps for the designed network in case of MURA defects. CONCLUSION: Using DL, MURA classification and localization had not been tried before. For the first time we demonstrated results for both classification and localization of MURA defects using state-of-the-art DL network with F1~80%. We also conclude that state-of-the-art network for general object detection can be reused with the help of Transfer Learning (TL) concept and fine-tuned with MURA domain specific optimizations mentioned in paper for optimal performances in MURA domain.