Enhancing Glass Defect Detection with Diffusion Models: Addressing Imbalanced Datasets in Manufacturing Quality Control
On the map: Diffusion models → Flow matching, time series and science
Abstract (arXiv)
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.
Key facts. Cited by 3 papers in the map (data through 14 Oct 2026). Placed in Flow matching, time series and science, one of 14 directions in Diffusion models. Builds on 17 of its 290 references that are in the map. First posted on arXiv 6 May 2025.
Builds on
17 of 290 references.
References · 17
- 2020Denoising Diffusion Probabilistic Modelscited by 21,076
- 2021High-Resolution Image Synthesis with Latent Diffusion Modelscited by 18,863
- 2021Diffusion Models Beat GANs on Image Synthesiscited by 8,255
- 2019EfficientNet: Rethinking Model Scaling for Convolutional Neural Networkscited by 6,442
- 2018MobileNetV2: Inverted Residuals and Linear Bottleneckscited by 4,795
- 2021Improved Denoising Diffusion Probabilistic Modelscited by 3,761
- 2022Elucidating the Design Space of Diffusion-Based Generative Modelscited by 3,045
- 2016Identity Mappings in Deep Residual Networkscited by 1,965
- 2017A systematic study of the class imbalance problem in convolutional neural networkscited by 513
- 2022Diffusion Models for Medical Anomaly Detectioncited by 232
- 2015Cost Sensitive Learning of Deep Feature Representations from Imbalanced Datacited by 167
- 2018Detection and Segmentation of Manufacturing Defects with Convolutional Neural Networks and Transfer Learningcited by 16
5 more
- 2025Prioritizing App Reviews for Developer Responses on Google Playcited by 3
- 2025Network Anomaly Detection for IoT Using Hyperdimensional Computing on NSL-KDDcited by 3
- 2024Addressing Small and Imbalanced Medical Image Datasets Using Generative Models: A Comparative Study of DDPM and PGGANs with Random and Greedy K Samplingcited by 2
- 2025Machine Learning Models for Reinforced Concrete Pipes Condition Prediction: The State-of-the-Art Using Artificial Neural Networks and Multiple Linear Regression in a Wisconsin Case Studycited by 2
- 2025rPPG-SysDiaGAN: Systolic-Diastolic Feature Localization in rPPG Using Generative Adversarial Network with Multi-Domain Discriminatorcited by 2
Built on by
Citing papers per year: 2025: 2; 2026: 1.
3 papers.
- 2025Sanitizing Manufacturing Dataset Labels Using Vision-Language Modelscited by 6
- 2025Quantifying Label-Induced Bias in Large Language Model Self- and Cross-Evaluationscited by 3
- 2026Spatial Attention Supervision for Defect Localization: Exploiting Ground-Truth Masks as Training Signal in Diffusion-Augmented Defect Detectioncited by 0
- 2026Spatial Attention Supervision for Defect Localization: Exploiting Ground-Truth Masks as Training Signal in Diffusion-Augmented Defect Detection5 Sep 2026
- 2025Quantifying Label-Induced Bias in Large Language Model Self- and Cross-Evaluations28 Aug 2025
- 2025Sanitizing Manufacturing Dataset Labels Using Vision-Language Models30 Jun 2025
Similar, not cited
Close in wording, no citation either way
- 2024Bring the Power of Diffusion Model to Defect Detection
- 2024DDPM-MoCo: Advancing Industrial Surface Defect Generation and Detection with Generative and Contrastive Learning
- 2026Industrial Surface Defect Detection via Diffusion Generation and Asymmetric Student-Teacher Network
- 2025Semi-Supervised Defect Detection via Conditional Diffusion and CLIP-Guided Noise Filtering
- 2024Diffusion-based Image Generation for In-distribution Data Augmentation in Surface Defect Detection
- 2026Improving Supervised Machine Learning Performance in Optical Quality Control via Generative AI for Dataset Expansion
- 2024Leveraging Latent Diffusion Models for Training-Free In-Distribution Data Augmentation for Surface Defect Detection
- 2025From Lab to Factory: Pitfalls and Guidelines for Self-/Unsupervised Defect Detection on Low-Quality Industrial Images