HyperAIHyperAI

Command Palette

Search for a command to run...

3 months ago

Industrial Anomaly Detection and Localization Using Weakly-Supervised Residual Transformers

Hanxi Li Jingqi Wu Deyin Liu Lin Wu Hao Chen Mingwen Wang Chunhua Shen

Industrial Anomaly Detection and Localization Using Weakly-Supervised Residual Transformers

Abstract

Recent advancements in industrial anomaly detection (AD) have demonstrated that incorporating a small number of anomalous samples during training can significantly enhance accuracy. However, this improvement often comes at the cost of extensive annotation efforts, which are impractical for many real-world applications. In this paper, we introduce a novel framework, Weak}ly-supervised RESidual Transformer (WeakREST), designed to achieve high anomaly detection accuracy while minimizing the reliance on manual annotations. First, we reformulate the pixel-wise anomaly localization task into a block-wise classification problem. Second, we introduce a residual-based feature representation called Positional Fast Anomaly Residuals (PosFAR) which captures anomalous patterns more effectively. To leverage this feature, we adapt the Swin Transformer for enhanced anomaly detection and localization. Additionally, we propose a weak annotation approach, utilizing bounding boxes and image tags to define anomalous regions. This approach establishes a semi-supervised learning context that reduces the dependency on precise pixel-level labels. To further improve the learning process, we develop a novel ResMixMatch algorithm, capable of handling the interplay between weak labels and residual-based representations. On the benchmark dataset MVTec-AD, our method achieves an Average Precision (AP) of $83.0\%$, surpassing the previous best result of $82.7\%$ in the unsupervised setting. In the supervised AD setting, WeakREST attains an AP of $87.6\%$, outperforming the previous best of $86.0\%$. Notably, even when using weaker annotations such as bounding boxes, WeakREST exceeds the performance of leading methods relying on pixel-wise supervision, achieving an AP of $87.1\%$ compared to the prior best of $86.0\%$ on MVTec-AD.

Benchmarks

BenchmarkMethodologyMetrics
anomaly-detection-on-btadWeakREST-Un
Detection AUROC: 94.4
Segmentation AP: 63.1
Segmentation AUPRO: 84.9
Segmentation AUROC: 98.7
anomaly-detection-on-mvtec-adWeakREST-Un
Detection AUROC: 99.6
FPS: 25.2
Segmentation AP: 83.0
Segmentation AUPRO: 97.6
Segmentation AUROC: 99.3
supervised-anomaly-detection-on-btadWeakREST-Block
Detection AUROC: 96.5
Segmentation AP: 84.6
Segmentation AUPRO: 90.8
Segmentation AUROC: 99.3
supervised-anomaly-detection-on-mvtec-adWeakREST-Block
Detection AUROC: 99.8
Segmentation AP: 87.6
Segmentation AUPRO: 98.4
Segmentation AUROC: 99.7
unsupervised-anomaly-detection-onWeakREST-Un
Segmentation AP: 76.9
Segmentation AUPRO: 98.5
Segmentation AUROC: 99.7

Build AI with AI

From idea to launch — accelerate your AI development with free AI co-coding, out-of-the-box environment and best price of GPUs.

AI Co-coding
Ready-to-use GPUs
Best Pricing
Get Started

Hyper Newsletters

Subscribe to our latest updates
We will deliver the latest updates of the week to your inbox at nine o'clock every Monday morning
Powered by MailChimp
Industrial Anomaly Detection and Localization Using Weakly-Supervised Residual Transformers | Papers | HyperAI