Share Email Print

Proceedings Paper

Exploiting confident information for weakly supervised prostate segmentation based on image-level labels
Format Member Price Non-Member Price
PDF $17.00 $21.00

Paper Abstract

Prostate segmentation on magnetic resonance images (MRI) is an important step for prostate cancer diagnosis and therapy. After the birth of deep convolution neural network (DCNN), prostate segmentation has achieved great success in supervised segmentation. However, these works are mostly based on abundant fully labeled pixel-level image data. In this work, we propose a weakly supervised prostate segmentation (WS-PS) method based on image-level labels. Although the image-level label is not sufficient for an exact prostate contour, it contains potential information which is helpful to make sure a coarse contour. This information is referred to confident information in this paper. Our WS-PS method includes two steps which are mask generation and prostate segmentation. First, the mask generation (MG) exploits a class activation maps (CAM) technique to generate a coarse probability map for MRI slices based on image-level label. These elements of the coarse map which have higher probability are considered to contain more confident information. To make use of confident information from coarse probability map, a similarity model (S-Model) is introduced to refine the coarse map. Second, the prostate segmentation (PS) uses a residual U-Net with a size constraint loss to segment prostate based on the refined mask obtained from MG. The proposed method achieves a mean Dice similarity coefficient (DSC) of 83.39% as compared to the manually delineated ground-truth. The experimental results indicate that our weakly supervised method can achieve a satisfactory segmentation on prostate MRI only with image-level labels.

Paper Details

Date Published: 16 March 2020
PDF: 7 pages
Proc. SPIE 11315, Medical Imaging 2020: Image-Guided Procedures, Robotic Interventions, and Modeling, 1131523 (16 March 2020); doi: 10.1117/12.2549260
Show Author Affiliations
Zhang Chen, Xi'an Jiaotong Univ. (China)
Zhiqiang Tian, Xi'an Jiaotong Univ. (China)
Xiaojian Li, Xi'an Jiaotong Univ. (China)
Yinshu Zhang, Xi'an Jiaotong Univ. (China)
James D. Dormer, The Univ. of Texas at Dallas (United States)

Published in SPIE Proceedings Vol. 11315:
Medical Imaging 2020: Image-Guided Procedures, Robotic Interventions, and Modeling
Baowei Fei; Cristian A. Linte, Editor(s)

© SPIE. Terms of Use
Back to Top
Sign in to read the full article
Create a free SPIE account to get access to
premium articles and original research
Forgot your username?