3 个月前

CaraNet:用于小医学目标分割的上下文轴向反向注意力网络

CaraNet:用于小医学目标分割的上下文轴向反向注意力网络

摘要

准确且可靠地分割医学图像对于疾病诊断与治疗具有重要意义。然而,由于目标对象在尺寸、形状以及成像模态上存在巨大差异,该任务极具挑战性。近年来,众多卷积神经网络(CNN)被设计用于图像分割任务,并取得了显著成果。然而,现有研究大多未能充分考虑目标对象的尺寸差异,导致在小目标分割方面表现不佳,这可能严重影响疾病的早期检测。为此,本文提出一种上下文轴向逆向注意力网络(Context Axial Reserve Attention Network, CaraNet),以显著提升小目标的分割性能,相较多个近期先进的分割模型展现出优越性。我们在脑肿瘤(BraTS 2018)以及息肉(Kvasir-SEG、CVC-ColonDB、CVC-ClinicDB、CVC-300 和 ETIS-LaribPolypDB)分割数据集上对所提CaraNet进行了实验验证。结果表明,CaraNet在平均Dice系数上达到最优水平,尤其在小尺寸医学目标的分割任务中展现出显著优势。

代码仓库

AngeLouCN/CaraNet
官方
pytorch

基准测试

基准方法指标
medical-image-segmentation-on-cvc-clinicdbCaraNet
Average MAE: 0.007
S-Measure: 0.954
mIoU: 0.887
max E-Measure: 0.991
mean Dice: 0.936
medical-image-segmentation-on-cvc-colondbCaraNet
Average MAE: 0.042
S-Measure: 0.853
mIoU: 0.689
max E-Measure: 0.902
mean Dice: 0.773
medical-image-segmentation-on-etisCaraNet
Average MAE: 0.017
S-Measure: 0.868
mIoU: 0.672
max E-Measure: 0.894
mean Dice: 0.747
medical-image-segmentation-on-kvasir-segCaraNet
Average MAE: 0.023
S-Measure: 0.929
mIoU: 0.865
max E-Measure: 0.968
mean Dice: 0.918

用 AI 构建 AI

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

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

Hyper Newsletters

订阅我们的最新资讯
我们会在北京时间 每周一的上午九点 向您的邮箱投递本周内的最新更新
邮件发送服务由 MailChimp 提供
CaraNet:用于小医学目标分割的上下文轴向反向注意力网络 | 论文 | HyperAI超神经