用粗标签知识增强对抗学习,提升异常检测精度。
Boosting Fine-Grained Visual Anomaly Detection with Coarse-Knowledge-Aware Adversarial Learning
- 引入粗粒度标签数据,指导重建特征分布对齐
- 在多个医疗与工业数据集上实现更高检测准确率
- 无需细粒度标注,适合标注成本高的场景
许多无监督视觉异常检测方法通过自编码器重建正常样本,并利用重构误差图进行异常检测与定位。然而,由于神经网络强大的建模与泛化能力,部分异常也能被良好重建,导致检测与定位效果不佳。本文首先收集了一个小型粗粒度标注的异常数据集,进而提出一种粗知识感知的对抗学习方法,使重构特征分布与正常特征分布对齐,从而有效抑制自编码器对异常的重建能力,提升检测精度。考虑到异常通常仅占异常图像中的极小区域,进一步设计了基于块(patch)级别的对抗学习策略。尽管缺乏块级异常信息,但严格证明:仅将异常图像中任意块特征视为异常,该知识感知方法仍可实现重构块特征与正常特征分布的对齐。在四个医学数据集和两个工业数据集上的实验结果表明,该方法显著提升了检测与定位性能。
原文摘要 · Abstract (English)
Many unsupervised visual anomaly detection methods train an auto-encoder to reconstruct normal samples and then leverage the reconstruction error map to detect and localize the anomalies. However, due to the powerful modeling and generalization ability of neural networks, some anomalies can also be well reconstructed, resulting in unsatisfactory detection and localization accuracy. In this paper, a small coarsely-labeled anomaly dataset is first collected. Then, a coarse-knowledge-aware adversarial learning method is developed to align the distribution of reconstructed features with that of normal features. The alignment can effectively suppress the auto-encoder's reconstruction ability on anomalies and thus improve the detection accuracy. Considering that anomalies often only occupy very small areas in anomalous images, a patch-level adversarial learning strategy is further developed. Although no patch-level anomalous information is available, we rigorously prove that by simply viewing any patch features from anomalous images as anomalies, the proposed knowledge-aware method can also align the distribution of reconstructed patch features with the normal ones. Experimental results on four medical datasets and two industrial datasets demonstrate the effectiveness of our method in improving the detection and localization performance.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。