编辑推荐:
为解决肺部结节分割难题,研究人员提出 CSEA-Net 模型,在多数据集表现卓越,辅助肺癌诊断。
肺癌,这个 “健康杀手”,长期霸占全球癌症致死原因的榜首。在肺癌的早期,患者往往只表现出肺部结节,没有其他明显症状,因此,及时发现并准确分割肺部结节,对提高患者的预后至关重要。在临床实践中,低剂量计算机断层扫描(CT)是筛查肺部结节的主要手段。然而,传统的手动分割肺部结节的方法,就像在布满迷雾的森林里寻找宝藏,不仅依赖人工,容易受到经验和主观因素的影响,还面临着 CT 图像中结节纹理、形状和位置多变,边界模糊等难题。而且,不同的放射科医生水平参差不齐,误诊、漏诊时有发生。因此,寻找一种高效、准确的计算机辅助诊断方法,成为了医学领域的迫切需求。
为了攻克这一难题,华中科技大学同济医学院附属协和医院等机构的研究人员展开了深入研究,他们的成果发表在《iScience》上。这项研究提出了一种名为 CSEA-Net(Channel-Spatial Enhanced Attention Network,通道 - 空间增强注意力网络)的全自动肺部结节分割模型。
研究人员在研究过程中用到了多个关键技术方法。在数据方面,使用了 LUNA16、MSD 和 LNDb 这三个公开的肺部 CT 图像数据集 。数据预处理时,将 CT 值裁剪到特定范围,进行随机选择技术处理和归一化,并按 7:2:1 的比例划分数据集。模型构建上,以 U-Net++ 为基础,引入 CSEA 模块,该模块包含通道增强残差模块(CERM)、空间特征增强模块(SFEM)和坐标注意力(CA)模块。训练模型时,采用 Adam 优化函数,设置相关参数,使用 PyTorch 框架在特定 GPU 环境下训练 300 个 epoch 。评估模型性能时,选用交并比(IoU)、Dice 相似系数(DSC)、BCE-Dice 损失和边界 IoU 等指标。
下面来看具体的研究结果:
- 数据集和数据预处理:研究人员选用了 LUNA16、MSD 和 LNDb 三个数据集,这些数据集涵盖了不同来源和特征的肺部 CT 图像。经过一系列的数据预处理步骤,包括 CT 值裁剪、随机变换和归一化等,将图像划分为训练集、测试集和验证集,为后续的模型训练和评估做好准备。
- 损失函数:由于肺部结节面积小,使用单一的 Dice 损失函数会导致训练误差曲线不稳定。因此,研究人员将 Dice 损失函数与二元交叉熵(BCE)损失函数结合,提出了新的损失函数,公式为L=0.5LDice+LBCE,其中LDice=1?∣X∣+∣Y∣+λ2∣X∩Y∣+λ ,LBCE=ylog(x)+(1?y)log(1?x),这样可以更稳定地分割小目标物体。
- 评估指标:为了全面评估模型的性能,研究人员使用了 IoU、DSC、BCE-Dice 损失和边界 IoU 等多个指标。这些指标从不同角度反映了模型分割结果与真实情况的匹配程度,能够更准确地评估模型在肺部结节分割任务中的表现。
- 性能量化:将 CSEA-Net 与 FCN、U-Net 等多种先进模型进行对比,结果令人惊喜。在所有三个数据集上,CSEA-Net 在各项指标上均优于其他模型。例如,在 LUNA16 数据集上,CSEA-Net 的 DSC 达到 92.69%,IoU 达到 86.56%;在 MSD 数据集上,DSC 为 97.17%,IoU 为 95.03%;在 LNDb 数据集上,DSC 高达 98.65%,IoU 为 97.58%。这充分表明 CSEA-Net 在肺部肿瘤分割方面具有强大的实力,能更精准地识别和分割肺部结节。
- 定性性能:通过定性评估,研究人员发现 FCN、U-Net 等模型在分割肺部结节时,难以准确描绘肿瘤边界;Attention U-Net 和 U-Net++ 虽然有所改进,但仍存在输出碎片化和轮廓不准确的问题。而 CSEA-Net 凭借其独特的空间和通道特征增强模块以及 CA 机制,能够更好地捕捉肿瘤的复杂形态,减少特征不兼容问题,在三个数据集上都能更精确地逼近真实情况,展现出强大的医学图像细节处理能力。
- 消融研究:研究人员进行了消融实验,分别去除模型中的 DCSE 模块和 CA 模块,以验证各模块的作用。结果发现,单独引入 DCSE 模块时,DSC 提升不明显,这是因为 DCSE 网络增加了参数,使网络变得复杂。而引入 CA 模块后,DSC 和 IoU 都显著提高,这突出了 CA 模块在捕捉关键信息、提升分割精度方面的重要作用。同时也证明了将 DCSE 模块和 CA 机制集成到 U-Net++ 框架中,对增强肺部肿瘤分割具有协同效应。
研究结论和讨论部分指出,深度学习在肺部结节分割领域已经逐渐取代传统方法,成为肺癌诊断的重要手段。CSEA-Net 在多个公开数据集上表现优异,能够有效分割肺部结节,且能保持微小结节边缘的完整性,是一个完全自动化的系统,无需人工干预。不过,该研究也存在一些局限性,比如模型性能依赖高质量标注数据集,在处理极不均质肿瘤或与周围组织对比度极低的肿瘤时,效果还需进一步验证,而且目前仅在 2D 医学图像上进行了研究,在 3D 医学图像上的适用性和性能有待探索。未来,研究人员将致力于提高模型的鲁棒性,优化模型,探索如自动图像增强等技术提升分割性能,并将模型扩展到 3D 肺部结节分割,以更好地应用于临床。
总的来说,CSEA-Net 的出现为肺部肿瘤的自动分割提供了新的解决方案,在肺癌的早期诊断中具有巨大的潜力,有望减轻放射科医生的工作负担,提高诊断效率和准确性,为肺癌患者带来新的希望。