arrow
返回

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

delete2023-10-01
delete13
PRE
AI
Y
Yuxuan Wang *
S
Sai‐Ho Chung
W
Waqar Ahmed Khan
T
Tianteng Wang
J
Jingjun Xu
DOI:10.1016/j.aei.2023.102205delete
delete原文链接
delete原文求助
delete分享
delete收藏
摘要

摘要

En 中文
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.
Keyword:
Industrial defect detection
Automatic data augmentation
Deep learning
Optimization

期刊

Advanced Engineering Informatics 封面图
Advanced Engineering Informatics
IF:
9.9
论文数:
4.4K
被引数:
1.7W

机构

H
hong kong polytechnic university
学者数:
3.0W
论文数: 4.1W
被引数: 921
C
City University of Hong Kong
学者数:
2.3W
论文数: 3.0W
被引数: 6.1W
U
University of Sharjah
学者数:
5.9K
论文数: 5.5K
被引数: 8.8K
学者 查看更多机构
引用论文

引用论文

A smart surface inspection system using faster R-CNN in cloud-edge computing environment
err2020-01-01
err87
errOAAI
errWang, Yuanbin; Liu, Minggao; Zheng, Pai; Yang, Huayong; Zou, Jun
err分享
err收藏
err分享
err收藏
Improved faster R-CNN for fabric defect detection based on Gabor filter with Genetic Algorithm optimization基于遗传算法优化Gabor滤波器的改进faster r-cnn织物疵点检测
err2022-01-01
err160
PREAI
errChen, Mengqi; Yu, Lingjie; Zhi, Chao; Sun, Runjun; Zhu, Shuangwu; Gao, Zhongyuan; Ke, Zhenxia; Zhu, Mengqiu; Zhang, Yuming
err分享
err收藏
学者 查看更多内容