3 个月前

基于ResUNet++、条件随机场与测试时增强的结直肠息肉分割综合研究

基于ResUNet++、条件随机场与测试时增强的结直肠息肉分割综合研究

摘要

结肠镜检查被认为是检测结直肠癌及其癌前病变的金标准。然而,现有的检查方法普遍存在较高的漏诊率,许多异常病灶未能被及时发现。基于先进机器学习算法的计算机辅助诊断(Computer-Aided Diagnosis, CAD)系统被视为一项颠覆性技术,能够识别内镜检查过程中医生可能忽略的结肠区域,从而帮助更准确地检测和表征病变。在前期工作中,我们提出了ResUNet++网络架构,并证实其在性能上优于U-Net和ResUNet等同类模型。本文进一步证明,通过引入条件随机场(Conditional Random Field, CRF)和测试时增强(Test-Time Augmentation, TTA)策略,可显著提升ResUNet++架构的整体预测性能。我们进行了广泛的实验评估,并在六个公开可用的数据集上验证了该方法的有效性:Kvasir-SEG、CVC-ClinicDB、CVC-ColonDB、ETIS-Larib Polyp DB、ASU-Mayo Clinic结肠镜视频数据库以及CVC-VideoClinicDB。此外,我们将所提出的架构及其训练模型与当前其他先进方法进行了系统比较。为评估ResUNet++在不同公开结肠息肉数据集上的泛化能力,以支持其在真实临床环境中的应用,我们开展了全面的跨数据集评估。实验结果表明,引入CRF与TTA不仅在单一数据集上提升了分割性能,在跨数据集场景下也表现出显著的性能增强,充分验证了该方法的鲁棒性与实用性。

基准测试

基准方法指标
medical-image-segmentation-on-cvcResUNet++ + TTA + CRF
Dice: 0.8130
Recall: 0.6875
mIoU: 0.8477
precision: 0.6276
medical-image-segmentation-on-cvcResUNet++ + TTA
Dice: 0.8125
Recall: 0.6896
mIoU: 0.8467
precision: 0.6421
medical-image-segmentation-on-cvcResUNet++ + CRF
Dice: 0.8811
Recall: 0.7743
mIoU: 0.8739
precision: 0.6706
medical-image-segmentation-on-cvc-clinicdbResUNet++ + TTA
mean Dice: 0.9020
medical-image-segmentation-on-cvc-clinicdbResUNet++ + CRF+ TTA
mean Dice: 0.9017
medical-image-segmentation-on-cvc-colondbResUNet++ + TTA
mIoU: 0.8466
mean Dice: 0.8474
medical-image-segmentation-on-etisResUNet++ + TTA
mIoU: 0.7458
mean Dice: 0.6136
medical-image-segmentation-on-kvasir-segResUNet++ + TTA + CRF
FPS: 69.59
mIoU: 0.7800
mean Dice: 0.8508

用 AI 构建 AI

从想法到上线——通过免费 AI 协同编程、开箱即用的环境和市场最优价格的 GPU 加速您的 AI 开发

AI 协同编程
即用型 GPU
最优价格
立即开始

Hyper Newsletters

订阅我们的最新资讯
我们会在北京时间 每周一的上午九点 向您的邮箱投递本周内的最新更新
邮件发送服务由 MailChimp 提供
基于ResUNet++、条件随机场与测试时增强的结直肠息肉分割综合研究 | 论文 | HyperAI超神经