Graph-convolutional-network-based interactive prostate segmentation in MR images

Zhiqiang Tian, Xiaojian Li, Yaoyue Zheng, Zhang Chen, Zhong Shi, Lizhi Liu, Baowei Fei

Research output: Contribution to journalArticlepeer-review

3 Scopus citations

Abstract

Purpose: Accurate and robust segmentation of the prostate from magnetic resonance (MR) images is extensively applied in many clinical applications in prostate cancer diagnosis and treatment. The purpose of this study is the development of a robust interactive segmentation method for accurate segmentation of the prostate from MR images. Methods: We propose an interactive segmentation method based on a graph convolutional network (GCN) to refine the automatically segmented results. An atrous multiscale convolutional neural network (CNN) encoder is proposed to learn representative features to obtain accurate segmentations. Based on the multiscale feature, a GCN block is presented to predict the prostate contour in both automatic and interactive manners. To preserve the prostate boundary details and effectively train the GCN, a contour matching loss is proposed. The performance of the proposed algorithm was evaluated on 41 in-house MR subjects and 30 PROMISE12 test subjects. Result: The proposed method yields mean Dice similarity coefficients of 93.8 ± 1.2% and 94.4 ± 1.0% on our in-house and PROMISE12 datasets, respectively. The experimental results show that the proposed method outperforms several state-of-the-art segmentation methods. Conclusion: The proposed interactive segmentation method based on the GCN can accurately segment the prostate from MR images. Our method has a variety of applications in prostate cancer imaging.

Original languageEnglish (US)
Pages (from-to)4164-4176
Number of pages13
JournalMedical physics
Volume47
Issue number9
DOIs
StatePublished - Sep 1 2020

Keywords

  • graph convolutional network
  • interactive segmentation
  • prostate MR image

ASJC Scopus subject areas

  • Biophysics
  • Radiology Nuclear Medicine and imaging

Fingerprint

Dive into the research topics of 'Graph-convolutional-network-based interactive prostate segmentation in MR images'. Together they form a unique fingerprint.

Cite this