Title
Bayesian VoxDRN: A Probabilistic Deep Voxelwise Dilated Residual Network for Whole Heart Segmentation from 3D MR Images.
Abstract
In this paper, we propose a probabilistic deep voxelwise dilated residual network, referred as Bayesian VoxDRN, to segment the whole heart from 3D MR images. Bayesian VoxDRN can predict voxelwise class labels with a measure of model uncertainty, which is achieved by a dropout-based Monte Carlo sampling during testing to generate a posterior distribution of the voxel class labels. Our method has three compelling advantages. First, the dropout mechanism encourages the model to learn a distribution of weights with better data-explanation ability and prevents over-fitting. Second, focal loss and Dice loss are well encapsulated into a complementary learning objective to segment both hard and easy classes. Third, an iterative switch training strategy is introduced to alternatively optimize a binary segmentation task and a multi-class segmentation task for a further accuracy improvement. Experiments on the MICCAI 2017 multi-modality whole heart segmentation challenge data corroborate the effectiveness of the proposed method.
Year
DOI
Venue
2018
10.1007/978-3-030-00937-3_65
Lecture Notes in Computer Science
Field
DocType
Volume
Voxel,Computer vision,Residual,Monte Carlo method,Pattern recognition,Computer science,Segmentation,Posterior probability,Artificial intelligence,Probabilistic logic,Dice,Bayesian probability
Conference
11073
ISSN
Citations 
PageRank 
0302-9743
4
0.40
References 
Authors
15
7
Name
Order
Citations
PageRank
Zenglin Shi140.40
Guodong Zeng2275.28
Le Zhang326832.16
Xiahai Zhuang441138.76
Lei Li5335.37
Guang Yang6155.73
Guoyan Zheng736356.10