MS UNet: Multi-scale 3D UNet for Brain Tumor Segmentation

Parvez Ahmad, Saqib Qamar, Linlin Shen, Syed Qasim Afser Rizvi, Aamir Ali, Girija Chetty

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

7 Citations (Scopus)

Abstract

A deep convolutional neural network (CNN) achieves remarkable performance for medical image analysis. UNet is the primary source in the performance of 3D CNN architectures for medical imaging tasks, including brain tumor segmentation. The skip connection in the UNet architecture concatenates multi-scale features from image data. The multi-scaled features play an essential role in brain tumor segmentation. Researchers presented numerous multi-scale strategies that have been excellent for the segmentation task. This paper proposes a multi-scale strategy that can further improve the final segmentation accuracy. We propose three multi-scale strategies in MS UNet. Firstly, we utilize densely connected blocks in the encoder and decoder for multi-scale features. Next, the proposed residual-inception blocks extract local and global information by merging features of different kernel sizes. Lastly, we utilize the idea of deep supervision for multiple depths at the decoder. We validate the MS UNet on the BraTS 2021 validation dataset. The dice (DSC) scores of the whole tumor (WT), tumor core (TC), and enhancing tumor (ET) are 91.938 %, 86.268 %, and 82.409 %, respectively.

Original languageEnglish
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
Pages30-41
Number of pages12
ISBN (Print)9783031090011
DOIs
Publication statusPublished - 2022
Externally publishedYes
Event7th International Brain Lesion Workshop, BrainLes 2021, held in conjunction with the Medical Image Computing and Computer Assisted Intervention, MICCAI 2021 - Virtual, Online
Duration: 27 Sept 202127 Sept 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
Period27/09/2127/09/21

Keywords

  • Brain tumor segmentation
  • CNN
  • Contextual information
  • Dense connections
  • Residual inception blocks
  • UNet

ASJC Scopus subject areas

  • Theoretical Computer Science
  • General Computer Science

Fingerprint

Dive into the research topics of 'MS UNet: Multi-scale 3D UNet for Brain Tumor Segmentation'. Together they form a unique fingerprint.

Cite this