Skip to main navigation Skip to search Skip to main content

Brain tumor segmentation using two-stage convolutional neural network for federated evaluation

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

Abstract

A deep learning method is proposed for brain tumor segmentation using a two-stage encoder-decoder convolutional neural network (CNN). To improve the generalization of the proposed network for federated evaluation, we propose a two-stage encoder-decoder CNN that performs coarse segmentation at stage-I and fine segmentation at stage-II. Stage-I consists of an ensemble of three predictions on the orthogonal slices of a subject. In stage-II, the predictions of the first stage are used to crop the region of interest consisting of the tumor region and a fine grain segmentation is performed on the cropped image. A single ResUNet was used for stage-I and seven different networks were used for stage-II. Heavy data augmentation consisting of geometric transformation and random contrast was used to avoid overfitting and improve the generalization. The mean dice scores on 21 imaging sites evaluated in a federated manner achieved dice scores of 0.8659, 0.7708, and 0.7714 for the whole tumor, tumor core, and enhancing tumor respectively. The method ranked second in the federated evaluation task.

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
Pages494-505
Number of pages12
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