Exploiting confident information for weakly supervised prostate segmentation based on image-level labels

Zhang Chen; Zhiqiang Tian; Xiaojian Li; Yinshu Zhang; James D. Dormer

doi:10.1117/12.2549260

16 March 2020 Exploiting confident information for weakly supervised prostate segmentation based on image-level labels

Zhang Chen, Zhiqiang Tian, Xiaojian Li, Yinshu Zhang, James D. Dormer

Proceedings Volume 11315, Medical Imaging 2020: Image-Guided Procedures, Robotic Interventions, and Modeling; 1131523 (2020) https://doi.org/10.1117/12.2549260
Event: SPIE Medical Imaging, 2020, Houston, Texas, United States

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.

Citation Download Citation

Zhang Chen, Zhiqiang Tian, Xiaojian Li, Yinshu Zhang, and James D. Dormer "Exploiting confident information for weakly supervised prostate segmentation based on image-level labels", Proc. SPIE 11315, Medical Imaging 2020: Image-Guided Procedures, Robotic Interventions, and Modeling, 1131523 (16 March 2020); https://doi.org/10.1117/12.2549260

ACCESS THE FULL ARTICLE

INSTITUTIONAL
Select your institution to access the SPIE Digital Library.

SELECT YOUR INSTITUTION

PERSONAL
Sign in with your SPIE account to access your personal subscriptions or to use specific features such as save to my library, sign up for alerts, save searches, etc.

PERSONAL SIGN IN

No SPIE Account? Create one

PURCHASE THIS CONTENT

SUBSCRIBE TO DIGITAL LIBRARY

50 downloads per 1-year subscription

Members: $195

Non-members: $335 ADD TO CART

25 downloads per 1 - year subscription

Members: $145

Non-members: $250 ADD TO CART

PURCHASE SINGLE ARTICLE

Includes PDF, HTML & Video, when available