Memory-Efficient Automatic Kidney and Tumor Segmentation Based on Non-local Context Guided 3D U-Net

Zhuoying Li, Junquan Pan, Huisi Wu, Zhenkun Wen, Jing Qin

Research output: Chapter in book / Conference proceedingConference article published in proceeding or bookAcademic researchpeer-review

8 Citations (Scopus)

Abstract

Automatic kidney and tumor segmentation from CT volumes is essential for clinical diagnosis and surgery planning. However, it is still a very challenging problem as kidney and tumor usually exhibit various scales, irregular shapes and blurred contours. In this paper, we propose a memory efficient automatic kidney and tumor segmentation algorithm based on non-local context guided 3D U-Net. Different from the traditional 3D U-Net, we implement a lightweight 3D U-Net with depthwise separable convolution (DSC), which can not only avoid over fitting but also improve the generalization ability. By encoding long range pixel-wise dependencies in features and recalibrating the weight of channels, we also develop a non-local context guided mechanism to capture global context and fully utilize the long range dependencies during the feature selection. Thanks to the non-local context guidance (NCG), we can successfully complement high-level semantic information with the spatial information simply based on a skip connection between encoder and decoder in the 3D U-Net, and finally realize a more accurate 3D kidney and tumor segmentation network. Our proposed method was validated and evaluated with KiTS dataset, including various 3D kidney and tumor patient cases. Convincing visual and statistical results verified effectiveness of our method. Comparisons with state-of-the-art methods were also conducted to demonstrate its advantages in terms of both efficiency and accuracy.

Original languageEnglish
Title of host publicationMedical Image Computing and Computer Assisted Intervention – MICCAI 2020 - 23rd International Conference, Proceedings
EditorsAnne L. Martel, Purang Abolmaesumi, Danail Stoyanov, Diana Mateus, Maria A. Zuluaga, S. Kevin Zhou, Daniel Racoceanu, Leo Joskowicz
PublisherSpringer Science and Business Media Deutschland GmbH
Pages197-206
Number of pages10
ISBN (Print)9783030597184
DOIs
Publication statusPublished - Oct 2020
Event23rd International Conference on Medical Image Computing and Computer-Assisted Intervention, MICCAI 2020 - Lima, Peru
Duration: 4 Oct 20208 Oct 2020

Publication series

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

Conference

Conference23rd International Conference on Medical Image Computing and Computer-Assisted Intervention, MICCAI 2020
Country/TerritoryPeru
CityLima
Period4/10/208/10/20

Keywords

  • Kidney and tumor segmentation
  • Memory-efficient network
  • Non-local context guided network
  • U-Net

ASJC Scopus subject areas

  • Theoretical Computer Science
  • Computer Science(all)

Fingerprint

Dive into the research topics of 'Memory-Efficient Automatic Kidney and Tumor Segmentation Based on Non-local Context Guided 3D U-Net'. Together they form a unique fingerprint.

Cite this