arXivDaily arXiv每日学术速递 周一至周五更新
arXiv周末暂无论文更新,休息一下吧,周末愉快~~
arXiv 2410.04289cs.CVcs.AIcs.LG

野外自监督异常检测:青睐联合嵌入方法

Self-Supervised Anomaly Detection in the Wild: Favor Joint Embeddings Methods

EXIT83;Brown University(布朗大学)

Daniel Otero, Rafael Mateus, Randall Balestriero

AI总结:

该研究针对自监督学习在异常检测中应用不足的问题,基于下水道巡检场景开展250组实验,评估多种SSL框架与轻量模型,发现联合嵌入方法优于重构类方法,且模型选择比骨干架构更关键,为该领域后续研究提供参考。

摘要:

精准的异常检测在基于视觉的基础设施巡检中至关重要,有助于预防代价高昂的故障并提升安全性。自监督学习(Self-Supervised Learning,SSL)通过从未标记数据中学习鲁棒表征,提供了一种颇具前景的途径。然而,其在异常检测中的应用仍未得到充分探索。本文针对这一空白,对面向真实世界异常检测的SSL方法开展了综合评估,研究聚焦于下水道基础设施场景。我们使用Sewer-ML数据集,在不同类别不平衡水平下,跨BYOL、Barlow Twins、SimCLR、DINO和MAE等SSL框架评估了ViT-Tiny和ResNet-18等轻量级模型。通过250组实验,我们对这些SSL方法的性能进行了严格评估,以确保评估的鲁棒性与全面性。研究结果表明,SimCLR、Barlow Twins这类联合嵌入方法的表现优于MAE等基于重构的方法,后者在类别不平衡场景下难以维持性能。此外,我们发现SSL模型的选择比骨干架构更为关键。我们还强调需要更优的无标记SSL表征评估方法,因为当前的RankMe等方法无法充分评估表征质量,导致无标记交叉验证无法实现。尽管SSL模型与监督模型之间仍存在性能差距,但这些发现凸显了SSL提升异常检测效果的潜力,为SSL应用这一未被充分探索的领域的后续研究铺平了道路。

英文摘要:

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.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑