Abstract:Medical image segmentation has made a significant contribution towards delivering affordable healthcare by facilitating the automatic identification of anatomical structures and other regions of interest. Although convolution neural networks have become prominent in the field of medical image segmentation, they suffer from certain limitations. In this study, we present a reliable framework for producing performant outcomes for the segmentation of pathological structures of 2D medical images. Our framework consists of a novel deep learning architecture, called deep multi-level attention dilated residual neural network (MADR-Net), designed to improve the performance of medical image segmentation. MADR-Net uses a U-Net encoder/decoder backbone in combination with multi-level residual blocks and atrous pyramid scene parsing pooling. To improve the segmentation results, channel-spatial attention blocks were added in the skip connection to capture both the global and local features and superseded the bottleneck layer with an ASPP block. Furthermore, we introduce a hybrid loss function that has an excellent convergence property and enhances the performance of the medical image segmentation task. We extensively validated the proposed MADR-Net on four typical yet challenging medical image segmentation tasks: (1) Left ventricle, left atrium, and myocardial wall segmentation from Echocardiogram images in the CAMUS dataset, (2) Skin cancer segmentation from dermoscopy images in ISIC 2017 dataset, (3) Electron microscopy in FIB-SEM dataset, and (4) Fluid attenuated inversion recovery abnormality from MR images in LGG segmentation dataset. The proposed algorithm yielded significant results when compared to state-of-the-art architectures such as U-Net, Residual U-Net, and Attention U-Net. The proposed MADR-Net consistently outperformed the classical U-Net by 5.43%, 3.43%, and 3.92% relative improvement in terms of dice coefficient, respectively, for electron microscopy, dermoscopy, and MRI. The experimental results demonstrate superior performance on single and multi-class datasets and that the proposed MADR-Net can be utilized as a baseline for the assessment of cross-dataset and segmentation tasks.

FocusNet: An attention-based Fully Convolutional Network for Medical Image Segmentation

Focal Attention Networks: optimising attention for biomedical image segmentation

A multi-attention and depthwise separable convolution network for medical image segmentation

Focus U-Net: A novel dual attention-gated CNN for polyp segmentation during colonoscopy

MADR-Net: multi-level attention dilated residual neural network for segmentation of medical images

CA-Net: Comprehensive Attention Convolutional Neural Networks for Explainable Medical Image Segmentation

TA-Net: Triple attention network for medical image segmentation

A Multi-Scale Context Aware Attention Model for Medical Image Segmentation

TSCA-Net: Transformer based spatial-channel attention segmentation network for medical images

Spatial-Frequency Dual Progressive Attention Network For Medical Image Segmentation

CFANet: Context Feature Fusion and Attention Mechanism Based Network for Small Target Segmentation in Medical Images

HD 2 A-Net: A novel dual gated attention network using comprehensive hybrid dilated convolutions for medical image segmentation

Segmentation of medical images using an attention embedded lightweight network

Y-CA-Net: A Convolutional Attention Based Network for Volumetric Medical Image Segmentation

A hybrid approach for improving U-Net variants in medical image segmentation

AEC-Net: Attention and Edge Constraint Network for Medical Image Segmentation

FANet: A Feedback Attention Network for Improved Biomedical Image Segmentation

CSCA U-Net: A channel and space compound attention CNN for medical image segmentation

DoubleU-NetPlus: A Novel Attention and Context Guided Dual U-Net with Multi-Scale Residual Feature Fusion Network for Semantic Segmentation of Medical Images

FCSU-Net: A novel full-scale Cross-dimension Self-attention U-Net with collaborative fusion of multi-scale feature for medical image segmentation