3 个月前

逐步特征融合:局部引导全局

逐步特征融合:局部引导全局

摘要

结肠镜检查目前是检测结肠息肉最高效且被广泛认可的技术,对于结直肠癌的早期筛查与预防具有重要意义。然而,由于结肠息肉在大小上存在差异,形态特征复杂,且其与黏膜组织之间的边界往往不清晰,因此实现息肉的精确分割仍面临挑战。深度学习在息肉分割任务中已展现出优异性能,成为实现高精度分割的主流方法。然而,受限于息肉图像的结构特性以及息肉形态的多样性,现有深度学习模型容易在当前数据集上出现过拟合现象,导致模型在面对未见过的结肠镜图像时泛化能力不足。为解决这一问题,本文提出一种新型的医学图像分割前沿模型——SSFormer。该模型采用金字塔式Transformer编码器,以提升模型的泛化能力。具体而言,我们设计的渐进式局部解码器(Progressive Locality Decoder)可与金字塔Transformer主干网络有效协同,强化局部特征表达,同时抑制注意力机制的过度分散。实验结果表明,SSFormer在模型学习能力与泛化性能评估方面均达到了当前最优水平。

代码仓库

Qiming-Huang/ssformer
官方
pytorch
GitHub 中提及

基准测试

基准方法指标
medical-image-segmentation-on-2018-dataSSFormer-L
Dice: 0.9230
mIoU: 0.8614
medical-image-segmentation-on-cvc-clinicdbSSFormer-L
mIoU: 0.8995
mean Dice: 0.9447
medical-image-segmentation-on-cvc-colondbSSFormer-L
mIoU: 0.721
mean Dice: 0.802
medical-image-segmentation-on-etisSSFormer-L
mIoU: 0.720
mean Dice: 0.796
medical-image-segmentation-on-kvasir-segSSFormer-L
mIoU: 0.8905
mean Dice: 0.9357

用 AI 构建 AI

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

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

Hyper Newsletters

订阅我们的最新资讯
我们会在北京时间 每周一的上午九点 向您的邮箱投递本周内的最新更新
邮件发送服务由 MailChimp 提供
逐步特征融合:局部引导全局 | 论文 | HyperAI超神经