通过个性化生成正常图像提升少样本异常检测精度
One-to-Normal: Anomaly Personalization for Few-shot Anomaly Detection
- 用定制生成模型将查询图转为类正常样态
- 在11个数据集上优于最新方法,提升检测准确率
- 可适配其他检测方法,适合复杂场景应用
传统异常检测方法主要依赖大量正常数据的无监督学习。近年随着大规模预训练视觉-语言模型的发展,少样本异常检测能力有所提升,但仍存在精度瓶颈。主要原因在于直接比较查询图像与少量正常图像特征,易导致精度损失,并难以拓展至复杂领域。为此,本文提出异常个性化方法,通过无异常的定制化生成模型对查询图像进行个性化一到正常转换,确保其与正常流形高度对齐。为进一步增强预测稳定性与鲁棒性,提出三元组对比异常推理策略,综合比较查询图像、生成的无异常数据池与提示信息。在三个领域的十一个数据集上广泛评估表明,本方法显著优于最新异常检测方法。此外,该方法可灵活迁移至其他检测框架,生成图像数据能有效提升其他方法性能。
原文摘要 · Abstract (English)
Traditional Anomaly Detection (AD) methods have predominantly relied on unsupervised learning from extensive normal data. Recent AD methods have evolved with the advent of large pre-trained vision-language models, enhancing few-shot anomaly detection capabilities. However, these latest AD methods still exhibit limitations in accuracy improvement. One contributing factor is their direct comparison of a query image's features with those of few-shot normal images. This direct comparison often leads to a loss of precision and complicates the extension of these techniques to more complex domains--an area that remains underexplored in a more refined and comprehensive manner. To address these limitations, we introduce the anomaly personalization method, which performs a personalized one-to-normal transformation of query images using an anomaly-free customized generation model, ensuring close alignment with the normal manifold. Moreover, to further enhance the stability and robustness of prediction results, we propose a triplet contrastive anomaly inference strategy, which incorporates a comprehensive comparison between the query and generated anomaly-free data pool and prompt information. Extensive evaluations across eleven datasets in three domains demonstrate our model's effectiveness compared to the latest AD methods. Additionally, our method has been proven to transfer flexibly to other AD methods, with the generated image data effectively improving the performance of other AD methods.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。