Unsupervised Image Anomaly Detection and Segmentation Based on Pretrained Feature Mapping

Abstract

Image anomaly detection and segmentation are important for the development of automatic product quality inspection in intelligent manufacturing. Because the normal data can be collected easily and abnormal ones are rarely existent, unsupervised methods based on reconstruction and embedding have been mainly studied for anomaly detection. But the detection performance and computing time require to be further improved. This article proposes a novel framework, named as pretrained feature mapping (PFM), for unsupervised image anomaly detection and segmentation. The proposed PFM maps the image from a pretrained feature space to another one to detect the anomalies effectively. The bidirectional and multihierarchical bidirectional PFM are further proposed and studied for improving the performance. The proposed framework achieves the better results on well-known MVTec AD dataset compared with state-of-the-art methods, with the area under the receiver operating characteristic curve of 97.5% for anomaly detection and of 97.3% for anomaly segmentation over all 15 categories. The proposed framework is also superior in terms of the computing time. The extensive experiments on ablation studies are also conducted to show the effectiveness and efficiency of the proposed framework.