4 个月前

G-CASCADE:用于2D医学图像分割的高效级联图卷积解码器

G-CASCADE:用于2D医学图像分割的高效级联图卷积解码器

摘要

近年来,医学图像分割已成为计算机辅助诊断领域的一个重要应用。本文首次提出了一种基于图卷积的新解码器——级联图卷积注意力解码器(G-CASCADE),用于二维医学图像分割。G-CASCADE 利用高效的图卷积模块逐步细化由分层变压器编码器生成的多阶段特征图。编码器通过自注意力机制捕捉长距离依赖关系,而解码器则利用图卷积模块的全局感受野来保留长距离信息,从而进一步优化特征图。我们对多个变压器编码器在五个医学图像分割任务(即腹部器官、心脏器官、息肉病变、皮肤病变和视网膜血管)上进行了严格的评估,结果显示我们的模型优于其他最先进的(SOTA)方法。此外,我们还证明了我们的解码器在参数量减少 80.8% 和浮点运算量减少 82.3% 的情况下,仍能实现比 SOTA CASCADE 解码器更高的 DICE 分数。我们的解码器可以轻松与其他分层编码器结合使用,适用于通用语义分割和医学图像分割任务。

代码仓库

SLDGroup/G-CASCADE
官方
pytorch
GitHub 中提及

基准测试

基准方法指标
medical-image-segmentation-on-automaticMERIT-GCASCADE
Avg DSC: 92.23
medical-image-segmentation-on-automaticPVT-GCASCADE
Avg DSC: 91.95
medical-image-segmentation-on-chase-db1PVT-GCASCADE
DSC: 0.8251
medical-image-segmentation-on-chase-db1MERIT-GCASCADE
DSC: 0.8267
medical-image-segmentation-on-cvc-clinicdbPVT-GCASCADE
mIoU: 0.9018
mean Dice: 0.9468
medical-image-segmentation-on-cvc-colondbPVT-GCASCADE
mIoU: 0.7460
mean Dice: 0.8261
medical-image-segmentation-on-drive-1MERIT-GCASCADE
F1 score: 0.8290
Recall: 0.8281
Specificity: 0.9844
mIoU: 0.7081
medical-image-segmentation-on-drive-1PVT-GCASCADE
F1 score: 0.8210
Recall: 0.83
Specificity: 0.9822
mIoU: 0.697
medical-image-segmentation-on-isic-2018-1PVT-GCASCADE
DSC: 91.51
mIoU: 86.53
medical-image-segmentation-on-kvasir-segPVT-GCASCADE
mIoU: 0.8790
mean Dice: 0.9274
medical-image-segmentation-on-miccai-2015-1MERIT-GCASCADE
Avg DSC: 84.54
Avg HD: 10.38
medical-image-segmentation-on-miccai-2015-1PVT-GCASCADE
Avg DSC: 83.28
Avg HD: 15.83
retinal-vessel-segmentation-on-chase_db1MERIT-GCASCADE
F1 score: 0.8267
Sensitivity: 0.8493
mIOU: 0.7050
retinal-vessel-segmentation-on-chase_db1PVT-GCASCADE
F1 score: 0.8251
Sensitivity: 0.8584
mIOU: 0.7024
retinal-vessel-segmentation-on-driveMERIT-GCASCADE
Accuracy: 0.9707
F1 score: 0.8290
Specificity: 0.9844
mIoU: 0.7081
sensitivity: 0.8281
retinal-vessel-segmentation-on-drivePVT-GCASCADE
Accuracy: 0.9689
F1 score: 0.8210
Specificity: 0.9822
mIoU: 0.6970
sensitivity: 0.83

用 AI 构建 AI

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

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

Hyper Newsletters

订阅我们的最新资讯
我们会在北京时间 每周一的上午九点 向您的邮箱投递本周内的最新更新
邮件发送服务由 MailChimp 提供
G-CASCADE:用于2D医学图像分割的高效级联图卷积解码器 | 论文 | HyperAI超神经