摘要
arXiv:2505.09264v1 交叉类型:公告
摘要:使用自注意力变换器的无监督重建网络在单个模型中实现了多类别(统一)异常检测的最先进性能。然而,这些自注意力重建模型主要操作目标特征,可能会由于上下文的一致性而导致完美地重建正常和异常特征,从而在检测异常方面失败。另外,由于在低空间分辨率的潜在空间中进行重建,这些模型通常会产生不准确的异常分割。为了解决同时拥有高效重建和增强统一异常检测的泛化能力的问题,我们提出了一种简单且有效的方法,即仅通过一个正常图像提示(OneNIP)进行正常特征的重建和异常特征的恢复。与以前的工作相比,OneNIP首次能够仅通过一个正常图像提示重建或恢复异常特征,从而显著提升统一异常检测的性能。此外,我们提出了一种监督精修器,通过使用真实正常图像和合成异常图像来回归重建误差,这极大地提高了像素级异常分割的精度。OneNIP在三个工业异常检测基准(MVTec、BTAD和VisA)中优于以前的方法。代码和预训练模型可在 https://github.com/gaobb/OneNIP 获取。