Abstract:Industrial defect detection is a critical and challenging task in the quality control of manufacturing production. Competent in feature extraction and pattern recognition, deep learning shows great power for classifying and locating defects in industrial products. However, insufficient data records and diverse categories of defects restrict the detection accuracy of data-driven neural networks . One direction to solve such a problem is data augmentation, which aims at generating synthetic copies from existing data and improving the generalizability of detection models. However, confirming a suitable augmentation policy involves either human experience or substantial experiments for augmentation parameters. To address these challenges, in this work, a lite automatic data augmentation (ALADA) framework is proposed to jointly optimize the data augmentation policies and the neural network for industrial defect detection. First, a lite search space is formulated to efficiently sample augmentation policies and generate augmented images for joint optimization. To reduce the hyperparameter tuning efforts for retraining with searched policies, a three-step bi-level optimization scheme is proposed to replace the retraining strategy and update the model and augmentation parameter alternately. To solve the non-differentiable problem in the joint optimization scheme, policy gradient sampling is implemented to estimate the gradient flow efficiently. Experimental results on three industrial defect detection datasets, namely, Tianchi-TILE, GC10-DET, and NEU-DET, reveal that our proposed automatic augmentation framework outperforms the state-of-the-art augmentation methods and effectively improves the accuracy of the baseline defect detection model. The proposed ALADA scheme also alleviates the missed detection of defects in four practical industrial circumstances: textured background, uneven brightness, low contrast, and intraclass difference.

AdaAug+: A Reinforcement Learning-Based Adaptive Data Augmentation for Change Detection

AdaAug: an Adaptive Data Augmentation Method for Change Detection

AdaAugment: A Tuning-Free and Adaptive Approach to Enhance Data Augmentation

SelectAugment: Hierarchical Deterministic Sample Selection for Data Augmentation

Universal Adaptive Data Augmentation

CSDACD: Domain-adaptive Change Detection Network for Cross-seasonal Remote Sensing Images

Background-Mixed Augmentation for Weakly Supervised Change Detection

Data Augmentation Using Image Generation for Change Detection

Adaptive Spatial and Difference Learning for Change Detection

Adversarial Instance Augmentation for Building Change Detection in Remote Sensing Images

Data augmentation and few-shot change detection in forest remote sensing

ALADA: A lite automatic data augmentation framework for industrial defect detection

Local Patch AutoAugment with Multi-Agent Collaboration

A domain adaptation neural network for change detection with heterogeneous optical and SAR remote sensing images

DADA: Differentiable Automatic Data Augmentation

Adaptive Data Augmentation for Contrastive Learning

Difference-Guided Aggregation Network with Multiimage Pixel Contrast for Change Detection.

A New Remote Sensing Change Detection Data Augmentation Method Based on Mosaic Simulation and Haze Image Simulation

Learning Better with Less: Effective Augmentation for Sample-Efficient Visual Reinforcement Learning

Domain Adaptive and Interactive Differential Attention Network for Remote Sensing Image Change Detection

Enhancing Monocular 3-D Object Detection Through Data Augmentation Strategies