Learning an augmentation strategy for sparse datasets

doi:10.1016/j.imavis.2021.104338

Learning an augmentation strategy for sparse datasets

2022-01-01

11

PRE

AI

DOI：10.1016/j.imavis.2021.104338

原文链接

原文求助

摘要

En 中文

The limited quantity of training data can hamper supervised machine learning methods, that generally need large amounts of data to avoid overfitting. Data augmentation has a long history of use with machine learning algorithms and is a straightforward method to overcome overfitting and improve model generalisation. However, data augmentation schemes are typically designed by hand and demand substantial domain knowledge to create suitable data transformations. This paper introduces a GAN based method that automatically learns an augmentation strategy appropriate for sparse datasets and can improve pixel-level semantic segmentation accuracy by filling the gaps in the training set. Our method can also be combined with other augmentation techniques to further improve performance. We evaluate the proposed method's feasibility on four datasets and three semantic segmentation models, leading to improvement in the mean intersection-over-union (mIoU) score of between 0.5 and 14 percentage points, under different circumstances. (c) 2021 Elsevier B.V. All rights reserved.

Keyword:

GAN

Data augmentation

Semantic segmentation

Learning an augmentation strategy for sparse datasets

摘要

期刊

机构