arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2001.04086cs.CV

GridMask数据增强

GridMask Data Augmentation

The Chinese University of Hong Kong(香港中文大学);City University of Hong Kong(香港城市大学)

Pengguang Chen, Shu Liu, Hengshuang Zhao, Xingquan Wang, Jiaya Jia

AI总结:

本研究提出名为GridMask的结构化信息丢弃式数据增强方法,通过删除输入图像区域实现增强,在ImageNet、COCO2017、Cityscapes等多个视觉任务上性能优于高计算成本的AutoAugment,验证了其有效性与通用性。

摘要:

本文提出了一种名为GridMask的新型数据增强方法。该方法通过信息移除在多种计算机视觉任务中取得了当前最优结果。我们分析了信息丢弃的要求,随后指出现有信息丢弃算法的局限性,并提出了我们的结构化方法,该方法简单且十分有效,其核心是对输入图像的区域进行删除。我们的大量实验表明,该方法的性能优于最新的AutoAugment(自动增强),而AutoAugment因使用强化学习寻找最优策略,计算成本要高得多。在用于图像识别的ImageNet数据集、COCO2017目标检测任务以及用于语义分割的Cityscapes数据集上,我们的方法相比基线模型均显著提升了性能。大量实验证明了该新方法的有效性与通用性。

英文摘要:

We propose a novel data augmentation method `GridMask' in this paper. It utilizes information removal to achieve state-of-the-art results in a variety of computer vision tasks. We analyze the requirement of information dropping. Then we show limitation of existing information dropping algorithms and propose our structured method, which is simple and yet very effective. It is based on the deletion of regions of the input image. Our extensive experiments show that our method outperforms the latest AutoAugment, which is way more computationally expensive due to the use of reinforcement learning to find the best policies. On the ImageNet dataset for recognition, COCO2017 object detection, and on Cityscapes dataset for semantic segmentation, our method all notably improves performance over baselines. The extensive experiments manifest the effectiveness and generality of the new method.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑