MTHM: Self-Supervised Multitask Anomaly Detection with Hard Example Mining

Chenkai Zhang,Yueming Wang,Wenming Tan
DOI: https://doi.org/10.1109/tim.2023.3276529
IF: 5.6
2023-01-01
IEEE Transactions on Instrumentation and Measurement
Abstract:It is still challenging to detect and locate anomalies by models trained only with normal samples. Methods using image reconstruction as a pretext task can provide precise localization but suffer from harnessing the reconstruction capability on unseen anomalies. This article proposes a new framework of multitasking and hard example mining (MTHM) for anomaly detection and localization. The self-supervised multitask setting creatively takes advantage of the competition among different tasks to learn more compact and efficient representations for detection tasks. Moreover, introducing other semantic tasks allows the shared encoder to learn beyond the pixel-to-pixel mapping of only a single image reconstruction task. Subsequent analysis experiments demonstrate that the proposed method can achieve a more suppressive reconstruction capability for anomalies. During the test process, the outputs of the other tasks can also provide valuable information for anomaly detection and localization. Furthermore, in combination with a novel hard example mining strategy, the byproducts of the image reconstruction task are inexpensively exploited as hard-to-detect samples for enhancing models’ detection ability. And as the model capability increases during training, the detection difficulty of these samples is able to increase adaptively. Our experiments show that hard samples generated in the later training stages can better approximate the real data distribution. With the help of the multitask framework and hard example mining strategy, our method surpasses many state-of-the-art methods.
What problem does this paper attempt to address?