对比多种自监督方法,发现联合嵌入模型更适合真实场景的异常检测。
Self-Supervised Anomaly Detection in the Wild: Favor Joint Embeddings Methods
- 采用联合嵌入框架(如SimCLR、Barlow Twins)提升异常检测性能
- 在极端类别不平衡下,重建类方法(如MAE)性能显著下降
- 模型选择比主干网络更重要,适合无标签场景评估研究者参考
准确的异常检测对基于视觉的基础设施检查至关重要,可预防高昂故障并提升安全性。自监督学习(SSL)通过从无标签数据中学习鲁棒表征,提供了有前景的解决方案,但其在异常检测中的应用仍不充分。本文通过全面评估多种SSL方法在真实世界异常检测中的表现,聚焦于污水管网检测。基于Sewer-ML数据集,我们测试了ViT-Tiny和ResNet-18等轻量级模型,在BYOL、Barlow Twins、SimCLR、DINO和MAE等框架下,覆盖不同类别不平衡水平,共完成250次实验。结果表明,联合嵌入方法(如SimCLR、Barlow Twins)优于重建类方法(如MAE),后者在类别不平衡下难以维持性能。此外,模型选择比主干架构更具影响。我们还指出,当前无标签评估方法(如RankMe)不足以衡量表征质量,导致无标签交叉验证困难。尽管自监督模型与有监督模型仍有差距,但本研究揭示了自监督在异常检测中的潜力,推动该未充分探索方向的研究发展。
原文摘要 · Abstract (English)
Accurate anomaly detection is critical in vision-based infrastructure inspection, where it helps prevent costly failures and enhances safety. Self-Supervised Learning (SSL) offers a promising approach by learning robust representations from unlabeled data. However, its application in anomaly detection remains underexplored. This paper addresses this gap by providing a comprehensive evaluation of SSL methods for real-world anomaly detection, focusing on sewer infrastructure. Using the Sewer-ML dataset, we evaluate lightweight models such as ViT-Tiny and ResNet-18 across SSL frameworks, including BYOL, Barlow Twins, SimCLR, DINO, and MAE, under varying class imbalance levels. Through 250 experiments, we rigorously assess the performance of these SSL methods to ensure a robust and comprehensive evaluation. Our findings highlight the superiority of joint-embedding methods like SimCLR and Barlow Twins over reconstruction-based approaches such as MAE, which struggle to maintain performance under class imbalance. Furthermore, we find that the SSL model choice is more critical than the backbone architecture. Additionally, we emphasize the need for better label-free assessments of SSL representations, as current methods like RankMe fail to adequately evaluate representation quality, making cross-validation without labels infeasible. Despite the remaining performance gap between SSL and supervised models, these findings highlight the potential of SSL to enhance anomaly detection, paving the way for further research in this underexplored area of SSL applications.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。