MeshCut data augmentation for deep learning in computer vision.

Journal: PloS one
Published Date:

Abstract

To solve overfitting in machine learning, we propose a novel data augmentation method called MeshCut, which uses a mesh-like mask to segment the whole image to achieve more partial diversified information. In our experiments, this strategy outperformed the existing augmentation strategies and achieved state-of-the-art results in a variety of computer vision tasks. MeshCut is also an easy-to-implement strategy that can efficiently improve the performance of the existing convolutional neural network models by a good margin without careful hand-tuning. The performance of such a strategy can be further improved by incorporating it into other augmentation strategies, which can make MeshCut a promising baseline strategy for future data augmentation algorithms.

Authors

  • Wei Jiang
    Department of Civil Engineering, Johns Hopkins System Institute, Johns Hopkins University, Baltimore, Maryland.
  • Kai Zhang
    Anhui Province Key Laboratory of Respiratory Tumor and Infectious Disease, First Affiliated Hospital of Bengbu Medical University, Bengbu, China.
  • Nan Wang
    Department of Gastroenterology, The First Affiliated Hospital of Harbin Medical University, Harbin, Heilongjiang, China.
  • Miao Yu
    Key Laboratory of Bioactive Substances and Resource Utilization of Chinese Herbal Medicine, Ministry of Education, China Institute of Medicinal Plant Development, Chinese Academy of Medical Sciences & Peking Union Medical College, Beijing, 100193, China; School of Chinese Materia Medica, Guangdong Pharmaceutical University, Guangzhou, 510006, China; Faculty of Arts and Sciences, Beijing Normal University, Zhuhai, 519087, China.