Abstract:Medical image segmentation is critical for efficient diagnosis of diseases and treatment planning. In recent years, convolutional neural networks (CNN)-based methods, particularly U-Net and its variants, have achieved remarkable results on medical image segmentation tasks. However, they do not always work consistently on images with complex structures and large variations in regions of interest (ROI). This could be due to the fixed geometric structure of the receptive fields used for feature extraction and repetitive down-sampling operations that lead to information loss. To overcome these problems, the standard U-Net architecture is modified in this work by replacing the convolution block with a dilated convolution block to extract multi-scale context features with varying sizes of receptive fields, and adding a dilated inception block between the encoder and decoder paths to alleviate the problem of information recession and the semantic gap between features. Furthermore, the input of each dilated convolution block is added to the output through a squeeze and excitation unit, which alleviates the vanishing gradient problem and improves overall feature representation by re-weighting the channel-wise feature responses. The original inception block is modified by reducing the size of the spatial filter and introducing dilated convolution to obtain a larger receptive field. The proposed network was validated on three challenging medical image segmentation tasks with varying size ROIs: lung segmentation on chest X-ray (CXR) images, skin lesion segmentation on dermoscopy images and nucleus segmentation on microscopy cell images. Improved performance compared to state-of-the-art techniques demonstrates the effectiveness and generalisability of the proposed Dilated Convolution and Inception blocks-based U-Net (DCI-UNet).

Generalization Capabilities of Neural Cellular Automata for Medical Image Segmentation: A Robust and Lightweight Approach

Edge Segmentation: Empowering Mobile Telemedicine with Compressed Cellular Neural Networks

Med-NCA: Robust and Lightweight Segmentation with Neural Cellular Automata

MS-NAS: Multi-scale Neural Architecture Search for Medical Image Segmentation

Unsupervised Training of Neural Cellular Automata on Edge Devices

M3D-NCA: Robust 3D Segmentation with Built-in Quality Control

Scalable model for segmenting Cells' Nuclei using the U-NET architecture

3D U$$^$$-Net: A 3D Universal U-Net for Multi-domain Medical Image Segmentation

Medical Image Segmentation Review: The success of U-Net

Attention Based Multi-Scale Nested Network for Biomedical Image Segmentation

EdgeMedNet: Lightweight and Accurate U-Net for Implementing Efficient Medical Image Segmentation on Edge Devices

DC-UNet: Rethinking the U-Net Architecture with Dual Channel Efficient CNN for Medical Images Segmentation

NCA-Morph: Medical Image Registration with Neural Cellular Automata

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

NCAdapt: Dynamic adaptation with domain-specific Neural Cellular Automata for continual hippocampus segmentation

Towards Generalizable Medical Image Segmentation with Pixel-wise Uncertainty Estimation

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

A low resource 3D U-Net based deep learning model for medical image analysis

Multimodal Biomedical Image Segmentation using Multi-Dimensional U-Convolutional Neural Network

DCSAU-Net: A deeper and more compact split-attention U-Net for medical image segmentation

Modality specific U-Net variants for biomedical image segmentation: a survey