关键词:
磁共振图像
脑肿瘤分割
卷积神经网络
注意力机制
轻量化
摘要:
医学图像分割是计算机视觉领域的重要研究方向之一,它可以帮助医生更准确、及时地诊断疾病,提高患者的诊疗效果和生存率。脑肿瘤是一种常见的神经系统疾病,具有患病率高、诊断困难的特点,如何精准且快速地定位脑肿瘤的位置和类型,对诊疗过程中的方案制定和预后评估至关重要。近年来,随着深度学习技术的发展,基于卷积神经网络的多模态脑肿瘤分割已经成为研究的热点。然而,由于不同个体的脑肿瘤的形态、位置和大小存在较大的异质性,同时受限于网络的复杂度,使得现有脑肿瘤分割算法的精度和速度仍有待提升。本文提出了3种基于深度卷积神经网络和多模态磁共振影像(Magnetic Resonance Imaging,MRI)的脑肿瘤分割算法,并在公开数据集上进行了算法性能测试。本文主要研究内容如下:(1)提出基于金字塔切分注意力和2D U-Net的脑肿瘤分割算法PSA-UNet。PSA-UNet提出改进的金字塔切分注意力模块,将其与2D U-Net网络中的卷积块拼接,获取脑部图像的多尺度信息。采用实例归一化对数据进行处理,减少不同脑肿瘤患者数据之间的干扰,提高模型的稳定性。使用Focal损失和多类别Dice损失函数,降低脑肿瘤患者数据像素类别不平衡对分割结果的影响。本文使用Bra TS2020数据集对PSA-UNet算法进行训练和测试,主要通过性能指标的分数和可视化的结果来对算法进行评估。结果表明,PSA-UNet在全肿瘤和核心区的Dice系数分别为0.901和0.804,核心区的Hausdorff95距离为10.518,并且在脑肿瘤形态各异和病灶像素占比较小的情况下也能较好地将其分割出来。由此可见,本文提出的PSA-UNet算法对脑肿瘤分割的性能具有一定的优势。(2)提出基于混合注意力和3D UNet的脑肿瘤分割算法3D-CAS-UNet。3D-CAS-UNet在3D U-Net网络中的跳跃连接处分别引入改进的3D Self Attention和3D Coordinate Attention模块,构建3D-SA-UNet和3D-CA-UNet子网络,能够更精准地提取肿瘤轮廓和细节信息。其次,3D-CAS-UNet使用残差模块和组归一化,可提高模型的稳定性和降低不同样本分布之间的偏差,并有效地保留原有的特征信息。使用Bra TS2020数据集和Bra TS2018数据集对提出的3D-CASUNet算法进行训练和测试。结果表明,在Bra TS2020和Bra TS2018中,3D-SAUNet和3D-CA-UNet的全肿瘤的Dice系数分别为0.900和0.898,0.899和0.903。两种模型在脑肿瘤区域不连通和小病灶较多的情况下,依然具有较好的分割效果。由此可见,本章提出的算法在分割脑部全肿瘤方面具有明显的优势,并且模型具备一定的泛化能力。(3)提出基于空洞空间金字塔和3D-CA-UNet的轻量化脑肿瘤分割算法DAS-CA-UNet。首先,在网络的最底层引入不同空洞率的空间金字塔池化模块,以提取脑肿瘤图像不同尺度下的特征信息。其次,将网络所有的残差模块中的标准卷积用3D深度可分离卷积代替,在减少模型参数的同时加快对图片的推理速度。最后,使用Bra TS2020数据集对提出的DAS-CA-UNet算法进行实验。实验表明,空洞空间金字塔池化以及3D深度可分离卷积可以在保证模型精度的同时达到轻量化性能。结果表明,DAS-CA-UNet在全肿瘤和增强区的Dice系数分别为0.896和0.727,核心区的Hausdorff95距离为6.538。尤其在脑肿瘤增强区边界模糊和形状大小不规则的情况时,DAS-CA-UNet仍然能够实现良好的分割效果。由此可见,本章提出的DAS-CA-UNet算法在达到轻量化的同时保持了较好的脑肿瘤分割性能。综上所述,本文基于深度卷积神经网络和多模态MRI提出了3种脑肿瘤分割算法PSA-UNet、3D-CAS-UNet和DAS-CA-UNet,并在Bra TS2018和Bra TS2020数据集上进行了算法性能测试。实验证明了本文提出的算法具有良好的脑肿瘤分割效果和轻量化优势,并在一定程度上能解决小病灶多、区域非连通、边界模糊、形状不规则等典型的分割问题。因此,本文提出的算法在机器学习理论和临床医学方面具有良好的研究意义和应用价值。