Automatic 3D cardiovascular MR segmentation with densely-connected volumetric convnets

Lequan Yu, Jie Zhi Cheng, Qi Dou, Xin Yang, Hao Chen, Jing Qin, Pheng Ann Heng

Research output: Chapter in book / Conference proceedingConference article published in proceeding or bookAcademic researchpeer-review

72 Citations (Scopus)

Abstract

Automatic and accurate whole-heart and great vessel segmentation from 3D cardiac magnetic resonance (MR) images plays an important role in the computer-assisted diagnosis and treatment of cardiovascular disease. However, this task is very challenging due to ambiguous cardiac borders and large anatomical variations among different subjects. In this paper, we propose a novel densely-connected volumetric convolutional neural network, referred as DenseVoxNet, to automatically segment the cardiac and vascular structures from 3D cardiac MR images. The DenseVoxNet adopts the 3D fully convolutional architecture for effective volume-to-volume prediction. From the learning perspective, our DenseVoxNet has three compelling advantages. First, it preserves the maximum information flow between layers by a densely-connected mechanism and hence eases the network training. Second, it avoids learning redundant feature maps by encouraging feature reuse and hence requires fewer parameters to achieve high performance, which is essential for medical applications with limited training data. Third, we add auxiliary side paths to strengthen the gradient propagation and stabilize the learning process. We demonstrate the effectiveness of DenseVoxNet by comparing it with the state-of-the-art approaches from HVSMR 2016 challenge in conjunction with MICCAI, and our network achieves the best dice coefficient. We also show that our network can achieve better performance than other 3D ConvNets but with fewer parameters.
Original languageEnglish
Title of host publicationMedical Image Computing and Computer Assisted Intervention − MICCAI 2017 - 20th International Conference, Proceedings
PublisherSpringer Verlag
Pages287-295
Number of pages9
ISBN (Print)9783319661841
DOIs
Publication statusPublished - 1 Jan 2017
Event20th International Conference on Medical Image Computing and Computer-Assisted Intervention, MICCAI 2017 - Quebec City, Canada
Duration: 11 Sep 201713 Sep 2017

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
Period11/09/1713/09/17

ASJC Scopus subject areas

  • Theoretical Computer Science
  • Computer Science(all)

Cite this