HI-Net: Hyperdense Inception 3D UNet for Brain Tumor Segmentation

Saqib Qamar, Parvez Ahmad, Linlin Shen

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

17 Citations (Scopus)

Abstract

The brain tumor segmentation task aims to classify tissue into the whole tumor (WT), tumor core (TC) and enhancing tumor (ET) classes using multimodel MRI images. Quantitative analysis of brain tumors is critical for clinical decision making. While manual segmentation is tedious, time-consuming, and subjective, this task is at the same time very challenging to automatic segmentation methods. Thanks to the powerful learning ability, convolutional neural networks (CNNs), mainly fully convolutional networks, have shown promising brain tumor segmentation. This paper further boosts the performance of brain tumor segmentation by proposing hyperdense inception 3D UNet (HI-Net), which captures multi-scale information by stacking factorization of 3D weighted convolutional layers in the residual inception block. We use hyper dense connections among factorized convolutional layers to extract more contexual information, with the help of features reusability. We use a dice loss function to cope with class imbalances. We validate the proposed architecture on the multi-modal brain tumor segmentation challenges (BRATS) 2020 testing dataset. Preliminary results on the BRATS 2020 testing set show that achieved by our proposed approach, the dice (DSC) scores of ET, WT, and TC are 0.79457, 0.87494, and 0.83712, respectively.

Original languageEnglish
Title of host publicationBrainlesion
Subtitle of host publicationGlioma, Multiple Sclerosis, Stroke and Traumatic Brain Injuries - 6th International Workshop, BrainLes 2020, Held in Conjunction with MICCAI 2020, Revised Selected Papers
EditorsAlessandro Crimi, Spyridon Bakas
PublisherSpringer Science and Business Media Deutschland GmbH
Pages50-57
Number of pages8
ISBN (Print)9783030720865
DOIs
Publication statusPublished - 2021
Externally publishedYes
Event6th International MICCAI Brainlesion Workshop, BrainLes 2020 Held in Conjunction with 23rd Medical Image Computing for Computer Assisted Intervention Conference, MICCAI 2020 - Virtual, Online
Duration: 4 Oct 20204 Oct 2020

Publication series

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

Conference

Conference6th International MICCAI Brainlesion Workshop, BrainLes 2020 Held in Conjunction with 23rd Medical Image Computing for Computer Assisted Intervention Conference, MICCAI 2020
CityVirtual, Online
Period4/10/204/10/20

Keywords

  • 3D UNet
  • Brain tumor
  • Deep learning
  • Dense connections
  • Factorized convolutional

ASJC Scopus subject areas

  • Theoretical Computer Science
  • General Computer Science

Fingerprint

Dive into the research topics of 'HI-Net: Hyperdense Inception 3D UNet for Brain Tumor Segmentation'. Together they form a unique fingerprint.

Cite this