Brain tumor segmentation with Deep Neural Networks

&NA; In this paper, we present a fully automatic brain tumor segmentation method based on Deep Neural Networks (DNNs). The proposed networks are tailored to glioblastomas (both low and high grade) pictured in MR images. By their very nature, these tumors can appear anywhere in the brain and have almost any kind of shape, size, and contrast. These reasons motivate our exploration of a machine learning solution that exploits a flexible, high capacity DNN while being extremely efficient. Here, we give a description of different model choices that we've found to be necessary for obtaining competitive performance. We explore in particular different architectures based on Convolutional Neural Networks (CNN), i.e. DNNs specifically adapted to image data. We present a novel CNN architecture which differs from those traditionally used in computer vision. Our CNN exploits both local features as well as more global contextual features simultaneously. Also, different from most traditional uses of CNNs, our networks use a final layer that is a convolutional implementation of a fully connected layer which allows a 40 fold speed up. We also describe a 2‐phase training procedure that allows us to tackle difficulties related to the imbalance of tumor labels. Finally, we explore a cascade architecture in which the output of a basic CNN is treated as an additional source of information for a subsequent CNN. Results reported on the 2013 BRATS test data‐set reveal that our architecture improves over the currently published state‐of‐the‐art while being over 30 times faster. HighlightsA fast and accurate fully automatic method for brain tumor segmentation which is competitive both in terms of accuracy and speed compared to the state of the art.The method is based on deep neural networks (DNN) and learns features that are specific to brain tumor segmentation.We present a new DNN architecture which exploits both local features as well as more global contextual features simultaneously.Using a GPU implementation and a convolutional output layer, the model is an order of magnitude faster than other state of the art methods.Introduces a novel cascaded architecture that allows the system to more accurately model local label dependencies. Graphical abstract Figure. No caption available.

[1]  Mark W. Schmidt,et al.  3D Variational Brain Tumor Segmentation using a High Dimensional Feature Set , 2007, 2007 IEEE 11th International Conference on Computer Vision.

[2]  Guido Gerig,et al.  A brain tumor segmentation framework based on outlier detection , 2004, Medical Image Anal..

[3]  S. Bauer,et al.  A survey of MRI-based medical image analysis for brain tumor studies , 2013, Physics in medicine and biology.

[4]  Lawrence O. Hall,et al.  Automatic tumor segmentation using knowledge-based techniques , 1998, IEEE Transactions on Medical Imaging.

[5]  Luca Maria Gambardella,et al.  Deep Neural Networks Segment Neuronal Membranes in Electron Microscopy Images , 2012, NIPS.

[6]  Mark W. Schmidt,et al.  Segmenting Brain Tumors with Conditional Random Fields and Support Vector Machines , 2005, CVBIA.

[7]  D. Louis Collins,et al.  Hierarchical Probabilistic Gabor and MRF Segmentation of Brain Tumours in MRI Volumes , 2013, MICCAI.

[8]  Trevor Darrell,et al.  Fully Convolutional Networks for Semantic Segmentation , 2017, IEEE Transactions on Pattern Analysis and Machine Intelligence.

[9]  Ronan Collobert,et al.  Recurrent Convolutional Neural Networks for Scene Labeling , 2014, ICML.

[10]  Yann LeCun,et al.  What is the best multi-stage architecture for object recognition? , 2009, 2009 IEEE 12th International Conference on Computer Vision.

[11]  Pascal Vincent,et al.  Representation Learning: A Review and New Perspectives , 2012, IEEE Transactions on Pattern Analysis and Machine Intelligence.

[12]  Antonio Criminisi,et al.  Segmentation of Brain Tumor Tissues with Convolutional Neural Networks , 2014 .

[13]  Isabelle Bloch,et al.  3D brain tumor segmentation in MRI using fuzzy classification, symmetry analysis and spatially constrained deformable models , 2009, Fuzzy Sets Syst..

[14]  Fred A. Hamprecht,et al.  Multi-modal Brain Tumor Segmentation using Deep Convolutional Neural Networks , 2014 .

[15]  Stefan Bauer,et al.  Fully Automatic Segmentation of Brain Tumor Images Using Support Vector Machine Classification in Combination with Hierarchical Conditional Random Field Regularization , 2011, MICCAI.

[16]  Camille Couprie,et al.  Learning Hierarchical Features for Scene Labeling , 2013, IEEE Transactions on Pattern Analysis and Machine Intelligence.

[17]  Yoshua Bengio,et al.  Maxout Networks , 2013, ICML.

[18]  Carlos A. Silva,et al.  Fully Automatic Brain Tumor Segmentation from Multiple MR Sequences using Hidden Markov Fields and Variational , 2013 .

[19]  Viren Jain,et al.  Deep and Wide Multiscale Recursive Networks for Robust Image Labeling , 2013, ICLR.

[20]  Nikos Paragios,et al.  Joint Tumor Segmentation and Dense Deformable Registration of Brain MR Images , 2012, MICCAI.

[21]  Olivier Clatz,et al.  Glioma Dynamics and Computational Models: A Review of Segmentation, Registration, and In Silico Growth Algorithms and their Clinical Applications , 2007 .

[22]  Jitendra Malik,et al.  Simultaneous Detection and Segmentation , 2014, ECCV.

[23]  Geoffrey E. Hinton,et al.  ImageNet classification with deep convolutional neural networks , 2012, Commun. ACM.

[24]  Guido Gerig,et al.  Robust Estimation for Brain Tumor Segmentation , 2003, MICCAI.

[25]  Martin Jägersand,et al.  3D variational brain tumor segmentation using Dirichlet priors on a clustered feature set , 2012, International Journal of Computer Assisted Radiology and Surgery.

[26]  Yoshua Bengio,et al.  Gradient-based learning applied to document recognition , 1998, Proc. IEEE.

[27]  Gözde B. Ünal,et al.  Tumor-Cut: Segmentation of Brain Tumors on Contrast Enhanced MR Images for Radiosurgery Applications , 2012, IEEE Transactions on Medical Imaging.

[28]  Yoshua Bengio,et al.  Domain Adaptation for Large-Scale Sentiment Classification: A Deep Learning Approach , 2011, ICML.

[29]  Razvan Pascanu,et al.  M L ] 2 0 A ug 2 01 3 Pylearn 2 : a machine learning research library , 2014 .

[30]  Geoffrey E. Hinton,et al.  Learning representations by back-propagating errors , 1986, Nature.

[31]  Trevor Darrell,et al.  DeCAF: A Deep Convolutional Activation Feature for Generic Visual Recognition , 2013, ICML.

[32]  Rob Fergus,et al.  Visualizing and Understanding Convolutional Networks , 2013, ECCV.

[33]  LeCunYann,et al.  Learning Hierarchical Features for Scene Labeling , 2013 .

[34]  Michael I. Jordan,et al.  A generalized mean field algorithm for variational inference in exponential families , 2002, UAI.

[35]  Nitish Srivastava,et al.  Dropout: a simple way to prevent neural networks from overfitting , 2014, J. Mach. Learn. Res..

[36]  Yoshua Bengio,et al.  Practical Recommendations for Gradient-Based Training of Deep Architectures , 2012, Neural Networks: Tricks of the Trade.

[37]  Brian B. Avants,et al.  The Multimodal Brain Tumor Image Segmentation Benchmark (BRATS) , 2015, IEEE Transactions on Medical Imaging.

[38]  G. Urban,et al.  ilastik for Multi-modal Brain Tumor Segmentation , 2014 .

[39]  Doina Precup,et al.  Iterative Multilevel MRF Leveraging Context and Voxel Information for Brain Tumour Segmentation in MRI , 2014, 2014 IEEE Conference on Computer Vision and Pattern Recognition.

[40]  Yann LeCun,et al.  Road Scene Segmentation from a Single Image , 2012, ECCV.

[41]  Mark W. Schmidt,et al.  Segmenting brain tumors using alignment-based features , 2005, Fourth International Conference on Machine Learning and Applications (ICMLA'05).

[42]  Ben Glocker,et al.  Decision Forests for Tissue-Specific Segmentation of High-Grade Gliomas in Multi-channel MR , 2012, MICCAI.

[43]  Terry M. Peters,et al.  Medical Image Computing and Computer-Assisted Intervention - MICCAI 2003 , 2003, Lecture Notes in Computer Science.

[44]  Mohammad Havaei,et al.  Efficient Interactive Brain Tumor Segmentation as Within-Brain kNN Classification , 2014, 2014 22nd International Conference on Pattern Recognition.

[45]  Simon Haykin,et al.  GradientBased Learning Applied to Document Recognition , 2001 .

[46]  Lawrence O. Hall,et al.  Automatic Tumor Segmentation Using Knowledge-Based Clustering , 1998, IEEE Trans. Medical Imaging.