Densely Dilated Spatial Pooling Convolutional Network Using Benign Loss Functions for Imbalanced Volumetric Prostate Segmentation

Page: [788 - 799] Pages: 12

  • * (Excluding Mailing and Handling)

Abstract

Background: The high incidence rate of prostate disease poses a requirement of accurate early detection. Magnetic Resonance Imaging (MRI) is one of the main imaging methods used for prostate cancer detection so far, but it has problems of imbalance and variation in appearance, therefore, automated prostate segmentation is still challenging.

Objective: Aiming to accurately segment the prostate from MRI, the focus was on designing a unique network with benign loss functions.

Methods: A novel Densely Dilated Spatial Pooling Convolutional Network (DDSP ConNet) in an encoderdecoder structure, with a unique DDSP block was proposed. By densely combining dilated convolution and global pooling layers, the DDSP block supplies coarse segmentation results and preserves hierarchical contextual information. Meanwhile, the DSC and Jaccard loss were adopted to train the DDSP ConNet. And it was proved theoretically that they have benign properties, including symmetry, continuity, and differentiability on the parameters of the network.

Results: Extensive experiments have been conducted to corroborate the effectiveness of the DDSP ConNet with DSC and Jaccard loss on the MICCAI PROMISE12 challenge dataset. In the test dataset, the DDSP ConNet achieved a score of 85.78.

Conclusion: In the conducted experiments, DDSP network with DSC and Jaccard loss outperformed most of the other competitors on the PROMISE12 dataset. Therefore, it has a better ability to extract hierarchical features and solve the imbalanced medical image problem.

Keywords: Prostate segmentation, imbalanced medical images, DDSP conNet, dilated convolution, global pooling layer, densely connection, DSC loss function, Jaccard loss function.

Graphical Abstract