arrow
返回

aD b

delete2023-05-01
delete13
PRE
AI
S
Shuanlong Niu
Y
Yaru Peng
李彬 封面图
李彬 (Bin Li) *
X
Xinggang Wang
DOI:10.1016/j.compind.2023.103860delete
delete原文链接
delete原文求助
delete分享
delete收藏
摘要

摘要

En 中文
Data augmentation is widely used in convolutional neural network (CNN) models to improve the performance of downstream tasks. The images generated by traditional data augmentation methods are usually random and can struggle to overcome the range limitation of the sampled data distribution, which can result in meaningless augmented data and poor diversity. As a result, a novel data augmentation algorithm is proposed to generate images that are outside the scope of the sampled data distribution along the feature direction with diversity weights. First, an image is mapped to the latent feature space based on the encoder. Then, the range loss function can restrict the latent variables within a hypersphere of radius k. The feature directions and diversity weights can be found based on the entropy variation in the latent feature space. During the CNN model's training process, probabilities are constructed based on diversity weights to select the feature direction. Then, the latent variable of the input image is transformed in the feature direction to the region beyond the hypersphere. Finally, the transformed latent variable is mapped to the pixel space based on the decoder to generate images outside the sampled data space to improve diversity. Experimentally, the proposed method considerably improves the performance of defect segmentation in different industrial scenes.
Keyword:
Data augmentation
Deep latent feature
Out-of-distribution
Generative adversarial network (GAN)

期刊

Computers in Industry 封面图
Computers in Industry
IF:
9.1
论文数:
2.9K
被引数:
1.1W

机构

暂无机构信息
引用论文

引用论文

Segmentation-based deep-learning approach for surface-defect detection
err2019-05-15
err544
errOAAI
errTabernik, Domen; Sela, Samo; Skvarc, Jure; Skocaj, Danijel
err分享
err收藏
ImageNet Large Scale Visual Recognition ChallengeImageNet大规模视觉识别挑战
err2015-04-11
err2.7W
PREAI
errRussakovsky, Olga; Deng, Jia; Su, Hao; Krause, Jonathan; Satheesh, Sanjeev; Ma, Sean; Huang, Zhiheng; Karpathy, Andrej; Khosla, Aditya; Bernstein, Michael; Berg, Alexander C.; Fei-Fei, Li
err分享
err收藏
学者 查看更多内容