用单步扩散模型精准生成带标签的缺陷图像,提升工业质检数据增强效果。
Ali-AUG: Innovative Approaches to Labeled Data Augmentation using One-Step Diffusion Model
- 基于单步扩散与跳连、LoRA结构,实现图像与掩码的高效融合。
- 相比其他方法提升模型准确率31%,无数据增强时提升45%。
- 支持成对与非成对数据,训练时间减少32%,适合制造业缺陷检测。
本文提出Ali-AUG,一种用于工业场景中高效有标签数据增强的单步扩散模型。该方法通过生成带有精确特征插入的合成标签图像,缓解标注数据不足问题。Ali-AUG采用具备跳跃连接和LoRA模块的稳定扩散架构,能高效整合掩码与图像,确保特征定位准确且不影响无关内容。在多个工业数据集上的实验表明,Ali-AUG能生成高质量缺陷增强图像,并实现快速单步推理。通过分类准确率得分(CAS)与朴素增强得分(NAS)验证,其相较其他增强方法提升模型性能31%,较无增强模型提升45%。此外,训练时间减少32%,支持成对与非成对数据集,显著提升数据准备灵活性,特别适用于制造过程中缺陷产品图像生成以训练缺陷检测模型。
原文摘要 · Abstract (English)
This paper introduces Ali-AUG, a novel single-step diffusion model for efficient labeled data augmentation in industrial applications. Our method addresses the challenge of limited labeled data by generating synthetic, labeled images with precise feature insertion. Ali-AUG utilizes a stable diffusion architecture enhanced with skip connections and LoRA modules to efficiently integrate masks and images, ensuring accurate feature placement without affecting unrelated image content. Experimental validation across various industrial datasets demonstrates Ali-AUG's superiority in generating high-quality, defect-enhanced images while maintaining rapid single-step inference. By offering precise control over feature insertion and minimizing required training steps, our technique significantly enhances data augmentation capabilities, providing a powerful tool for improving the performance of deep learning models in scenarios with limited labeled data. Ali-AUG is especially useful for use cases like defective product image generation to train AI-based models to improve their ability to detect defects in manufacturing processes. Using different data preparation strategies, including Classification Accuracy Score (CAS) and Naive Augmentation Score (NAS), we show that Ali-AUG improves model performance by 31% compared to other augmentation methods and by 45% compared to models without data augmentation. Notably, Ali-AUG reduces training time by 32% and supports both paired and unpaired datasets, enhancing flexibility in data preparation.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。