3d u-net based brain tumor segmentation and survival days prediction

Feifan Wang, Runzhou Jiang, Liqin Zheng, Chun Meng, Bharat Biswal

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

84 Scopus citations

Abstract

Past few years have witnessed the prevalence of deep learning in many application scenarios, among which is medical image processing. Diagnosis and treatment of brain tumors requires an accurate and reliable segmentation of brain tumors as a prerequisite. However, such work conventionally requires brain surgeons significant amount of time. Computer vision techniques could provide surgeons a relief from the tedious marking procedure. In this paper, a 3D U-net based deep learning model has been trained with the help of brain-wise normalization and patching strategies for the brain tumor segmentation task in the BraTS 2019 competition. Dice coefficients for enhancing tumor, tumor core, and the whole tumor are 0.737, 0.807 and 0.894 respectively on the validation dataset. These three values on the test dataset are 0.778, 0.798 and 0.852. Furthermore, numerical features including ratio of tumor size to brain size and the area of tumor surface as well as age of subjects are extracted from predicted tumor labels and have been used for the overall survival days prediction task. The accuracy could be 0.448 on the validation dataset, and 0.551 on the final test dataset.

Original languageEnglish (US)
Title of host publicationBrainlesion
Subtitle of host publicationGlioma, Multiple Sclerosis, Stroke and Traumatic Brain Injuries - 5th International Workshop, BrainLes 2019, Held in Conjunction with MICCAI 2019, Revised Selected Papers
EditorsAlessandro Crimi, Spyridon Bakas
PublisherSpringer
Pages131-141
Number of pages11
ISBN (Print)9783030466398
DOIs
StatePublished - 2020
Event5th International MICCAI Brainlesion Workshop, BrainLes 2019, held in conjunction with the Medical Image Computing for Computer Assisted Intervention, MICCAI 2019 - Shenzhen, China
Duration: Oct 17 2019Oct 17 2019

Publication series

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

Conference

Conference5th International MICCAI Brainlesion Workshop, BrainLes 2019, held in conjunction with the Medical Image Computing for Computer Assisted Intervention, MICCAI 2019
Country/TerritoryChina
CityShenzhen
Period10/17/1910/17/19

All Science Journal Classification (ASJC) codes

  • Theoretical Computer Science
  • General Computer Science

Keywords

  • 3D U-Net
  • Brain tumor segmentation
  • Survival days prediction

Fingerprint

Dive into the research topics of '3d u-net based brain tumor segmentation and survival days prediction'. Together they form a unique fingerprint.

Cite this