头条资讯网_今日热点_娱乐才是你关心的时事

今日热点 时事资讯
娱乐头条才是你关心的新闻
首页 > 头条资讯 > 科技

基于多尺度卷积神经网络的脑肿瘤分类与分割深度学习方法

研究背景

医学图像的自动分割和分类在脑肿瘤的诊断、预测和治疗中起着重要作用。使用传统的手动程序对数据库中的脑肿瘤图像进行定位和分类需要耗费大量的精力和时间成本。深度学习技术的出现为这一问题提供了解决方案。2021年2月2日,来自西班牙巴利亚多利德大学通信工程学院的DavidGonzález-Ortega教授及其团队在Healthcare期刊发表本文,提出了一种利用多尺度卷积神经网络进行脑肿瘤全自动分类和分割的深度学习方法。该方法在233名患者的MRI图像数据集上进行训练和测试,并取得了良好的分割和分类效果。与之前的方法相比,该方法具有更好的性能和准确度。

研究方法

卷积神经网络及其实现细节

本文首先提出了一种用于肿瘤分割的多通路CNN架构。该架构逐像素处理MRI图像,并使用四种可能的输出标签(0-健康区域,1-脑膜瘤肿瘤,2-神经胶质瘤肿瘤,3-垂体肿瘤)对每个像素进行分类。每个窗口都通过三个卷积路径进行处理,其中不同尺度的内核提取特征。三个路径的尺度特征在一个卷积层中连接起来,然后进入一个完全连接的阶段,最后通过softmax激活函数得到最终的分类结果。所提出的CNN使用Pytorch实现,共有约300万个可训练参数。在数据预处理阶段,使用弹性变换进行数据增强以防止过度拟合。训练数据集包含233名患者的3064张切片,每个切片都有相应的肿瘤类型标签和肿瘤边界信息。训练过程耗时5天,平均预测时间为57.5秒。

图1.卷积神经网络(CNN)架构。

数据集与数据预处理

在临床环境中使用脑CE-MRI切片进行肿瘤识别和建模。临床提供的通常只有一定数量的大切片间隙的数据可用,难以构建3D模型。本文方法基于的数据集包括了233名患者的3064个切片,涵盖了脑膜瘤、神经胶质瘤和垂体瘤的常见视图。这些切片被用于训练和性能测量,数据集提供了5倍交叉验证指数。图像具有512×512像素的面内分辨率,像素大小为0.49×0.49mm2,切片厚度为6mm,切片间隙为1mm。肿瘤边界由三名放射科医生手动描绘,每个切片都有相应的患者信息、肿瘤类型标签和肿瘤掩模。训练过程中使用弹性变换进行数据增强以防止过度拟合,并且从每个训练图像中提取训练示例。

图2.左:原始切片;右:图像变换;在两张图像中,肿瘤的边缘都以红色突出显示。

神经网络训练和性能测量

本研究使用CNN进行肿瘤分类和分割的性能评估过程。首先,研究人员使用给定的数据集进行5倍交叉验证,每个折叠使用随机梯度下降(SGD)优化器进行80个时期的训练,其参数设置为学习率0.005,动量系数0.9和dropout参数0.5。然后,在每个折叠中,使用滑动窗口方法在612个图像上测试模型的性能,通过将大小为65×65像素的测试窗口输入训练好的CNN来预测肿瘤类型标签。将标准化的训练窗口像素值用于测试窗口的标准化。研究人员根据预测的标签和标记的标签,计算分类函数的关系,根据预测的标签大小关系和置信度阈值来确定最终预测的标签。性能评估方面,研究计算了混淆矩阵、Dice系数和灵敏度分数,并定义了预测肿瘤类型比率评分pttas来评估预测肿瘤类型的准确性。pttas指数是根据预测肿瘤和真实肿瘤之间的大小关系来计算的。

评估指标使用以下公式计算:

研究结果

研究通过使用数据集中的5倍交叉验证训练/测试子组,测试了神经网络的性能。结果显示,在三种脑肿瘤(脑膜瘤、神经胶质瘤和垂体肿瘤)的分割中取得了较好的效果。

图3.图像显示肿瘤分割:检测到的区域显示为红色,而真实区域显示为绿色;结果,交叉区域显示为黄色。

通过分割指标的定量结果(Dice系数、敏感度和pttas指数),可以看出该方法对于脑膜瘤的分割效果最好,垂体肿瘤的分割效果最差。此外,此方法还进行了肿瘤分类,结果表明其准确性为0.973,优于其他深度学习方法。

研究总结

本文提出了一种基于CNN的全自动脑肿瘤分割和分类方法,并使用公开的MRI图像数据集进行性能评估。通过弹性变换的数据增强来增加训练数据集防止过度拟合。与其他七种脑肿瘤分类方法相比,此方法获得了最高的肿瘤分类精度0.973。平均Dice指数和灵敏度分数分别为0.828和0.940,平均pttas值为0.967。该方法可以辅助医生进行脑肿瘤诊断,并可应用于其他医学成像问题。未来的工作包括与使用相同数据集的FCN架构进行比较,并研究多尺度CNN在其他领域的适用性。

原文出自Healthcare期刊

未经允许不得转载:头条资讯网_今日热点_娱乐才是你关心的时事 » 基于多尺度卷积神经网络的脑肿瘤分类与分割深度学习方法

分享到:更多 ()
来源:MDPI开放科学 编辑:科技

评论

留言/评论 共有条点评
昵称:
验证码:
匿名发表