Brain Tumor Segmentation with Patch-Based 3D Attention UNet from Multi-parametric MRI

Xue Feng, Harrison Bai, Daniel Kim, Georgios Maragkos, Jan Machaj, Ryan Kellogg

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

Abstract

Accurate segmentation of different sub-regions of gliomas including peritumoral edema, necrotic core, enhancing and non-enhancing tumor core from multiparametric MRI scans has important clinical relevance in diagnosis, prognosis and treatment of brain tumors. However, due to the highly heterogeneous appearance and shape, segmentation of the sub-regions is very challenging. Recent development using deep learning models has proved its effectiveness in the past several brain segmentation challenges as well as other semantic and medical image segmentation problems. In this paper we developed a deep-learning-based segmentation method using a patch-based 3D UNet with the attention block. Hyper-parameters tuning and training and testing augmentations were applied to increase the model performance. Preliminary results showed effectiveness of the segmentation model and achieved mean Dice scores of 0.806 (ET), 0.863 (TC) and 0.918 (WT) in the validation dataset.

Original languageEnglish (US)
Title of host publicationBrainlesion
Subtitle of host publicationGlioma, Multiple Sclerosis, Stroke and Traumatic Brain Injuries - 7th International Workshop, BrainLes 2021, Held in Conjunction with MICCAI 2021, Revised Selected Papers
EditorsAlessandro Crimi, Spyridon Bakas
PublisherSpringer Science and Business Media Deutschland GmbH
Pages90-96
Number of pages7
ISBN (Print)9783031090011
DOIs
StatePublished - 2022
Event7th International Brain Lesion Workshop, BrainLes 2021, held in conjunction with the Medical Image Computing and Computer Assisted Intervention, MICCAI 2021 - Virtual, Online
Duration: Sep 27 2021Sep 27 2021

Publication series

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

Conference

Conference7th International Brain Lesion Workshop, BrainLes 2021, held in conjunction with the Medical Image Computing and Computer Assisted Intervention, MICCAI 2021
CityVirtual, Online
Period9/27/219/27/21

Keywords

  • 3D U-Net
  • Attention block
  • Brain tumor segmentation
  • Deep learning

ASJC Scopus subject areas

  • Theoretical Computer Science
  • General Computer Science

Fingerprint

Dive into the research topics of 'Brain Tumor Segmentation with Patch-Based 3D Attention UNet from Multi-parametric MRI'. Together they form a unique fingerprint.

Cite this