高级检索
罗亚威, 于俊清. 可微风格搜索:一种在线自动数据增强方法[J]. 计算机辅助设计与图形学学报, 2023, 35(4): 553-561. DOI: 10.3724/SP.J.1089.2023.19348
引用本文: 罗亚威, 于俊清. 可微风格搜索:一种在线自动数据增强方法[J]. 计算机辅助设计与图形学学报, 2023, 35(4): 553-561. DOI: 10.3724/SP.J.1089.2023.19348
Luo Yawei, and Yu Junqing. Differentiable Style Searching: An Online Automatic Data Augmentation Method[J]. Journal of Computer-Aided Design & Computer Graphics, 2023, 35(4): 553-561. DOI: 10.3724/SP.J.1089.2023.19348
Citation: Luo Yawei, and Yu Junqing. Differentiable Style Searching: An Online Automatic Data Augmentation Method[J]. Journal of Computer-Aided Design & Computer Graphics, 2023, 35(4): 553-561. DOI: 10.3724/SP.J.1089.2023.19348

可微风格搜索:一种在线自动数据增强方法

Differentiable Style Searching: An Online Automatic Data Augmentation Method

  • 摘要: 针对现有数据增强方法中所存在的离线图像变换和搜索空间受限等问题,提出一种在线自动数据增强(ODA)方法.ODA的核心是一个可微风格搜索模块,该模块可以通过直接回传训练损失的方式感知任务模型当前所需的数据增强,以对抗搜索的方式在线生成更难的风格化图片以扩展训练集,从而高效地帮助模型完成在多种未知风格上的泛化.在跨领域图像分类任务的MNIST,MNIST-M,SVHN和USPS数据集以及跨领域场景语义分割任务的Cityscapes和GTA5数据集上,与其他5种典型的数据增强方法进行对比实验表明,在Acc指标上ODA方法能带来至少2%的分类精度提升,而在语义分割任务的mIoU指标上ODA方法能带来3%到7%的提升,证明了ODA扩充了传统自动数据增强方法在图像风格方向上的搜索空间,增强了网络的泛化能力.

     

    Abstract: Aiming at the problems of offline image transformation and limited search space in existing data enhancement methods, an online automatic data enhancement (ODA) method is proposed. The core of ODA is a micro-style search module, which can perceive the current data enhancement required by the task model by directly transmitting training losses, and generate more difficult stylized pictures online to expand the training set by means of anti-search, so as to effectively help the model complete the generalization on a variety of unknown styles. On the cross-domain image classification task of MNIST, MNIST-M, SVHN and USPS data sets, as well as the cross-domain scene semantic segmentation task of Cityscapes and GTA5 data sets, compared with other five typical data enhancement methods, ODA method can improve classification accuracy by at least 2% under Acc metric and 3% to 7% under mIoU metric of semantic segmentation task, which proves that ODA extends the search space of traditional automatic data enhancement methods in the direction of image style and enhances the generalization ability of network.

     

/

返回文章
返回