RP-Unet: a Unet-based network with RNNPool enables computation-efficient polyp segmentation

Yue Chen; Zhiwen Liu; Yonggang Shi

doi:10.1117/12.2604803

6 August 2021 RP-Unet: a Unet-based network with RNNPool enables computation-efficient polyp segmentation

Yue Chen, Zhiwen Liu, Yonggang Shi

Proceedings Volume 11913, Sixth International Workshop on Pattern Recognition; 1191302 (2021) https://doi.org/10.1117/12.2604803
Event: Sixth International Workshop on Pattern Recognition, 2021, Chengdu, China

Abstract

The incidence of colon cancer has shown an upward trend in recent years, and the appearance of colon polyps is one of the signs of colon cancer. The detection and segmentation of colon polyps are one of the doctors' auxiliary diagnostic methods. However, the increasing number of model parameters and inference memory requirements make the engineering of polyp segmentation models a challenging task. In this paper, an efficient polyp segmentation model based on Unet and RNNPool named RP-Unet is proposed. The first two blocks consisted of two convolutional and max pooling layers in Unet are replaced with the proposed RNNPool Down and Fuse (RDF) modules to rapidly downsample and fuse the input feature maps, and they also provide feature maps for skip connection. The last two blocks in the encoder are replaced with the proposed Double Convolution with Residual connection and RNNPool (DCRR) modules, in which the convolution layers are residually connected, and the max pooling layer is replaced directly with RNNPool. In the two proposed modules, up mapping and channel mapping are used to strengthen feature propagation by mapping activation maps logically instead of allocating unnecessary memory. The proposed RP-Unet is evaluated on two polyp segmentation datasets, and experiments show that the peak inference memory is reduced by almost 22%, while the segmentation accuracy is not significantly reduced.

Citation Download Citation

Yue Chen, Zhiwen Liu, and Yonggang Shi "RP-Unet: a Unet-based network with RNNPool enables computation-efficient polyp segmentation", Proc. SPIE 11913, Sixth International Workshop on Pattern Recognition, 1191302 (6 August 2021); https://doi.org/10.1117/12.2604803

ACCESS THE FULL ARTICLE

INSTITUTIONAL
Select your institution to access the SPIE Digital Library.

SELECT YOUR INSTITUTION

PERSONAL
Sign in with your SPIE account to access your personal subscriptions or to use specific features such as save to my library, sign up for alerts, save searches, etc.

PERSONAL SIGN IN

No SPIE Account? Create one

PURCHASE THIS CONTENT

SUBSCRIBE TO DIGITAL LIBRARY

50 downloads per 1-year subscription

Members: $195

Non-members: $335 ADD TO CART

25 downloads per 1 - year subscription

Members: $145

Non-members: $250 ADD TO CART

PURCHASE SINGLE ARTICLE

Includes PDF, HTML & Video, when available