Semi-supervised learning for network-based cardiac MR image segmentation

Wenjia Bai, Ozan Oktay, Matthew Sinclair, Hideaki Suzuki, Martin Rajchl, Giacomo Tarroni, Ben Glocker, Andrew King, Paul M. Matthews, Daniel Rueckert

Research output: Chapter in Book/Report/Conference proceedingConference contribution

105 Citations (Scopus)

Abstract

Training a fully convolutional network for pixel-wise (or voxel-wise) image segmentation normally requires a large number of training images with corresponding ground truth label maps. However, it is a challenge to obtain such a large training set in the medical imaging domain, where expert annotations are time-consuming and difficult to obtain. In this paper, we propose a semi-supervised learning approach, in which a segmentation network is trained from both labelled and unlabelled data. The network parameters and the segmentations for the unlabelled data are alternately updated. We evaluate the method for short-axis cardiac MR image segmentation and it has demonstrated a high performance, outperforming a baseline supervised method. The mean Dice overlap metric is 0.92 for the left ventricular cavity, 0.85 for the myocardium and 0.89 for the right ventricular cavity. It also outperforms a state-of-the-art multi-atlas segmentation method by a large margin and the speed is substantially faster.

Original languageEnglish
Title of host publicationMedical Image Computing and Computer Assisted Intervention − MICCAI 2017 - 20th International Conference, Proceedings
EditorsPierre Jannin, Simon Duchesne, Maxime Descoteaux, Alfred Franz, D. Louis Collins, Lena Maier-Hein
PublisherSpringer Verlag
Pages253-260
Number of pages8
ISBN (Print)9783319661841
DOIs
Publication statusPublished - 2017
Externally publishedYes
Event20th International Conference on Medical Image Computing and Computer-Assisted Intervention, MICCAI 2017 - Quebec City, Canada
Duration: 2017 Sep 112017 Sep 13

Publication series

NameLecture Notes in Computer Science (including subseries Lecture Notes in Artificial Intelligence and Lecture Notes in Bioinformatics)
Volume10434 LNCS
ISSN (Print)0302-9743
ISSN (Electronic)1611-3349

Conference

Conference20th International Conference on Medical Image Computing and Computer-Assisted Intervention, MICCAI 2017
Country/TerritoryCanada
CityQuebec City
Period17/9/1117/9/13

ASJC Scopus subject areas

  • Theoretical Computer Science
  • Computer Science(all)

Fingerprint

Dive into the research topics of 'Semi-supervised learning for network-based cardiac MR image segmentation'. Together they form a unique fingerprint.

Cite this