4 个月前

TGANet:文本引导注意力机制以改进息肉分割

TGANet:文本引导注意力机制以改进息肉分割

摘要

结肠镜检查是诊断结直肠疾病的金标准,但其效果高度依赖于操作者的技术水平。自动息肉分割技术可以减少漏诊率,并在早期阶段及时治疗结肠癌。尽管已经开发了多种深度学习方法用于此任务,但息肉大小的变化可能会影响模型训练,从而导致模型对大多数训练样本的大小属性进行优化,而对不同大小的息肉产生次优结果。在本研究中,我们利用了与息肉大小和数量相关的特征,在训练过程中引入了文本注意力机制。我们引入了一个辅助分类任务来加权基于文本的嵌入向量,使网络能够学习额外的特征表示,从而更好地适应不同大小的息肉,并且能够处理多发性息肉的情况。实验结果显示,这些添加的文本嵌入向量显著提高了模型的整体性能,优于现有的最先进的分割方法。我们探索了四个不同的数据集,并提供了针对特定大小息肉改进的见解。所提出的文本引导注意力网络(TGANet)在不同数据集中对各种大小的息肉具有良好的泛化能力。

代码仓库

基准测试

基准方法指标
medical-image-segmentation-on-bkai-ighTGANet
Average Dice: 0.9023
mIoU: 0.8409
medical-image-segmentation-on-kvasir-segTGA-Net
mIoU: 0.8330
mean Dice: 0.8982
polyp-segmentation-on-kvasir-segTGA-Net
mDice: 0.8982
mIoU: 0.8330

用 AI 构建 AI

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

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

Hyper Newsletters

订阅我们的最新资讯
我们会在北京时间 每周一的上午九点 向您的邮箱投递本周内的最新更新
邮件发送服务由 MailChimp 提供
TGANet:文本引导注意力机制以改进息肉分割 | 论文 | HyperAI超神经