Towards increased trustworthiness of deep learning segmentation methods on cardiac MRI

Jorg Sander, Bob D. de Vos, Jelmer M. Wolterink, Ivana Išgum

Research output: Chapter in Book/Report/Conference proceedingConference contributionAcademicpeer-review

39 Citations (Scopus)

Abstract

Current state-of-the-art deep learning segmentation methods have not yet made a broad entrance into the clinical setting in spite of high demand for such automatic methods. One important reason is the lack of reliability caused by models that fail unnoticed and often locally produce anatomically implausible results that medical experts would not make. This paper presents an automatic image segmentation method based on (Bayesian) dilated convolutional networks (DCNN) that generate segmentation masks and spatial uncertainty maps for the input image at hand. The method was trained and evaluated using segmentation of the left ventricle (LV) cavity, right ventricle (RV) endocardium and myocardium (Myo) at end-diastole (ED) and end-systole (ES) in 100 cardiac 2D MR scans from the MICCAI 2017 Challenge (ACDC). Combining segmentations and uncertainty maps and employing a human-in-the-loop setting, we provide evidence that image areas indicated as highly uncertain, regarding the obtained segmentation, almost entirely cover regions of incorrect segmentations. The fused information can be harnessed to increase segmentation performance. Our results reveal that we can obtain valuable spatial uncertainty maps with low computational effort using DCNNs.
Original languageEnglish
Title of host publicationMedical Imaging 2019: Image Processing
Subtitle of host publicationImage Processing
EditorsBennett A. Landman, Elsa D. Angelini
PublisherSPIE
Volume10949
ISBN (Electronic)9781510625457
DOIs
Publication statusPublished - 2019
EventMedical Imaging 2019: Image Processing - San Diego, United States
Duration: 19 Feb 201921 Feb 2019

Publication series

NameProgress in Biomedical Optics and Imaging - Proceedings of SPIE

Conference

ConferenceMedical Imaging 2019: Image Processing
Country/TerritoryUnited States
CitySan Diego
Period19/02/201921/02/2019

Keywords

  • Bayesian neural networks
  • cardiac MRI segmentation
  • deep learning
  • loss functions
  • uncertainty estimation

Cite this