4 个月前

PraNet:用于息肉分割的并行反向注意力网络

PraNet:用于息肉分割的并行反向注意力网络

摘要

结肠镜检查是一种有效的检测结直肠息肉的技术,而这些息肉与结直肠癌高度相关。在临床实践中,从结肠镜图像中分割息肉具有重要意义,因为它为诊断和手术提供了宝贵的信息。然而,准确的息肉分割是一项具有挑战性的任务,主要原因有两个:(i)同一类型的息肉在大小、颜色和纹理上存在多样性;(ii)息肉与其周围黏膜之间的边界不明显。为了应对这些挑战,我们提出了一种并行反向注意力网络(PraNet),用于结肠镜图像中的准确息肉分割。具体而言,我们首先使用并行部分解码器(PPD)聚合高层特征。基于这些组合特征,我们生成一个全局图作为后续组件的初始引导区域。此外,我们利用反向注意力(RA)模块挖掘边界线索,该模块能够建立区域与边界线索之间的关系。得益于区域和边界之间的反复合作机制,我们的PraNet能够校正任何错位预测,从而提高分割精度。对五个具有挑战性的数据集进行的六项指标的定量和定性评估表明,我们的PraNet显著提高了分割精度,并在泛化能力和实时分割效率方面表现出诸多优势。

代码仓库

GewelsJI/PNS-Net
pytorch
GitHub 中提及
DengPingFan/PraNet
官方
pytorch
yuwenlo/hardnet-dfus
pytorch
GitHub 中提及

基准测试

基准方法指标
camouflaged-object-segmentation-on-camoPraNet
MAE: 0.094
S-Measure: 0.769
Weighted F-Measure: 0.663
camouflaged-object-segmentation-on-pcod-1200PraNet
S-Measure: 0.904
medical-image-segmentation-on-cvc-clinicdbPraNet
mean Dice: 0.8990
medical-image-segmentation-on-cvc-colondbPraNet
Average MAE: 0.045
S-Measure: 0.819
mIoU: 0.649
max E-Measure: 0.869
mean Dice: 0.709
medical-image-segmentation-on-etisPraNet
Average MAE: 0.031
S-Measure: 0.794
mIoU: 0.5670
max E-Measure: 0.841
mean Dice: 0.6280
medical-image-segmentation-on-kvasir-segPraNet
Average MAE: 0.030
S-Measure: 0.915
mIoU: 0.849
max E-Measure: 0.948
mean Dice: 0.898
video-polyp-segmentation-on-sun-seg-easyPraNet
Dice: 0.621
S measure: 0.733
Sensitivity: 0.524
mean E-measure: 0.753
mean F-measure: 0.632
weighted F-measure: 0.572
video-polyp-segmentation-on-sun-seg-hardPraNet
Dice: 0.598
S-Measure: 0.717
Sensitivity: 0.512
mean E-measure: 0.735
mean F-measure: 0.607
weighted F-measure: 0.544

用 AI 构建 AI

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

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

Hyper Newsletters

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