HyperAIHyperAI

Command Palette

Search for a command to run...

3 months ago

SLSG: Industrial Image Anomaly Detection by Learning Better Feature Embeddings and One-Class Classification

Minghui Yang Jing Liu Zhiwei Yang Zhaoyang Wu

SLSG: Industrial Image Anomaly Detection by Learning Better Feature Embeddings and One-Class Classification

Abstract

Industrial image anomaly detection under the setting of one-class classification has significant practical value. However, most existing models struggle to extract separable feature representations when performing feature embedding and struggle to build compact descriptions of normal features when performing one-class classification. One direct consequence of this is that most models perform poorly in detecting logical anomalies which violate contextual relationships. Focusing on more effective and comprehensive anomaly detection, we propose a network based on self-supervised learning and self-attentive graph convolution (SLSG) for anomaly detection. SLSG uses a generative pre-training network to assist the encoder in learning the embedding of normal patterns and the reasoning of position relationships. Subsequently, SLSG introduces the pseudo-prior knowledge of anomaly through simulated abnormal samples. By comparing the simulated anomalies, SLSG can better summarize the normal features and narrow down the hypersphere used for one-class classification. In addition, with the construction of a more general graph structure, SLSG comprehensively models the dense and sparse relationships among elements in the image, which further strengthens the detection of logical anomalies. Extensive experiments on benchmark datasets show that SLSG achieves superior anomaly detection performance, demonstrating the effectiveness of our method.

Benchmarks

BenchmarkMethodologyMetrics
anomaly-detection-on-mvtec-loco-adSLSG
Avg. Detection AUROC: 90.3
Detection AUROC (only logical): 89.6
Detection AUROC (only structural): 91.4
Segmentation AU-sPRO (until FPR 5%): 67.3

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
SLSG: Industrial Image Anomaly Detection by Learning Better Feature Embeddings and One-Class Classification | Papers | HyperAI