arXiv:2409.10104cs.CVcs.LG2024-09

小数据下用迁移学习训练视觉模型,效果不降反升

A Comparative Study of Open Source Computer Vision Models for Application on Small Data: The Case of CFRP Tape Laying

  • 用迁移学习在少量数据上训练开源视觉模型
  • 极少量数据即可达成有效质量检测性能
  • 小模型也能保持高性能,适合工业小样本场景

在工业制造领域,人工智能正越来越多地应用于流程自动化与新材料研发。然而,在数据量有限的中小型实验性工艺中,如何训练AI模型仍面临挑战。本文探讨了迁移学习在小样本场景下的应用潜力,重点研究构建功能型AI模型所需的最小数据量。以航空航天制造中的碳纤维增强聚合物(CFRP)铺放质量控制为例,利用光学传感器采集数据,系统评估不同开源计算机视觉模型在持续减少训练数据条件下的表现。结果表明,成功训练AI模型所需的数据量可大幅降低,且使用小型模型并不必然导致性能下降。

原文摘要 · Abstract (English)

In the realm of industrial manufacturing, Artificial Intelligence (AI) is playing an increasing role, from automating existing processes to aiding in the development of new materials and techniques. However, a significant challenge arises in smaller, experimental processes characterized by limited training data availability, questioning the possibility to train AI models in such small data contexts. In this work, we explore the potential of Transfer Learning to address this challenge, specifically investigating the minimum amount of data required to develop a functional AI model. For this purpose, we consider the use case of quality control of Carbon Fiber Reinforced Polymer (CFRP) tape laying in aerospace manufacturing using optical sensors. We investigate the behavior of different open-source computer vision models with a continuous reduction of the training data. Our results show that the amount of data required to successfully train an AI model can be drastically reduced, and the use of smaller models does not necessarily lead to a loss of performance.

小样本学习视觉检测迁移学习工业质检

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