Abstract:Medical image segmentation plays a crucial role in diagnosing and staging diseases. It facilitates image analysis and quantification in multiple applications, but building the right appropriate solutions is essential and highly reliant on the features of different datasets and computational resources. Most existing approaches provide segmentation for a specific anatomical region of interest and are limited to multiple imaging modalities in a clinical setting due to their generalizability with high computational requirements. To mitigate these issues, we propose a robust and lightweight deep learning real-time segmentation network for multi-modality medical images called MISegNet. We incorporate discrete wavelet transform (DWT) of the input to extract salient features in the frequency domain. This mechanism allows the neurons’ receptive field to enlarge within the network. We propose a self-attention-based global context-aware (SGCA) module with varying dilation rates to enlarge the field of view and designate the importance of each scale that enhances the network’s ability to discriminate features. We build a residual shuffle attention (RSA) mechanism to improve the feature representation of the proposed model and formulate a new boundary-aware loss function called Farid End Point Error (FEPE) that correctly segments regions with ambiguous boundaries for edge detection. We confirm the versatility of the proposed model by performing experiments against eleven state-of-the-art segmentation methods on four datasets of different organs, including two publicly available datasets (i.e., ISBI2017, and COVID-19 CT) and two private datasets (i.e., ovary and liver ultrasound images). Experimental results prove that the MISegNet with 1.5M parameters, outperforms the state-of-the-art methods by 1.5%–7% (i.e., dice coefficient score) with a corresponding 23× decrease in the number of parameters and multiply-accumulate operations respectively compared to U-Net.

A Medical Image Segmentation Methods Based On Wavelet Transforms And Som Neural Network

A memristive RBF neural network and its application in unsupervised medical image segmentation

Improved watershed transform for medical image segmentation using prior information

An Image Segmentation for Different Medical Image Modalities using Wavelet Transform Technique

Intelligent Diagnosis Method of MRI Brain Image Using Parallel Self-Organizing Feature Maps Neural Network

Local Feature Extraction and Information Bottleneck-Based Segmentation of Brain Magnetic Resonance (MR) Images.

WRANet: wavelet integrated residual attention U-Net network for medical image segmentation

Reliable segmentation of multiple lesions from medical images

An Improved Wavelet Neural Network Medical Image Segmentation Algorithm with Combined Maximum Entropy

Prior wavelet knowledge for multi-modal medical image segmentation using a lightweight neural network with attention guided features

Review of Medical Image Segmentation Methods

Medical image segmentation based on self-supervised hybrid fusion network

Multispectral MR Images Segmentation Using SOM Network

Brain image segmentation based on improved BP-Adaboost neural network

A New Hybrid Segmentation Method for MR Images

ω-net: Dual supervised medical image segmentation with multi-dimensional self-attention and diversely-connected multi-scale convolution

A lightweight multi-modality medical image semantic segmentation network base on the novel UNeXt and Wave-MLP

The Method of Multimodal MRI Brain Image Segmentation Based on Differential Geometric Features

Segmentation for Multimodal Brain Tumor Images Using Dual-Tree Complex Wavelet Transform and Deep Reinforcement Learning

Neural-network-based segmentation of multi-modal medical images: a comparative and prospective study

Deep Neural Network Approach Based Segmentation, Detection and Classification of Brain Tumor