arXiv:2409.15980cs.CVcs.AI2024-09被引 3

用少量正常样本和低价设备,90秒完成高精度异常检测。

Leveraging Unsupervised Learning for Cost-Effective Visual Anomaly Detection

  • 基于预训练模型与无监督学习,仅需10张正常图训练。
  • 在树莓派上90秒内完成训练与推理,F1超0.95。
  • 适合预算有限的中小企业快速部署自动化质检。

传统机器学习视觉检测系统需大量数据收集和重复训练,依赖昂贵相机、计算设备及专业技能,对中小型企业负担沉重。本研究探索利用预训练模型与无监督学习方法,在低成本硬件上构建经济高效的视觉异常检测系统。系统采用Anomalib中的无监督学习模型,并通过openVINO部署于廉价的Raspberry Pi设备。实验表明,该系统仅使用10张正常产品图像,即可在树莓派上于90秒内完成异常检测的训练与推理,实现超过0.95的F1宏平均得分。尽管系统对光照、产品位置或背景变化略有敏感,但仍为中小型制造企业提供了快速、低成本的工厂自动化质检方案。代码已开源:https://github.com/Yunbo-max/Cost-Effective-Visual-Anomaly-Detection-using-Unsupervised-Learning。

原文摘要 · Abstract (English)

Traditional machine learning-based visual inspection systems require extensive data collection and repetitive model training to improve accuracy. These systems typically require expensive camera, computing equipment and significant machine learning expertise, which can substantially burden small and medium-sized enterprises. This study explores leveraging unsupervised learning methods with pre-trained models and low-cost hardware to create a cost-effective visual anomaly detection system. The research aims to develop a low-cost visual anomaly detection solution that uses minimal data for model training while maintaining generalizability and scalability. The system utilises unsupervised learning models from Anomalib and is deployed on affordable Raspberry Pi hardware through openVINO. The results show that this cost-effective system can complete anomaly defection training and inference on a Raspberry Pi in just 90 seconds using only 10 normal product images, achieving an F1 macro score exceeding 0.95. While the system is slightly sensitive to environmental changes like lighting, product positioning, or background, it remains a swift and economical method for factory automation inspection for small and medium-sized manufacturers. The code is available at https://github.com/Yunbo-max/Cost-Effective-Visual-Anomaly-Detection-using-Unsupervised-Learning.

异常检测无监督学习边缘计算低成本

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