CSTA-2P1D UNet: Consecutive Spatio-Temporal Attention for Multi-Scale 3D Pancreas Segmentation
Bharat Giddwani, Shruti Pandey, Hitesh Tekchandani, Shrish Verma · 2020
Accurate pancreas segmentation is a crucial task for the diagnosis of diseases as well as for developing modern surgical tools. In this paper, a novel architecture, named Consecutive Spatio-Temporal Attention (CSTA) 2 Plus 1 Dimensional (2P1D) UNet, is proposed for the volumetric segmentation of the pancreas in CT images. CSTA mechanism is inspired by recent advancements in Squeeze & Excitation (SE) module. It recalibrates feature maps, by multiplying learnable spatial and channel-wise attention maps, sequentially. Further, incorporation of Multi-Sale Output (MSO) and Group Normalization (GN) improves the localization performance and training stability of the model, respectively. Moreover, 2P1D convolution is introduced to enhance the segmentation results of our proposed architecture. Smoothed Dice Loss (SDL) is used as a loss function for training of the proposed model. Collectively, a state-of-the-art performance of 83.70%, 87.70%, and 85.60% on dice score, sensitivity, and precision is achieved, verifying the model's effectiveness for the pancreas segmentation.