Semi-supervised learning for pelvic MR image segmentation based on multi-task residual fully convolutional networks

Zishun Feng, Dong Nie, Li Wang, Dinggang Shen

Research output: Chapter in Book/Report/Conference proceedingConference contribution

9 Citations (Scopus)

Abstract

Accurate segmentation of pelvic organs from magnetic resonance (MR) images plays an important role in image-guided radiotherapy. However, it is a challenging task due to inconsistent organ appearances and large shape variations. Fully convolutional network (FCN) has recently achieved state-of-the-art performance in medical image segmentation, but it requires a large amount of labeled data for training, which is usually difficult to obtain in real situation. To address these challenges, we propose a deep learning based semi-supervised learning framework. Specifically, we first train an initial multi-task residual fully convolutional network (FCN) based on a limited number of labeled MRI data. Based on the initially trained FCN, those unlabeled new data can be automatically segmented and some reasonable segmentations (after manual/automatic checking) can be included into the training data to fine-tune the network. This step can be repeated to progressively improve the training of our network, until no reasonable segmentations of new data can be included. Experimental results demonstrate the effectiveness of our proposed progressive semi-supervised learning fashion as well as its advantage in terms of accuracy.

Original languageEnglish
Title of host publication2018 IEEE 15th International Symposium on Biomedical Imaging, ISBI 2018
PublisherIEEE Computer Society
Pages885-888
Number of pages4
Volume2018-April
ISBN (Electronic)9781538636367
DOIs
Publication statusPublished - 2018 May 23
Externally publishedYes
Event15th IEEE International Symposium on Biomedical Imaging, ISBI 2018 - Washington, United States
Duration: 2018 Apr 42018 Apr 7

Other

Other15th IEEE International Symposium on Biomedical Imaging, ISBI 2018
CountryUnited States
CityWashington
Period18/4/418/4/7

Keywords

  • Fully convolutional network
  • Multi-task learning
  • Neural network
  • Pelvic MRI segmentation
  • Semi-supervised learning

ASJC Scopus subject areas

  • Biomedical Engineering
  • Radiology Nuclear Medicine and imaging

Fingerprint Dive into the research topics of 'Semi-supervised learning for pelvic MR image segmentation based on multi-task residual fully convolutional networks'. Together they form a unique fingerprint.

  • Cite this

    Feng, Z., Nie, D., Wang, L., & Shen, D. (2018). Semi-supervised learning for pelvic MR image segmentation based on multi-task residual fully convolutional networks. In 2018 IEEE 15th International Symposium on Biomedical Imaging, ISBI 2018 (Vol. 2018-April, pp. 885-888). IEEE Computer Society. https://doi.org/10.1109/ISBI.2018.8363713