Orthogonal-nets: a large ensemble of 2D neural networks for 3D brain tumor segmentation

Kamlesh Pawar, Shenjun Zhong, Dilshan Sasanka Goonatillake, Gary Egan, Zhaolin Chen

Research output: Chapter in Book/Report/Conference proceedingConference PaperResearchpeer-review

2 Citations (Scopus)

Abstract

We propose Orthogonal-Nets consisting of a large number of ensembles of 2D encoder-decoder convolutional neural networks. The Orthogonal-Nets takes 2D slices of the image from axial, sagittal, and coronal views of the 3D brain volume and predicts the probability for the tumor segmentation region. The predicted probability distributions from all three views are averaged to generate a 3D probability distribution map that is subsequently used to predict the tumor regions for the 3D images. In this work, we propose a two-stage Orthogonal-Nets. Stage-I predicts the brain tumor labels for the whole 3D image using the axial, sagittal, and coronal views. The labels from the first stage are then used to crop only the tumor region. Multiple Orthogonal-Nets were then trained in stage-II, which takes only the cropped region as input. The two-stage strategy substantially reduces the computational burden on the stage-II networks and thus many Orthogonal-Nets can be used in stage-II. We used one Orthogonal-Net for stage-I and 28 Orthogonal-Nets for stage-II. The mean dice score on the testing datasets was 0.8660, 0.8776, 0.9118 for enhancing tumor, core tumor, and whole tumor respectively.

Original languageEnglish
Title of host publicationBrainlesion
Subtitle of host publicationMultiple Sclerosis, Stroke and Traumatic Brain Injuries
EditorsAlessandro Crimi, Spyridon Bakas
Place of PublicationCham Switzerland
PublisherSpringer
Pages54-67
Number of pages14
Edition1st
ISBN (Electronic)9783031090028
ISBN (Print)9783031090011
DOIs
Publication statusPublished - 2022
EventInternational MICCAI Brain Lesion Workshop 2021 - Online, United States of America
Duration: 27 Sept 202127 Sept 2021
Conference number: 7th
https://link.springer.com/book/10.1007/978-3-031-08999-2 (Proceedings)

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

ConferenceInternational MICCAI Brain Lesion Workshop 2021
Abbreviated titleBrainLes 2021
Country/TerritoryUnited States of America
Period27/09/2127/09/21
Otherheld in conjunction with the Medical Image Computing for Computer Assisted Intervention, MICCAI 2021
Internet address

Keywords

  • Brain tumor segmentation
  • Convolutional neural network
  • Medical imaging

Cite this