Automatic Brain Tumor Segmentation using Convolutional Neural Networks with Test-Time Augmentation

Abstract

Automatic brain tumor segmentation plays an important role for diagnosis, surgical planning and treatment assessment of brain tumors. Deep convolutional neural networks (CNNs) have been widely used for this task. Due to the relatively small data set for training, data augmentation at training time has been commonly used for better performance of CNNs. Recent works also demonstrated the usefulness of using augmentation at test time, in addition to training time, for achieving more robust predictions. We investigate how test-time augmentation can improve CNNs' performance for brain tumor segmentation. We used different underpinning network structures and augmented the image by 3D rotation, flipping, scaling and adding random noise at both training and test time. Experiments with BraTS 2018 training and validation set show that test-time augmentation helps to improve the brain tumor segmentation accuracy and obtain uncertainty estimation of the segmentation results.

arXiv comment12 pages, 3 figures, MICCAI BrainLes 2018

References in corpus (17)

• Adam: A Method for Stochastic Optimization
• TensorFlow: A system for large-scale machine learning
• mixup: Beyond Empirical Risk Minimization
• Efficient Multi-Scale 3D CNN with Fully Connected CRF for Accurate Brain Lesion Segmentation
• Brain Tumor Segmentation with Deep Neural Networks
• Efficient Inference in Fully Connected CRFs with Gaussian Edge Potentials
• What Uncertainties Do We Need in Bayesian Deep Learning for Computer Vision?
• Fully Convolutional Networks for Semantic Segmentation
• Dropout as a Bayesian Approximation: Representing Model Uncertainty in Deep Learning
• Multi-Scale Context Aggregation by Dilated Convolutions
• Interactive Medical Image Segmentation using Deep Learning with Image-specific Fine-tuning
• Aleatoric uncertainty estimation with test-time augmentation for medical image segmentation with convolutional neural networks
• NiftyNet: a deep-learning platform for medical imaging
• Automatic Brain Tumor Segmentation using Cascaded Anisotropic Convolutional Neural Networks
• Image Classification of Melanoma, Nevus and Seborrheic Keratosis by Deep Neural Network Ensemble
• Generalised Wasserstein Dice Score for Imbalanced Multi-class Segmentation using Holistic Convolutional Networks
• No New-Net

Cited by in corpus (12)

• A Review of Uncertainty Quantification in Deep Learning: Techniques, Applications and Challenges
• Attention-Guided Version of 2D UNet for Automatic Brain Tumor Segmentation
• Deep convolutional neural networks for multi-planar lung nodule detection: improvement in small nodule identification
• Learning Loss for Test-Time Augmentation
• Uncertainty Aware Training to Improve Deep Learning Model Calibration for Classification of Cardiac MR Images
• A Survey and Analysis on Automated Glioma Brain Tumor Segmentation and Overall Patient Survival Prediction
• A Survey on Deep Learning Methods for Semantic Image Segmentation in Real-Time
• Exploiting Uncertainties from Ensemble Learners to Improve Decision-Making in Healthcare AI
• Hyperspectral Data Augmentation
• A Multi-task Contextual Atrous Residual Network for Brain Tumor Detection & Segmentation
• Brain MR Image Segmentation in Small Dataset with Adversarial Defense and Task Reorganization
• Adaptive Test-Time Augmentation for Low-Power CPU