
摘要
近年来,医学图像分割已成为计算机辅助诊断领域的一个重要应用。本文首次提出了一种基于图卷积的新解码器——级联图卷积注意力解码器(G-CASCADE),用于二维医学图像分割。G-CASCADE 利用高效的图卷积模块逐步细化由分层变压器编码器生成的多阶段特征图。编码器通过自注意力机制捕捉长距离依赖关系,而解码器则利用图卷积模块的全局感受野来保留长距离信息,从而进一步优化特征图。我们对多个变压器编码器在五个医学图像分割任务(即腹部器官、心脏器官、息肉病变、皮肤病变和视网膜血管)上进行了严格的评估,结果显示我们的模型优于其他最先进的(SOTA)方法。此外,我们还证明了我们的解码器在参数量减少 80.8% 和浮点运算量减少 82.3% 的情况下,仍能实现比 SOTA CASCADE 解码器更高的 DICE 分数。我们的解码器可以轻松与其他分层编码器结合使用,适用于通用语义分割和医学图像分割任务。
代码仓库
SLDGroup/G-CASCADE
官方
pytorch
GitHub 中提及
基准测试
| 基准 | 方法 | 指标 |
|---|---|---|
| medical-image-segmentation-on-automatic | MERIT-GCASCADE | Avg DSC: 92.23 |
| medical-image-segmentation-on-automatic | PVT-GCASCADE | Avg DSC: 91.95 |
| medical-image-segmentation-on-chase-db1 | PVT-GCASCADE | DSC: 0.8251 |
| medical-image-segmentation-on-chase-db1 | MERIT-GCASCADE | DSC: 0.8267 |
| medical-image-segmentation-on-cvc-clinicdb | PVT-GCASCADE | mIoU: 0.9018 mean Dice: 0.9468 |
| medical-image-segmentation-on-cvc-colondb | PVT-GCASCADE | mIoU: 0.7460 mean Dice: 0.8261 |
| medical-image-segmentation-on-drive-1 | MERIT-GCASCADE | F1 score: 0.8290 Recall: 0.8281 Specificity: 0.9844 mIoU: 0.7081 |
| medical-image-segmentation-on-drive-1 | PVT-GCASCADE | F1 score: 0.8210 Recall: 0.83 Specificity: 0.9822 mIoU: 0.697 |
| medical-image-segmentation-on-isic-2018-1 | PVT-GCASCADE | DSC: 91.51 mIoU: 86.53 |
| medical-image-segmentation-on-kvasir-seg | PVT-GCASCADE | mIoU: 0.8790 mean Dice: 0.9274 |
| medical-image-segmentation-on-miccai-2015-1 | MERIT-GCASCADE | Avg DSC: 84.54 Avg HD: 10.38 |
| medical-image-segmentation-on-miccai-2015-1 | PVT-GCASCADE | Avg DSC: 83.28 Avg HD: 15.83 |
| retinal-vessel-segmentation-on-chase_db1 | MERIT-GCASCADE | F1 score: 0.8267 Sensitivity: 0.8493 mIOU: 0.7050 |
| retinal-vessel-segmentation-on-chase_db1 | PVT-GCASCADE | F1 score: 0.8251 Sensitivity: 0.8584 mIOU: 0.7024 |
| retinal-vessel-segmentation-on-drive | MERIT-GCASCADE | Accuracy: 0.9707 F1 score: 0.8290 Specificity: 0.9844 mIoU: 0.7081 sensitivity: 0.8281 |
| retinal-vessel-segmentation-on-drive | PVT-GCASCADE | Accuracy: 0.9689 F1 score: 0.8210 Specificity: 0.9822 mIoU: 0.6970 sensitivity: 0.83 |