论文标题
使用轻量级3D深度学习体系结构中T1加权MRI量的快速脑膜瘤分割
Fast meningioma segmentation in T1-weighted MRI volumes using a lightweight 3D deep learning architecture
论文作者
论文摘要
T1加权MRI体积和相应的体积评估中的自动且一致的脑膜瘤分割用于诊断,治疗计划和肿瘤生长评估。在本文中,我们使用大量经过手术治疗的脑膜瘤和未经治疗的脑膜瘤进行了分割和加工速度性能,随后在门诊诊所进行了。我们研究了两个不同的3D神经网络体系结构:(i)类似于3D U-NET的简单编码器解码器,以及(ii)轻量级的多尺度体系结构(PLS-NET)。此外,我们研究了不同培训方案的影响。在验证研究中,我们使用了挪威特朗德海姆的圣奥拉夫大学医院的698个T1加权MR卷。根据检测准确性,分割精度和训练/推理速度评估模型。尽管两个架构平均得分平均达到70%,但PLS-NET更准确,F1得分高达88%。对于最大的脑膜瘤,达到了最高的精度。从速度方面,PLS-NET架构倾向于在大约50小时内收敛,而U-NET需要130小时。使用PLS-NET的推断在GPU上需要不到一秒钟,在CPU上进行了约15秒。总体而言,通过使用混合精度培训,可以使用轻质PLS-NET架构在相对较短的时间内训练竞争性分割模型。将来,应将重点放在小型脑膜瘤(小于2ML)的分割上,以提高自动和早期诊断的临床相关性以及增长速度。
Automatic and consistent meningioma segmentation in T1-weighted MRI volumes and corresponding volumetric assessment is of use for diagnosis, treatment planning, and tumor growth evaluation. In this paper, we optimized the segmentation and processing speed performances using a large number of both surgically treated meningiomas and untreated meningiomas followed at the outpatient clinic. We studied two different 3D neural network architectures: (i) a simple encoder-decoder similar to a 3D U-Net, and (ii) a lightweight multi-scale architecture (PLS-Net). In addition, we studied the impact of different training schemes. For the validation studies, we used 698 T1-weighted MR volumes from St. Olav University Hospital, Trondheim, Norway. The models were evaluated in terms of detection accuracy, segmentation accuracy and training/inference speed. While both architectures reached a similar Dice score of 70% on average, the PLS-Net was more accurate with an F1-score of up to 88%. The highest accuracy was achieved for the largest meningiomas. Speed-wise, the PLS-Net architecture tended to converge in about 50 hours while 130 hours were necessary for U-Net. Inference with PLS-Net takes less than a second on GPU and about 15 seconds on CPU. Overall, with the use of mixed precision training, it was possible to train competitive segmentation models in a relatively short amount of time using the lightweight PLS-Net architecture. In the future, the focus should be brought toward the segmentation of small meningiomas (less than 2ml) to improve clinical relevance for automatic and early diagnosis as well as speed of growth estimates.