Medical Image Segmentation Using Advanced Attention UNet

EasyChair Preprint 15696

Abstract

Medical image segmentation across various modalities and domains is a challenging task. This paper presents a novel model designed to address these challenges effectively. Our proposed model incorporates several key innovations. In the decoder path, we utilize bilinear unpooling, global attention gates, and residual convolution blocks (RCBs) to up-sample and refine feature maps. The encoder path alternates between max pooling and RCBs to progressively capture higher-level features, ensuring efficient feature extraction. We incorporate skip connections between the corresponding encoder and decoder layers to preserve spatial information. The final segmented output is generated through a 1 X 1 convolution layer. We combine dice loss and cross-entropy loss for training to optimize segmentation performance. We evaluated our method on multiple state-of-the-art datasets, achieving an average accuracy exceeding 96.5% across all modalities and data types—outperforming current state-of-the-art approaches.

Keyphrases: Bilinear Unpooling, Cross Entropy Loss, Dice loss, Encoder-Decoder Architecture, Global Attention Gates, Medical image segmentation, Residual Convolution Blocks (RCBs)

@booklet{EasyChair:15696,
  author    = {Rikathi Pal and Somoballi Ghoshal and Amlan Chakrabarti},
  title     = {Medical Image Segmentation Using Advanced Attention UNet},
  howpublished = {EasyChair Preprint 15696},
  year      = {EasyChair, 2025}}