Omni-frequency Channel-selection Representations for Unsupervised Anomaly Detection

TL;DR

This paper introduces OCR-GAN, a novel frequency-aware reconstruction network for unsupervised anomaly detection that decouples images into frequency components and adaptively selects channels, achieving state-of-the-art results without extra training data.

Contribution

It proposes a frequency decoupling and channel selection framework for reconstruction-based anomaly detection, improving performance and surpassing existing methods.

Findings

Achieves 98.3% AUC on MVTec AD dataset, a new state-of-the-art.

Outperforms baseline by +38.1 in detection accuracy.

No extra training data required.

Abstract

Density-based and classification-based methods have ruled unsupervised anomaly detection in recent years, while reconstruction-based methods are rarely mentioned for the poor reconstruction ability and low performance. However, the latter requires no costly extra training samples for the unsupervised training that is more practical, so this paper focuses on improving this kind of method and proposes a novel Omni-frequency Channel-selection Reconstruction (OCR-GAN) network to handle anomaly detection task in a perspective of frequency. Concretely, we propose a Frequency Decoupling (FD) module to decouple the input image into different frequency components and model the reconstruction process as a combination of parallel omni-frequency image restorations, as we observe a significant difference in the frequency distribution of normal and abnormal images. Given the correlation among multiple frequencies, we further propose a Channel Selection (CS) module that performs frequency interaction among different encoders by adaptively selecting different channels. Abundant experiments demonstrate the effectiveness and superiority of our approach over different kinds of methods, e.g., achieving a new state-of-the-art 98.3 detection AUC on the MVTec AD dataset without extra training data that markedly surpasses the reconstruction-based baseline by +38.1 and the current SOTA method by +0.3. Source code is available at https://github.com/zhangzjn/OCR-GAN.