arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2307.08059cs.CV

LafitE:用于无监督多类别异常检测的特征编辑潜在扩散模型

LafitE: Latent Diffusion Model with Feature Editing for Unsupervised Multi-class Anomaly Detection

Tsinghua University(清华大学);Microsoft Research Asia(微软亚洲研究院)

Haonan Yin, Guanlong Jiao, Qianhui Wu, Borje F. Karlsson, Biqing Huang, Chin Yew Lin

AI总结:

针对仅用正常数据的无监督多类别异常检测,本文提出 LafitE,通过潜在扩散重建与特征编辑缓解恒等捷径,并用合成异常的伪验证集选择超参数,在 MVTec-AD 和 MPDD 上取得领先平均 AUROC。

摘要:

在柔性制造系统需要以最少重构生产不同类型和数量产品的背景下,本文研究无监督多类别异常检测问题:在仅能获取正常数据时,开发一个统一模型,以检测来自多个类别对象的异常。我们首先探索基于生成的方法,并研究用于重建的潜在扩散模型,以缓解基于自编码器方法中臭名昭著的“恒等捷径”问题。随后,我们引入一种特征编辑策略,修改扩散模型的输入特征空间,以进一步缓解“恒等捷径”,同时提高正常区域的重建质量,从而产生更少的假阳性预测。此外,我们首次提出无监督异常检测中的超参数选择问题,并提出一种合成异常数据以构建伪验证集的解决方案来解决该问题。在基准数据集 MVTec-AD 和 MPDD 上的大量实验表明,所提出的 LafitE,即带特征编辑的潜在扩散模型,在平均 AUROC 方面显著优于最先进方法。通过我们的伪验证集选择的超参数与真实测试集良好匹配。

英文摘要:

In the context of flexible manufacturing systems that are required to produce different types and quantities of products with minimal reconfiguration, this paper addresses the problem of unsupervised multi-class anomaly detection: develop a unified model to detect anomalies from objects belonging to multiple classes when only normal data is accessible. We first explore the generative-based approach and investigate latent diffusion models for reconstruction to mitigate the notorious ``identity shortcut'' issue in auto-encoder based methods. We then introduce a feature editing strategy that modifies the input feature space of the diffusion model to further alleviate ``identity shortcuts'' and meanwhile improve the reconstruction quality of normal regions, leading to fewer false positive predictions. Moreover, we are the first who pose the problem of hyperparameter selection in unsupervised anomaly detection, and propose a solution of synthesizing anomaly data for a pseudo validation set to address this problem. Extensive experiments on benchmark datasets MVTec-AD and MPDD show that the proposed LafitE, \ie, Latent Diffusion Model with Feature Editing, outperforms state-of-art methods by a significant margin in terms of average AUROC. The hyperparamters selected via our pseudo validation set are well-matched to the real test set.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑