Autofocus Net: Auto-focused 3D CNN for Brain Tumour Segmentation.

Andreas Stefani, Roushanak Rahmat, David Cameron Christopher Harris-Birtill

Research output: Chapter in Book/Report/Conference proceedingChapter (peer-reviewed)peer-review

Abstract

Several approaches based on convolutional neural networks (CNNs) are only able to process 2D images while most brain data consists of 3D volumes. Recent network architectures which have demonstrated promising results are able to process 3D images. In this work, we propose an adapted approach based on a CNN to process 3D contextual information in brain MRI scans for the challenging task of brain tumour segmentation. Our CNN is trained end-to-end on multi-modal MRI volumes and is able to predict segmentation for the binary case, which segments the whole tumour, and multi-class case, which segments the whole tumour (WT), tumour core (TC) and enhancing tumour (ET). Our network includes multiple layers of dilated convolutions and autofocus convolutions with residual connections to improve segmentation performance. Autofocus layers consist of multiple parallel convolutions each with a different dilation rate. We replaced standard convolutional layers with autofocus layers to adaptively change the size of the effective receptive field to generate more powerful features. Experiments with our autofocus settings on the BraTS 2018 glioma dataset show that the proposed method achieved average Dice scores of 83.92 for WT in the binary case and 66.88, 55.16, 64.13 for WT, TC and ET, respectively, in the multi-class case. We introduce the first publicly and freely available NiftyNet-based implementation of the autofocus convolutional layer for semantic image segmentation.
Original languageEnglish
Title of host publicationIn Annual Conference on Medical Image Understanding and Analysis
Subtitle of host publicationPart of the Communications in Computer and Information Science book series (CCIS)
PublisherSpringer
Pages43-55
Number of pages13
Volume1248
ISBN (Electronic)978-3-030-52791-4
Publication statusPublished - 8 Jul 2020

Fingerprint

Dive into the research topics of 'Autofocus Net: Auto-focused 3D CNN for Brain Tumour Segmentation.'. Together they form a unique fingerprint.

Cite this