arXiv:2409.20353cs.CVcs.LG2024-09NeurIPS被引 11

专家标注的电力电缆异常检测数据集,助力真实场景下缺陷识别研究

CableInspect-AD: An Expert-Annotated Anomaly Detection Dataset

  • 由电力专家采集并标注的真实电力线路高分辨率图像数据集
  • 提出增强版PatchCore算法,支持少样本场景下的异常检测
  • 适合从事工业视觉检测、少样本学习与真实世界应用的研究者

机器学习模型在现实场景中的应用日益广泛,但针对特定关键任务的可迁移性研究仍不足。以机器人电力线路巡检中的视觉异常检测(VAD)为例,现有方法在受控环境中表现良好,却难以应对真实世界中多样且未知的异常。为此,我们引入CableInspect-AD——一个由加拿大公共事业公司Hydro-Québec领域专家创建并标注的高质量公开数据集。该数据集包含高分辨率图像,涵盖不同程度的缺陷,具有真实世界挑战性。为解决有限标注数据下设定检测阈值的难题,我们改进了经典的PatchCore算法,使其适用于小样本场景。同时提出基于交叉验证的综合评估协议,对增强版PatchCore(少样本/多样本)及视觉-语言模型(零样本)进行评估。尽管表现良好,现有模型仍无法检测所有异常,凸显该数据集作为挑战性基准的价值。

原文摘要 · Abstract (English)

Machine learning models are increasingly being deployed in real-world contexts. However, systematic studies on their transferability to specific and critical applications are underrepresented in the research literature. An important example is visual anomaly detection (VAD) for robotic power line inspection. While existing VAD methods perform well in controlled environments, real-world scenarios present diverse and unexpected anomalies that current datasets fail to capture. To address this gap, we introduce $\textit{CableInspect-AD}$, a high-quality, publicly available dataset created and annotated by domain experts from Hydro-Québec, a Canadian public utility. This dataset includes high-resolution images with challenging real-world anomalies, covering defects with varying severity levels. To address the challenges of collecting diverse anomalous and nominal examples for setting a detection threshold, we propose an enhancement to the celebrated PatchCore algorithm. This enhancement enables its use in scenarios with limited labeled data. We also present a comprehensive evaluation protocol based on cross-validation to assess models' performances. We evaluate our $\textit{Enhanced-PatchCore}$ for few-shot and many-shot detection, and Vision-Language Models for zero-shot detection. While promising, these models struggle to detect all anomalies, highlighting the dataset's value as a challenging benchmark for the broader research community. Project page: https://mila-iqia.github.io/cableinspect-ad/.

异常检测电力巡检少样本学习专家标注

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。