Prior Normality Prompt Transformer for Multi-class Industrial Image Anomaly Detection

Fuente: arXiv
Saved in:
Bibliographic Details
Main Authors: Yao, Haiming, Cao, Yunkang, Luo, Wei, Zhang, Weihang, Yu, Wenyong, Shen, Weiming
Format: Preprint
Published: 2024
Subjects:
Online Access:
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866914836731920384
author Yao, Haiming
Cao, Yunkang
Luo, Wei
Zhang, Weihang
Yu, Wenyong
Shen, Weiming
author_facet Yao, Haiming
Cao, Yunkang
Luo, Wei
Zhang, Weihang
Yu, Wenyong
Shen, Weiming
contents Image anomaly detection plays a pivotal role in industrial inspection. Traditional approaches often demand distinct models for specific categories, resulting in substantial deployment costs. This raises concerns about multi-class anomaly detection, where a unified model is developed for multiple classes. However, applying conventional methods, particularly reconstruction-based models, directly to multi-class scenarios encounters challenges such as identical shortcut learning, hindering effective discrimination between normal and abnormal instances. To tackle this issue, our study introduces the Prior Normality Prompt Transformer (PNPT) method for multi-class image anomaly detection. PNPT strategically incorporates normal semantics prompting to mitigate the "identical mapping" problem. This entails integrating a prior normality prompt into the reconstruction process, yielding a dual-stream model. This innovative architecture combines normal prior semantics with abnormal samples, enabling dual-stream reconstruction grounded in both prior knowledge and intrinsic sample characteristics. PNPT comprises four essential modules: Class-Specific Normality Prompting Pool (CS-NPP), Hierarchical Patch Embedding (HPE), Semantic Alignment Coupling Encoding (SACE), and Contextual Semantic Conditional Decoding (CSCD). Experimental validation on diverse benchmark datasets and real-world industrial applications highlights PNPT's superior performance in multi-class industrial anomaly detection.
format Preprint
id arxiv_https___arxiv_org_abs_2406_11507
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle Prior Normality Prompt Transformer for Multi-class Industrial Image Anomaly Detection
Yao, Haiming
Cao, Yunkang
Luo, Wei
Zhang, Weihang
Yu, Wenyong
Shen, Weiming
Computer Vision and Pattern Recognition
Image anomaly detection plays a pivotal role in industrial inspection. Traditional approaches often demand distinct models for specific categories, resulting in substantial deployment costs. This raises concerns about multi-class anomaly detection, where a unified model is developed for multiple classes. However, applying conventional methods, particularly reconstruction-based models, directly to multi-class scenarios encounters challenges such as identical shortcut learning, hindering effective discrimination between normal and abnormal instances. To tackle this issue, our study introduces the Prior Normality Prompt Transformer (PNPT) method for multi-class image anomaly detection. PNPT strategically incorporates normal semantics prompting to mitigate the "identical mapping" problem. This entails integrating a prior normality prompt into the reconstruction process, yielding a dual-stream model. This innovative architecture combines normal prior semantics with abnormal samples, enabling dual-stream reconstruction grounded in both prior knowledge and intrinsic sample characteristics. PNPT comprises four essential modules: Class-Specific Normality Prompting Pool (CS-NPP), Hierarchical Patch Embedding (HPE), Semantic Alignment Coupling Encoding (SACE), and Contextual Semantic Conditional Decoding (CSCD). Experimental validation on diverse benchmark datasets and real-world industrial applications highlights PNPT's superior performance in multi-class industrial anomaly detection.
title Prior Normality Prompt Transformer for Multi-class Industrial Image Anomaly Detection
topic Computer Vision and Pattern Recognition
url https://arxiv.org/abs/2406.11507