用类别信息提升多类异常检测,解决模型混淆问题
Salvaging the Overlooked: Leveraging Class-Aware Contrastive Learning for Multi-Class Anomaly Detection
- 引入类别感知对比学习,利用真实类别标签指导特征优化
- 在5个数据集上显著提升性能,优于现有最先进方法
- 即使使用伪类别标签也能达到相近效果,适合实际部署
对于异常检测,早期方法通常为每个类别训练独立模型,虽性能优异但难以扩展。近期研究转向单模型处理多类别,但直接迁移原有方法常导致性能下降。本文分析发现,重建类方法在多类场景下性能退化主因是类间混淆:模型会将某类样本错误重建为另一类,加剧重建误差。为此,提出类别感知对比学习(CCL),利用原始类别信息(如carpet、wood)作为监督信号,通过局部对比学习优化多尺度密集特征,全局对比学习获得更紧凑的正常模式表征,有效适配多类场景。在五个数据集上的实验验证了方法有效性,显著优于当前最优方法。消融实验表明,伪类别标签亦可取得相当性能。
原文摘要 · Abstract (English)
For anomaly detection (AD), early approaches often train separate models for individual classes, yielding high performance but posing challenges in scalability and resource management. Recent efforts have shifted toward training a single model capable of handling multiple classes. However, directly extending early AD methods to multi-class settings often results in degraded performance. In this paper, we investigate this performance degradation observed in reconstruction-based methods, identifying the key issue: inter-class confusion. This confusion emerges when a model trained in multi-class scenarios incorrectly reconstructs samples from one class as those of another, thereby exacerbating reconstruction errors. To this end, we propose a simple yet effective modification, called class-aware contrastive learning (CCL). By explicitly leveraging raw object category information (\eg carpet or wood) as supervised signals, we introduce local CL to refine multiscale dense features, and global CL to obtain more compact feature representations of normal patterns, thereby effectively adapting the models to multi-class settings. Experiments across five datasets validate the effectiveness of our approach, demonstrating significant improvements and superior performance compared to state-of-the-art methods. Notably, ablation studies indicate that pseudo-class labels can achieve comparable performance.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。