Banerjee Subhashis, Mitra Sushmita
Machine Intelligence Unit, Indian Statistical Institute, Kolkata, India.
Department of CSE, University of Calcutta, Kolkata, India.
Front Comput Neurosci. 2020 Jan 24;14:3. doi: 10.3389/fncom.2020.00003. eCollection 2020.
A novel deep learning based model called Multi-Planar Spatial Convolutional Neural Network (MPS-CNN) is proposed for effective, automated segmentation of different sub-regions . peritumoral edema (), necrotic core (), enhancing and non-enhancing tumor core (/), from multi-modal MR images of the brain. An encoder-decoder type CNN model is designed for pixel-wise segmentation of the tumor along three anatomical planes (axial, sagittal, and coronal) at the slice level. These are then combined, by incorporating a consensus fusion strategy with a fully connected Conditional Random Field (CRF) based post-refinement, to produce the final volumetric segmentation of the tumor and its constituent sub-regions. Concepts, such as spatial-pooling and unpooling are used to preserve the spatial locations of the edge pixels, for reducing segmentation error around the boundaries. A new aggregated loss function is also developed for effectively handling data imbalance. The MPS-CNN is trained and validated on the recent Multimodal Brain Tumor Segmentation Challenge (BraTS) 2018 dataset. The Dice scores obtained for the validation set for whole tumor ( :/ + +), tumor core (:/ +), and enhancing tumor () are 0.90216, 0.87247, and 0.82445. The proposed MPS-CNN is found to perform the best (based on leaderboard scores) for and segmentation tasks, in terms of both the quantitative measures (viz. Dice and Hausdorff). In case of the segmentation it also achieved the second highest accuracy, with a score which was only 1% less than that of the best performing method.
提出了一种名为多平面空间卷积神经网络(MPS-CNN)的新型深度学习模型,用于从脑部多模态磁共振图像中有效地自动分割不同的子区域,即瘤周水肿、坏死核心、强化和非强化肿瘤核心。设计了一种编码器-解码器类型的卷积神经网络模型,用于在切片级别沿三个解剖平面(轴向、矢状和冠状)对肿瘤进行逐像素分割。然后,通过将共识融合策略与基于全连接条件随机场(CRF)的后处理相结合,将这些分割结果合并,以生成肿瘤及其组成子区域的最终体积分割。使用空间池化和解池化等概念来保留边缘像素的空间位置,以减少边界周围的分割误差。还开发了一种新的聚合损失函数,以有效处理数据不平衡问题。MPS-CNN在最近的2018年多模态脑肿瘤分割挑战赛(BraTS)数据集上进行了训练和验证。验证集上全肿瘤(:/+ +)、肿瘤核心(:/ +)和强化肿瘤()的Dice分数分别为0.90216、0.87247和0.82445。结果发现,所提出的MPS-CNN在和分割任务中,在定量测量(即Dice和豪斯多夫距离)方面表现最佳(基于排行榜分数)。在分割的情况下,它也取得了第二高的准确率,得分仅比表现最佳的方法低1%。