Synthetic Data Augmentation Using GAN For Improved Automated Visual Inspection

• URL: http://arxiv.org/abs/2212.09317v1
• Date: Mon, 19 Dec 2022 09:31:15 GMT
• Title: Synthetic Data Augmentation Using GAN For Improved Automated Visual Inspection
• Authors: Jo\v{z}e M. Ro\v{z}anec, Patrik Zajec, Spyros Theodoropoulos, Erik Koehorst, Bla\v{z} Fortuna, Dunja Mladeni\'c
• Abstract summary: State-of-the-art unsupervised defect detection does not match the performance of supervised models. Best classification performance was achieved considering GAN-based data generation with AUC ROC scores equal to or higher than 0,9898.
• Score: 0.440401067183266
• License: http://arxiv.org/licenses/nonexclusive-distrib/1.0/
• Abstract: Quality control is a crucial activity performed by manufacturing companies to ensure their products conform to the requirements and specifications. The introduction of artificial intelligence models enables to automate the visual quality inspection, speeding up the inspection process and ensuring all products are evaluated under the same criteria. In this research, we compare supervised and unsupervised defect detection techniques and explore data augmentation techniques to mitigate the data imbalance in the context of automated visual inspection. Furthermore, we use Generative Adversarial Networks for data augmentation to enhance the classifiers' discriminative performance. Our results show that state-of-the-art unsupervised defect detection does not match the performance of supervised models but can be used to reduce the labeling workload by more than 50%. Furthermore, the best classification performance was achieved considering GAN-based data generation with AUC ROC scores equal to or higher than 0,9898, even when increasing the dataset imbalance by leaving only 25\% of the images denoting defective products. We performed the research with real-world data provided by Philips Consumer Lifestyle BV.

Related papers

• Enhancing Glass Defect Detection with Diffusion Models: Addressing Imbalanced Datasets in Manufacturing Quality Control [0.0]
  This paper presents a novel approach using Denoising Diffusion Probabilistic Models (DDPMs) to generate synthetic defective glass product images for data augmentation.<n>The methodology significantly enhances image classification performance of standard CNN architectures.<n>The most dramatic improvement was observed in ResNet50V2's overall classification accuracy, which increased from 78% to 93% when trained with the augmented data.
  arXiv  Detail & Related papers  (2025-05-06T03:16:56Z)
• Trimming the Risk: Towards Reliable Continuous Training for Deep Learning Inspection Systems [10.121053770426759]
  The industry increasingly relies on deep learning (DL) technology for manufacturing inspections. finite labeled datasets often fail to encompass all natural variations. We develop a robust CT-based maintenance approach that updates DL models using reliable data selections.
  arXiv  Detail & Related papers  (2024-09-13T15:02:13Z)
• DetDiffusion: Synergizing Generative and Perceptive Models for Enhanced Data Generation and Perception [78.26734070960886]
  Current perceptive models heavily depend on resource-intensive datasets. We introduce perception-aware loss (P.A. loss) through segmentation, improving both quality and controllability. Our method customizes data augmentation by extracting and utilizing perception-aware attribute (P.A. Attr) during generation.
  arXiv  Detail & Related papers  (2024-03-20T04:58:03Z)
• Imbalanced Data Classification via Generative Adversarial Network with Application to Anomaly Detection in Additive Manufacturing Process [5.225026952905702]
  This paper proposes a novel data augmentation method based on a generative adversarial network (GAN) using additive manufacturing process image sensor data. The diverse and high-quality generated samples provide balanced training data to the classifier. The effectiveness of the proposed method is validated by both open-source data and real-world case studies in polymer and metal AM processes.
  arXiv  Detail & Related papers  (2022-10-28T16:08:21Z)
• An Outlier Exposure Approach to Improve Visual Anomaly Detection Performance for Mobile Robots [76.36017224414523]
  We consider the problem of building visual anomaly detection systems for mobile robots. Standard anomaly detection models are trained using large datasets composed only of non-anomalous data. We tackle the problem of exploiting these data to improve the performance of a Real-NVP anomaly detection model.
  arXiv  Detail & Related papers  (2022-09-20T15:18:13Z)
• Active Learning and Approximate Model Calibration for Automated Visual Inspection in Manufacturing [0.415623340386296]
  This research compares three active learning approaches (with single and multiple oracles) to visual inspection. We propose a novel approach to probabilities calibration of classification models and two new metrics to assess the performance of the calibration without the need for ground truth.
  arXiv  Detail & Related papers  (2022-09-12T15:00:29Z)
• Augmentation-Aware Self-Supervision for Data-Efficient GAN Training [68.81471633374393]
  Training generative adversarial networks (GANs) with limited data is challenging because the discriminator is prone to overfitting. We propose a novel augmentation-aware self-supervised discriminator that predicts the augmentation parameter of the augmented data. We compare our method with state-of-the-art (SOTA) methods using the class-conditional BigGAN and unconditional StyleGAN2 architectures.
  arXiv  Detail & Related papers  (2022-05-31T10:35:55Z)
• Incorporating Semi-Supervised and Positive-Unlabeled Learning for Boosting Full Reference Image Quality Assessment [73.61888777504377]
  Full-reference (FR) image quality assessment (IQA) evaluates the visual quality of a distorted image by measuring its perceptual difference with pristine-quality reference. Unlabeled data can be easily collected from an image degradation or restoration process, making it encouraging to exploit unlabeled training data to boost FR-IQA performance. In this paper, we suggest to incorporate semi-supervised and positive-unlabeled (PU) learning for exploiting unlabeled data while mitigating the adverse effect of outliers.
  arXiv  Detail & Related papers  (2022-04-19T09:10:06Z)
• Generative Modeling Helps Weak Supervision (and Vice Versa) [87.62271390571837]
  We propose a model fusing weak supervision and generative adversarial networks. It captures discrete variables in the data alongside the weak supervision derived label estimate. It is the first approach to enable data augmentation through weakly supervised synthetic images and pseudolabels.
  arXiv  Detail & Related papers  (2022-03-22T20:24:21Z)
• Streaming Machine Learning and Online Active Learning for Automated Visual Inspection [0.6299766708197884]
  We compare five streaming machine learning algorithms applied to visual defect inspection with real-world data provided by Philips Consumer Lifestyle BV. Our results show that active learning reduces the data labeling effort by almost 15% on average for the worst case. The use of machine learning models for automated visual inspection are expected to speed up the quality inspection up to 40%.
  arXiv  Detail & Related papers  (2021-10-15T09:39:04Z)
• Active Learning for Automated Visual Inspection of Manufactured Products [0.6299766708197884]
  We compare three active learning approaches and five machine learning algorithms applied to visual defect inspection with real-world data. Our results show that active learning reduces the data labeling effort without detriment to the models' performance.
  arXiv  Detail & Related papers  (2021-09-06T13:44:25Z)
• Negative Data Augmentation [127.28042046152954]
  We show that negative data augmentation samples provide information on the support of the data distribution. We introduce a new GAN training objective where we use NDA as an additional source of synthetic data for the discriminator. Empirically, models trained with our method achieve improved conditional/unconditional image generation along with improved anomaly detection capabilities.
  arXiv  Detail & Related papers  (2021-02-09T20:28:35Z)

This list is automatically generated from the titles and abstracts of the papers in this site.

This site does not guarantee the quality of this site (including all information) and is not responsible for any consequences.