用扩散模型生成缺陷玻璃图像,解决制造质检中样本不平衡问题。
Enhancing Glass Defect Detection with Diffusion Models: Addressing Imbalanced Datasets in Manufacturing Quality Control
- 用去噪扩散模型生成合成缺陷图像,扩充少数类样本。
- ResNet50V2准确率从78%提升至93%,召回率显著提高。
- 适合工业质检中缺陷样本稀少的场景,可推广至其他制造业。
工业玻璃制造中的视觉缺陷检测因缺陷产品频率低,导致数据严重不均衡,制约了深度学习模型与计算机视觉系统的性能。本文提出一种基于去噪扩散概率模型(DDPM)的新方法,生成合成缺陷玻璃图像用于数据增强,有效缓解制造质量控制中类别不平衡问题,并提升自动化视觉检测能力。该方法显著提升了标准CNN架构(ResNet50V2、EfficientNetB0、MobileNetV2)在异常检测上的表现,大幅改善了所有测试模型对缺陷样本的召回率,同时保持验证集上100%的精确率。其中,ResNet50V2的整体分类准确率从78%提升至93%。本研究提供了一种可扩展、低成本的自动化缺陷检测方案,适用于存在类似类别不平衡挑战的其他工业质量保证系统。
原文摘要 · Abstract (English)
Visual defect detection in industrial glass manufacturing remains a critical challenge due to the low frequency of defective products, leading to imbalanced datasets that limit the performance of deep learning models and computer vision systems. This paper presents a novel approach using Denoising Diffusion Probabilistic Models (DDPMs) to generate synthetic defective glass product images for data augmentation, effectively addressing class imbalance issues in manufacturing quality control and automated visual inspection. The methodology significantly enhances image classification performance of standard CNN architectures (ResNet50V2, EfficientNetB0, and MobileNetV2) in detecting anomalies by increasing the minority class representation. Experimental results demonstrate substantial improvements in key machine learning metrics, particularly in recall for defective samples across all tested deep neural network architectures while maintaining perfect precision on the validation set. The most dramatic improvement was observed in ResNet50V2's overall classification accuracy, which increased from 78\% to 93\% when trained with the augmented data. This work provides a scalable, cost-effective approach to enhancing automated defect detection in glass manufacturing that can potentially be extended to other industrial quality assurance systems and industries with similar class imbalance challenges.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。