Pelvis segmentation using multi-pass U-Net and iterative shape estimation

Chunliang Wang*, Bryan Connolly, Pedro Filipe de Oliveira Lopes, Alejandro F. Frangi, Örjan Smedby

*Corresponding author for this work

Research output: Chapter in Book/Conference proceedingConference contributionpeer-review

Abstract

In this report, an automatic method for segmentation of the pelvis in three-dimensional (3D) computed tomography (CT) images is proposed. The method is based on a 3D U-net which has as input the 3D CT image and estimated volumetric shape models of the targeted structures and which returns the probability maps of each structure. During training, the 3D U-net is initially trained using blank shape context inputs to generate the segmentation masks, i.e. relying only on the image channel of the input. The preliminary segmentation results are used to estimate a new shape model, which is then fed to the same network again, with the input images. With the additional shape context information, the U-net is trained again to generate better segmentation results. During the testing phase, the input image is fed through the same 3D U-net multiple times, first with blank shape context channels and then with iteratively re-estimated shape models. Preliminary results show that the proposed multi-pass U-net with iterative shape estimation outperforms both 2D and 3D conventional U-nets without the shape model.

Original languageEnglish
Title of host publicationComputational Methods and Clinical Applications in Musculoskeletal Imaging - 6th International Workshop, MSKI 2018, Held in Conjunction with MICCAI 2018, Revised Selected Papers
EditorsGuoyan Zheng, Tomaž Vrtovec, Jianhua Yao, Jose M. Pozo
PublisherSpringer-Verlag Italia
Pages49-57
Number of pages9
ISBN (Print)9783030111656
DOIs
Publication statusPublished - 2019
Event6th International Workshop on Computational Methods and Clinical Applications in Musculoskeletal Imaging, MSKI 2018 was held in conjunction with Medical Image Computing and Computer-Assisted Intervention, MICCAI 2018 - Granada, Spain
Duration: 16 Sept 201820 Sept 2018

Publication series

NameLecture Notes in Computer Science (including subseries Lecture Notes in Artificial Intelligence and Lecture Notes in Bioinformatics)
Volume11404 LNCS
ISSN (Print)0302-9743
ISSN (Electronic)1611-3349

Conference

Conference6th International Workshop on Computational Methods and Clinical Applications in Musculoskeletal Imaging, MSKI 2018 was held in conjunction with Medical Image Computing and Computer-Assisted Intervention, MICCAI 2018
Country/TerritorySpain
CityGranada
Period16/09/1820/09/18

Keywords

  • Deep learning
  • Multi-pass U-net
  • Pelvis segmentation
  • Shape context
  • Statistic shape model

Fingerprint

Dive into the research topics of 'Pelvis segmentation using multi-pass U-Net and iterative shape estimation'. Together they form a unique fingerprint.

Cite this