论文标题
在深卷积神经网络中预处理的球形坐标转换用于MRI中的脑肿瘤分割
Spherical coordinates transformation pre-processing in Deep Convolution Neural Networks for brain tumor segmentation in MRI
论文作者
论文摘要
磁共振成像(MRI)用于日常临床实践来评估脑肿瘤。已经引入了几种自动或半自动分割算法,以分割脑肿瘤并实现专家样精度。深度卷积神经网络(DCNN)最近显示出非常有希望的结果,但是,DCNN模型仍然远非临床上有意义的结果,这主要是由于模型缺乏概括。 DCNN模型需要大量注释的数据集来实现良好的性能。通常在对其训练的域数据集上进行了优化,然后当将同一模型应用于来自不同机构的不同数据集时使任务失败。原因之一是由于缺乏对不同模型和MR机器进行调整的数据标准化所致。在这项工作中,假设在预处理阶段的3D球形坐标变换以提高DCNN模型的准确性,即使在小型且异构的数据集中训练该模型并将其转换为不同的领域,也可以允许更具概括性的结果。实际上,球形坐标系避免了几个标准化问题,因为它独立于分辨率和成像设置。使用BRATS 2019数据集,在两个具有相同网络结构的DCNN模型中评估了笛卡尔和球形体积。对球形变换进行培训的模型预处理输入,在预测神经胶质瘤对肿瘤核心上的分割和增强肿瘤类别(分别增加0.011和0.014上的验证数据集上的分割)方面,具有优于训练有素的训练模型的性能(验证数据集上的0.014),通过将两个模型融合在一起,从而进一步提高了准确的改进。此外,球形变换不是分辨率依赖性的,并且在不同的输入分辨率上获得相同的结果。
Magnetic Resonance Imaging (MRI) is used in everyday clinical practice to assess brain tumors. Several automatic or semi-automatic segmentation algorithms have been introduced to segment brain tumors and achieve an expert-like accuracy. Deep Convolutional Neural Networks (DCNN) have recently shown very promising results, however, DCNN models are still far from achieving clinically meaningful results mainly because of the lack of generalization of the models. DCNN models need large annotated datasets to achieve good performance. Models are often optimized on the domain dataset on which they have been trained, and then fail the task when the same model is applied to different datasets from different institutions. One of the reasons is due to the lack of data standardization to adjust for different models and MR machines. In this work, a 3D Spherical coordinates transform during the pre-processing phase has been hypothesized to improve DCNN models' accuracy and to allow more generalizable results even when the model is trained on small and heterogeneous datasets and translated into different domains. Indeed, the spherical coordinate system avoids several standardization issues since it works independently of resolution and imaging settings. Both Cartesian and spherical volumes were evaluated in two DCNN models with the same network structure using the BraTS 2019 dataset. The model trained on spherical transform pre-processed inputs resulted in superior performance over the Cartesian-input trained model on predicting gliomas' segmentation on tumor core and enhancing tumor classes (increase of 0.011 and 0.014 respectively on the validation dataset), achieving a further improvement in accuracy by merging the two models together. Furthermore, the spherical transform is not resolution-dependent and achieve same results on different input resolution.