用双原型扩散模型实现大规模多类别异常检测,性能随类别增加更稳定。
Dual Prototype-Conditioned Diffusion Model for Scalable Multi-Class Unsupervised Anomaly Detection in Large Category Spaces

- 通过局部与全局原型协同建模正常样本的细粒度和整体结构特征。
- 在160类数据集上,图像级和像素级AUROC分别提升5.3和2.9点。
- 适合需要高扩展性的工业缺陷检测场景,尤其类别繁多时。
多类别异常检测旨在构建跨多样产品类别的统一模型。然而,随着类别数量增加,因正常分布日益复杂且异质,性能常下降。为此,我们提出DPDiff-AD:一种双原型条件扩散模型,用于大规模多类别异常检测。该模型通过互补的局部与全局原型刻画异质正常分布:局部原型通过最近原型聚合捕获细粒度结构模式,全局原型通过最优传输正则化调控整体特征几何。二者共同定义结构化的正常空间,并通过原型感知注意力机制,在双原型条件下进行扩散重建以优化该空间。联合利用双原型生成,使模型实现精准的正常性建模,保持类别数增长下的结构可分性,支持可扩展异常判别。在五个基准上的大量实验验证了其有效性与可扩展性。在160类的大规模数据集上,图像级和像素级AUROC分别比先前最优方法Dinomaly+提升5.3和2.9点,且类别数增加时性能保持稳定。
原文摘要 · Abstract (English)
Multi-class anomaly detection aims to build unified models across diverse product categories. However, as the number of categories grows, its performance often degrades due to increasingly complex and heterogeneous normal distributions. To address this challenge, we propose DPDiff-AD, a Dual Prototype-conditioned Diffusion model for large-scale multi-class Anomaly Detection. DPDiff-AD models heterogeneous normal distributions through complementary local and global prototypes. Local prototypes capture representative fine-grained structural patterns via nearest-prototype aggregation, while global prototypes regulate holistic feature geometry through optimal transport regularization. Together, these dual-scale representations define a structured normality space. This space is refined through diffusion-based reconstruction conditioned on both local and global prototypes via prototype-aware attention. By jointly leveraging dual prototypes during generation, DPDiff-AD achieves precise normality modeling, preserves structured separability as category cardinality grows, and enables scalable anomaly discrimination. Extensive experiments across five benchmarks demonstrate the effectiveness and scalability of DPDiff-AD. On the 160-category large-scale dataset, it improves image- and pixel-level AUROC by 5.3 and 2.9 points over the previous state-of-the-art method Dinomaly+, while maintaining stable performance as category cardinality increases.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。