Université de Sherbrooke, Sherbrooke, Qc, Canada.
École Normale supérieure, Paris, France.
Med Image Anal. 2017 Jan;35:18-31. doi: 10.1016/j.media.2016.05.004. Epub 2016 May 19.
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.
本文提出了一种基于深度神经网络(DNN)的全自动脑肿瘤分割方法。所提出的网络针对磁共振图像中的低级别和高级别胶质瘤进行了定制。由于这些肿瘤的性质,它们可以出现在大脑的任何部位,并且具有几乎任何形状、大小和对比度。这些原因促使我们探索一种利用灵活、大容量 DNN 的机器学习解决方案,同时具有极高的效率。在这里,我们描述了不同的模型选择,这些选择对于获得竞争性能是必要的。我们特别探讨了基于卷积神经网络(CNN)的不同架构,即专门针对图像数据的 DNN。我们提出了一种新的 CNN 架构,与传统的计算机视觉中使用的架构不同。我们的 CNN 同时利用局部特征和更全局的上下文特征。此外,与大多数传统的 CNN 用法不同,我们的网络使用最后一层,这是一个全连接层的卷积实现,允许速度提高 40 倍。我们还描述了一种两阶段训练过程,该过程允许我们解决与肿瘤标签不平衡相关的困难。最后,我们探索了级联架构,其中基本 CNN 的输出被视为后续 CNN 的附加信息源。在 2013 年 BRATS 测试数据集上报告的结果表明,我们的架构在提高速度的同时,也优于目前公布的最先进技术。