A coarse-to-fine 3D U-net network for semantic segmentation of kidney CT scans

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

6 Citations (Scopus)

Abstract

The number of kidney cancer patients is increasing each year. Computed Tomography (CT) scans of the kidneys are useful to assess tumors and study tumor morphology. Semantic segmentation techniques enable the identification of kidney and surrounding anatomy on the pixel level. This allows clinicians to provide accurate treatment plans and improve efficiency. The large size of CT volumes poses challenges for deep segmentation methods as it cannot be accommodated on a single GPU in its original resolution. Downsampling CT scans influences the segmentation performance. In this paper, we present a coarse-to-fine cascaded network based on 3D U-Net architecture for semantic segmentation of kidney CT volumes into three classes kidney, tumor, and cyst. A two stage approach is implemented where a 3D U-Net model is first trained on downsampled CT volumes to delineate kidney region. This is followed by another 3D U-Net model which is trained using the full resolution images cropped around the areas of interest generated by first stage segmentation results. A set of 300 CT scans were used for training and evaluation. The proposed approach scored 0.9748, 0.8813, 0.8710 average dice for kidney, tumor and cyst using 3D cascade U-Net model. The performance of the cascade network outperformed other trained U-Net models based on 2D, 3D low resolution and 3D full resolution. The model also achieved the 3 rd place in the leaderboard of KiTS21 challenge with a mean sampled average dice score of 0.8944 and a mean sampled average surface dice score of 0.8140 using a test set of 100 CT scans.

Original languageEnglish
Title of host publicationMICCAI 2021 Challenge, KiTS 2021, Held in Conjunction with MICCAI 2021, Proceedings
EditorsNicholas Heller, Fabian Isensee, Darya Trofimova, Resha Tejpaul, Nikolaos Papanikolopoulos, Christopher Weight
Place of PublicationCham Switzerland
PublisherSpringer
Pages137-142
Number of pages6
ISBN (Electronic)9783030983857
ISBN (Print)9783030983840
DOIs
Publication statusPublished - 2022
EventInternational Challenge on Kidney and Kidney Tumor Segmentation 2021 - Strasbourg, France
Duration: 27 Sept 202127 Sept 2021
Conference number: 2nd
https://link.springer.com/book/10.1007/978-3-030-98385-7 (Proceedings)

Publication series

NameLecture Notes in Computer Science
PublisherSprringer
Volume13168
ISSN (Print)0302-9743
ISSN (Electronic)1611-3349

Conference

ConferenceInternational Challenge on Kidney and Kidney Tumor Segmentation 2021
Abbreviated titleKiTS 2021
Country/TerritoryFrance
CityStrasbourg
Period27/09/2127/09/21
Internet address

Keywords

  • 3D U-Net
  • Cascaded network
  • Medical image diagnostics
  • Semantic segmentation

Cite this