A data augmentation approach to train fully convolutional networks for left ventricle segmentation

Magn Reson Imaging. 2020 Feb:66:152-164. doi: 10.1016/j.mri.2019.08.004. Epub 2019 Aug 30.

Abstract

Left ventricle (LV) segmentation plays an important role in the diagnosis of cardiovascular diseases. The cardiac contractile function can be quantified by measuring the segmentation results of LVs. Fully convolutional networks (FCNs) have been proven to be able to segment images. However, a large number of annotated images are required to train the network to avoid overfitting, which is a challenge for LV segmentation owing to the limited small number of available training samples. In this paper, we analyze the influence of augmenting training samples used in an FCN for LV segmentation, and propose a data augmentation approach based on shape models to train the FCN from a few samples. We show that the balanced training samples affect the performance of FCNs greatly. Experiments on four public datasets demonstrate that the FCN trained by our augmented data outperforms most existing automated segmentation methods with respect to several commonly used evaluation measures.

Keywords: Convolutional neural network; Data augmentation; Image segmentation; Shape models.

Publication types

  • Research Support, Non-U.S. Gov't

MeSH terms

  • Datasets as Topic
  • Heart Ventricles / anatomy & histology
  • Heart Ventricles / diagnostic imaging*
  • Humans
  • Image Processing, Computer-Assisted / methods*
  • Magnetic Resonance Imaging / methods*
  • Neural Networks, Computer*