关键词:
医学影像
胸腔CT
深度学习
肋骨分割
肋骨骨折检测
噪声标签
摘要:
肋骨是人体的重要组成部分,肋骨骨折是一种常见的肋骨损伤。胸腔CT成像是辅助诊断肋骨相关疾病的首选工具。近年来,深度学习技术快速发展,将其应用于肋骨相关疾病的辅助诊断,有着巨大的潜力。本文针对胸腔三维CT中的肋骨成像,开展了针对肋骨分割和肋骨骨折检测任务的研究,开发相关深度学习算法,以辅助医疗诊断。本研究的内容主要包括以下三个部分:(1)基于传统图像算法的肋骨分割算法分割精度不高,且易丢失肋骨。基于深度学习的3D分割算法能结合更多的信息,但是资源消耗大,往往需要多次下采样或者使用分块推断的方式进行分割,会影响分割精度;而2D网络资源需求更少,但不能很好地结合各张CT切片间的信息。为了解决这些问题,本研究针对胸腔三维CT成像中的肋骨分割任务,提出了一种名为ISF-UNet(Inter Slice Fusion UNet)的2.5D网络。针对肋骨区域目标较小的问题,ISF-UNet减少了下采样次数,并使用Dense ASPP来提取多尺度信息。ISF-UNet的输入为一组CT切片,网络利用2D CNN来提取切片内的信息,并使用 ISF-Transformer(Inter Slice Fusion Transformer)来融合各张切片间的信息。在RibSeg数据集上的实验结果显示,ISF-UNet取得了 96.79%Dice值和99.44%肋骨召回率,在肋骨分割任务上的表现超越了其他深度学习分割算法。(2)肋骨骨折是一种常见的肋骨损伤,准确诊断肋骨骨折十分重要。已有一些工作对使用深度学习算法进行肋骨骨折检测进行了初步尝试,但由于肋骨骨折区域较小,形状多样且不规则,该任务仍有较大挑战。为了解决这个问题,本文提出了一种肋骨骨折检测算法,包括肋骨分割,3D滑动窗口分割和后处理三个步骤,用于更准确地检测肋骨骨折。该算法利用由双注意力模块增强的分割网络CSFG-UNet(Channel-wise Fusion Attention&Spatial-wise Group Attention UNet),在 3D 滑动窗分割阶段准确分割肋骨骨折区域。双注意力模块包括通道融合注意力模块(Channel-wise Fusion Attention Module,CFAM)和空间分组注意力模块(Spatial-wise Group Attention Module,SGAM)。CFAM关注特征图通道间的关系,使用通道注意力机制来沿通道维度重新加权特征映射。SGAM使用分组卷积技术生成空间注意力图,以保留更多细粒度的空间信息,帮助网络从空间维度调整特征图。最后,本研究在自建的肋骨骨折检测数据集上进行了实验,结果显示,本文所提出的肋骨骨折检测算法达到了89.74%的最高敏感度,平均FROC得分为81.41%,均优于现有方法。(3)在大规模的医学影像数据集中,难免会有噪声标签的存在,深度学习网络容易过拟合噪声标签,从而对网络性能产生不利影响。为了解决这个问题,本文提出了一种对噪声标签鲁棒的框架,来帮助深度学习网络从带有噪声标签的数据中学习。该方法通过Mean Teacher模型对训练过程中的神经网络参数进行集成,从而提升教师模型预测的鲁棒性,并将教师模型的输出作为伪标签监督网络的学习。此外,该框架通过衡量伪标签和真实标签之间的相似度来筛选噪声标签,动态决定标签监督和伪标签监督之间的权重。伪标签中也会存在噪声,该框架通过不确定度估计的方式排除伪标签中噪声的监督信号。最后,本文在RibSeg数据集和肋骨骨折检测数据集上模拟了噪声标签,并进行了实验。实验结果显示,对于肋分割算法,在25%和50%噪声的情况下,使用框架训练,分割Dice值分别提升了 2.72和3.31个百分点,对于肋骨骨折检测算法,检测的平均FROC值提升了 3.27个百分点,说明了本文所提出的方法有效性。