Image Anomaly Detection with Semantic- Enhanced Augmentation and Distributional Kernel.

Mingxi Wang,Ye Zhu,Gang Li,Gang Liu,Bo Yang
DOI: https://doi.org/10.1109/hpcc-dss-smartcity-dependsys57074.2022.00054
2022-01-01
Abstract:In the field of image anomaly detection, self-supervised learning using only normal samples has become a hot research topic since it is difficult to acquire sufficient abnormal data. However, existing self-supervised methods usually consider the impact of the superficial information rather than the semantics of images, resulting in limited generalisation performance. Furthermore, the small amount of training data further limits the performance of anomaly detection. To overcome both issues, we propose a two-stage self-supervised learning framework based on CutPaste with a feature extraction stage followed by an anomaly detection stage. We first add a diminished semantics augmentation to generate harder abnormal samples, so that the feature extraction model pays more attention to semantic information. Furthermore, we increase the density of normal samples with normal sample augmentations and integrate a data-dependent distributional kernel to improve the detection of abnormal images in the second stage, based on the adaptive distributional similarity. Our extensive experiments on the MVTec anomaly detection dataset demonstrate the superior performance of the proposed method.
What problem does this paper attempt to address?