4 个月前

逐步归一化自注意力网络用于视频息肉分割

逐步归一化自注意力网络用于视频息肉分割

摘要

现有的视频息肉分割(VPS)模型通常采用卷积神经网络(CNNs)来提取特征。然而,由于其有限的感受野,CNNs无法充分利用连续视频帧中的全局时序和空间信息,导致出现假阳性分割结果。在本文中,我们提出了一种新颖的PNS-Net(Progressively Normalized Self-attention Network,逐步归一化自注意力网络),该网络能够在单个RTX 2080 GPU上以实时速度(约140帧/秒)高效地从息肉视频中学习表示,并且无需任何后处理。我们的PNS-Net仅基于一个基本的归一化自注意力块,完全配备了递归机制和卷积神经网络。在具有挑战性的VPS数据集上的实验表明,所提出的PNS-Net达到了最先进的性能。我们还进行了广泛的实验,研究了通道拆分、软注意力和逐步学习策略的有效性。实验结果表明,我们的PNS-Net在不同的设置下均表现出色,使其成为解决VPS任务的一个有前景的方案。

代码仓库

GewelsJI/PNS-Net
pytorch
GitHub 中提及
gewelsji/vps
pytorch
GitHub 中提及
GewelsJI/FSNet
pytorch
GitHub 中提及

基准测试

基准方法指标
video-polyp-segmentation-on-sun-seg-easyPNSNet
Dice: 0.676
S measure: 0.767
Sensitivity: 0.574
mean E-measure: 0.744
mean F-measure: 0.664
weighted F-measure: 0.616
video-polyp-segmentation-on-sun-seg-hardPNSNet
Dice: 0.675
S-Measure: 0.767
Sensitivity: 0.579
mean E-measure: 0.755
mean F-measure: 0.656
weighted F-measure: 0.609

用 AI 构建 AI

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

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

Hyper Newsletters

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