Progressively Normalized Self-Attention Network for Video Polyp Segmentation

Fuente: arXiv
Saved in:
Bibliographic Details
Main Authors: Ji, Ge-Peng, Chou, Yu-Cheng, Fan, Deng-Ping, Chen, Geng, Fu, Huazhu, Jha, Debesh, Shao, Ling
Format: Preprint
Published: 2021
Subjects:
Online Access:
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866911779067527168
author Ji, Ge-Peng
Chou, Yu-Cheng
Fan, Deng-Ping
Chen, Geng
Fu, Huazhu
Jha, Debesh
Shao, Ling
author_facet Ji, Ge-Peng
Chou, Yu-Cheng
Fan, Deng-Ping
Chen, Geng
Fu, Huazhu
Jha, Debesh
Shao, Ling
contents Existing video polyp segmentation (VPS) models typically employ convolutional neural networks (CNNs) to extract features. However, due to their limited receptive fields, CNNs can not fully exploit the global temporal and spatial information in successive video frames, resulting in false-positive segmentation results. In this paper, we propose the novel PNS-Net (Progressively Normalized Self-attention Network), which can efficiently learn representations from polyp videos with real-time speed (~140fps) on a single RTX 2080 GPU and no post-processing. Our PNS-Net is based solely on a basic normalized self-attention block, equipping with recurrence and CNNs entirely. Experiments on challenging VPS datasets demonstrate that the proposed PNS-Net achieves state-of-the-art performance. We also conduct extensive experiments to study the effectiveness of the channel split, soft-attention, and progressive learning strategy. We find that our PNS-Net works well under different settings, making it a promising solution to the VPS task.
format Preprint
id arxiv_https___arxiv_org_abs_2105_08468
institution arXiv
publishDate 2021
record_format arxiv
spellingShingle Progressively Normalized Self-Attention Network for Video Polyp Segmentation
Ji, Ge-Peng
Chou, Yu-Cheng
Fan, Deng-Ping
Chen, Geng
Fu, Huazhu
Jha, Debesh
Shao, Ling
Computer Vision and Pattern Recognition
Existing video polyp segmentation (VPS) models typically employ convolutional neural networks (CNNs) to extract features. However, due to their limited receptive fields, CNNs can not fully exploit the global temporal and spatial information in successive video frames, resulting in false-positive segmentation results. In this paper, we propose the novel PNS-Net (Progressively Normalized Self-attention Network), which can efficiently learn representations from polyp videos with real-time speed (~140fps) on a single RTX 2080 GPU and no post-processing. Our PNS-Net is based solely on a basic normalized self-attention block, equipping with recurrence and CNNs entirely. Experiments on challenging VPS datasets demonstrate that the proposed PNS-Net achieves state-of-the-art performance. We also conduct extensive experiments to study the effectiveness of the channel split, soft-attention, and progressive learning strategy. We find that our PNS-Net works well under different settings, making it a promising solution to the VPS task.
title Progressively Normalized Self-Attention Network for Video Polyp Segmentation
topic Computer Vision and Pattern Recognition
url https://arxiv.org/abs/2105.08468