用视觉语言模型自动清理制造图像数据的错误标签。
Sanitizing Manufacturing Dataset Labels Using Vision-Language Models
- 将图像和标签映射到统一语义空间,通过相似度匹配识别问题标签。
- 在Factorynet数据集上减少标签数量并提升一致性,效果显著。
- 适合需要高质量标注但人力成本高的工业场景使用。
机器学习在工业应用中的成功高度依赖训练数据集的质量。然而,大规模数据集(尤其是众包或网络爬取构建的)常存在标签噪声、不一致和错误,制造领域尤为严重,因高质量标注耗时费力。本文提出基于视觉语言模型的标签清洗与优化框架VLSR,利用CLIP将图像与文本标签嵌入共享语义空间,通过计算图像-标签嵌入的余弦相似度,实现两步处理:一是识别无关、拼写错误或语义弱的标签,选出最匹配的标签;二是对文本嵌入进行密度聚类并迭代合并,将语义相近标签归为统一类别。实验基于包含人工标注与网络爬取噪声标签的Factorynet数据集,结果表明该方法有效识别问题标签,显著提升标签一致性,并通过聚类大幅缩减标签词表,以最小人工干预增强数据质量,利于训练鲁棒的工业级机器学习模型。
原文摘要 · Abstract (English)
The success of machine learning models in industrial applications is heavily dependent on the quality of the datasets used to train the models. However, large-scale datasets, specially those constructed from crowd-sourcing and web-scraping, often suffer from label noise, inconsistencies, and errors. This problem is particularly pronounced in manufacturing domains, where obtaining high-quality labels is costly and time-consuming. This paper introduces Vision-Language Sanitization and Refinement (VLSR), which is a vision-language-based framework for label sanitization and refinement in multi-label manufacturing image datasets. This method embeds both images and their associated textual labels into a shared semantic space leveraging the CLIP vision-language model. Then two key tasks are addressed in this process by computing the cosine similarity between embeddings. First, label sanitization is performed to identify irrelevant, misspelled, or semantically weak labels, and surface the most semantically aligned label for each image by comparing image-label pairs using cosine similarity between image and label embeddings. Second, the method applies density-based clustering on text embeddings, followed by iterative cluster merging, to group semantically similar labels into unified label groups. The Factorynet dataset, which includes noisy labels from both human annotations and web-scraped sources, is employed to evaluate the effectiveness of the proposed framework. Experimental results demonstrate that the VLSR framework successfully identifies problematic labels and improves label consistency. This method enables a significant reduction in label vocabulary through clustering, which ultimately enhances the dataset's quality for training robust machine learning models in industrial applications with minimal human intervention.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。