基于模糊粗糙集损失的深度学习精确医学图像分割方法

时间:2026年1月25日
来源:Computerized Medical Imaging and Graphics

编辑推荐:

提出基于模糊粗糙集的FRS损失函数,通过模糊相似性建模和边界不确定性模型解决医学图像分割中的边界模糊和类别不平衡问题,实验表明其Dice分数平均提升2.1%,显著优于基线方法,且计算高效。

广告
   X   

Mohsin Furkh Dar | Avatharam Ganivada
印度德赫拉敦UPES计算机科学学院,248007

摘要

医学图像的准确分割对于诊断和治疗计划至关重要,但由于病变边界模糊、类别不平衡以及复杂的解剖结构,这一任务仍然具有挑战性。我们提出了一种基于模糊粗糙集(FRS)的新损失函数,该函数通过将像素的模糊相似性关系与边界不确定性模型结合在凸组合方法中来解决这些挑战。为了获得边界不确定性模型,使用了像素集的模糊下界和上界近似以及隶属权重。FRS损失函数通过其两个组成部分增强了边界敏感性并处理了预测不确定性:一个模糊相似性项用于捕捉病变边界的渐进过渡,另一个边界不确定性模型用于处理不确定性并缓解类别不平衡。在五个不同的医学成像数据集(乳腺超声、胃肠道息肉、脑部磁共振成像(MRI)、胸部计算机断层扫描(CT)和皮肤病变)上的广泛实验证明了我们方法的有效性。与最佳基线方法相比,FRS损失函数在Dice分数上平均提高了2.1%,并且在所有评估指标上都显示出统计学上的显著改进(p < 0.001)。FRS损失函数在保持计算效率的同时(平均推理时间为每张图像0.075–0.12秒,内存消耗4.5 MB),表现出对中等程度类别不平衡的鲁棒性。这些结果表明,FRS损失函数为精确的医学图像分割提供了一个鲁棒且可解释的框架,特别是在边界模糊和类别不平衡的情况下。代码:https://github.com/MohsinFurkh/Fuzzy-Rough-Set-Loss

引言

医学图像分割是计算机辅助诊断中的一个基本任务,涉及从超声、MRI和CT扫描等不同模态获得的医学图像中精确勾画解剖结构或感兴趣的区域(Du等人,2023年)。深度学习的发展,特别是通过卷积神经网络(Dar和Ganivada,2023年;Ning等人,2022年)以及基于变压器的模型(Kai等人,2023年;Zhou等人,2023年),显著推动了医学图像分割领域的发展。这些模型的性能在很大程度上受到损失函数选择的影响,损失函数用于量化模型预测与真实值之间的差异,以指导学习过程(Ansari等人,2024年)。常用的损失函数(如交叉熵、Dice、Tversky和Hausdorff距离)往往难以处理医学图像中病变边界的固有模糊性和变异性(Qiao等人,2024年)。这种限制通常会导致次优的分割结果,尤其是在病变边界不清晰的情况下(Chen等人,2023年)。
在基于深度学习(DL)的分割模型中常用的损失函数包括像素级(例如,交叉熵)和区域级(例如,Dice损失)损失函数。像素级损失函数仅关注单个像素,没有考虑相邻像素之间的空间关系和对象边界的结构完整性。区域级损失函数在测量预测区域与真实区域之间的重叠方面是有效的,但它们对细微边界细节的敏感度较低。尽管这些方法推动了该领域的发展,但在应用于医学图像时存在根本性限制,因为在医学图像中精确的边界勾画对于准确的诊断和治疗计划至关重要(Chen等人,2023年)。
尽管现有研究中的DL模型已经使用了损失函数,但在图像分割过程中仍面临边界模糊和数据不平衡的问题。此外,基于模糊的损失函数在医学图像中的可行性尚未得到充分研究。如图1所示,恶性肿瘤边界的不规则形状和重叠区域导致了模糊性。准确分割医学图像是一个重大挑战,尤其是在勾画模糊的病变边界时。传统损失函数在处理模糊边界时表现不佳,这降低了DL模型的鲁棒性和可靠性,而数据不平衡问题限制了它们在医学图像分割中的应用,从而导致过拟合和DL模型泛化能力的下降。
为了解决这些问题,尽管模糊相似性成分在数学上较为复杂,本研究提出了一种基于模糊粗糙集的损失函数。具体来说,它在设计中使用了模糊相似性成分。目前还缺乏将模糊相似性关系与能够处理边界模糊和数据不平衡问题的损失函数相结合的方法。为了弥合这一差距,本研究提出了一种新的损失函数,通过将模糊粗糙集与损失函数结合,利用模糊相似性关系来定义模糊集的下界和上界近似。所提出的损失函数提高了深度神经网络的训练稳定性。将模糊相似性与损失函数结合使用,提升了DL模型的分割性能。
模糊粗糙损失函数的方法如下:FRS损失函数是通过将新的模糊相似性损失和边界不确定性模型与凸组合相结合而开发的。在这里,边界不确定性模型是通过使用模糊下界和上界近似以及隶属权重来形成的。模糊相似性损失涉及像素的模糊加权隶属关系和预测分割与真实分割之间的相似性关系。FRS损失函数受到模糊粗糙集原理的启发,旨在以可微分且计算效率高的方式模拟病变外观的不确定性和变异性。具体而言,它利用模糊集隶属函数来表示每个像素属于病变类或背景类的程度。它还利用边界近似来考虑定义这些类别之间精确边界的不确定性。
本研究的主要贡献有三方面:
  • 设计了一种新的FRS损失函数,该函数使用模糊相似性关系、模糊下界和上界近似(模糊粗糙集)以及权重隶属关系。它有效地处理了医学图像分割中的边界模糊问题。
  • 在五个不同的医学成像数据集上对该损失函数进行了广泛实验,证明了其在分割准确性和边界精度方面的持续改进。
  • 对我们的方法的计算效率和参数敏感性进行了全面分析,展示了其在临床环境中的实际应用性。
本文的其余部分组织如下:第2节回顾了医学图像分割损失函数和模糊粗糙集理论的相关工作。第3节详细介绍了所提出的FRS损失函数及其组成部分。第4节描述了实验设置,包括数据集、实现细节和评估指标。第5节展示了实验结果并进行了分析。第6节讨论了结果。最后,第7节总结了本文并讨论了潜在的未来研究方向。

部分摘录

背景

本节全面概述了构成我们提出方法基础的关键概念和相关工作。

提出的模糊粗糙损失函数

本节详细介绍了所提出的模糊粗糙集(FRS)损失函数,该函数将模糊相似性指标与边界不确定性模型相结合,以提高医学图像分割的效果。

实验设置

本节详细介绍了用于评估所提出的FRS损失函数的实验框架,包括数据集、实现细节和评估指标。

结果与分析

本节对所提出的FRS损失函数在多个医学图像分割任务上的性能进行了全面评估。我们首先分析了各个数据集上的性能,然后进行了跨数据集比较、消融研究和计算效率分析。

讨论

实验结果表明,所提出的FRS损失函数在多个数据集和网络架构上一致地提高了医学图像分割性能。本节在现有文献的背景下解释了这些发现,并讨论了其含义。
重要的是,结果还揭示了在极端类别不平衡条件下,以区域重叠为导向的目标与对边界敏感的目标之间的特征性权衡。

结论

本文提出了一种用于医学图像分割的新FRS损失函数,解决了两个关键问题:边界模糊和类别不平衡。所提出的方法结合了模糊相似性指标(通过像素级隶属评估捕获细节)和边界不确定性模型(通过下界和上界处理边界不确定性和类别不平衡)。
实验结果表明,FRS在五个不同的医学成像任务上取得了优异的性能。

CRediT作者贡献声明

Mohsin Furkh Dar:撰写 – 审稿与编辑、原始草稿撰写、可视化、验证、软件、资源、方法论、概念化。Avatharam Ganivada:撰写 – 审稿与编辑、验证、项目管理、方法论。

利益冲突声明

作者声明他们没有已知的竞争性财务利益或个人关系可能影响本文报告的工作。

生物通微信公众号
微信
新浪微博


生物通 版权所有