Cardiac Medical Image Segmentation Based on Parallel Fusion Network Architecture

Wenhao Zhang,Lin Chen,Jingkun Mao,Fengxi Li,Kun Liu
DOI: https://doi.org/10.1109/CCISP59915.2023.10355768
2023-11-17
Abstract:Cardiovascular disease is the leading cause of death worldwide. Accurate segmentation of cardiac images is able to assist physicians in quantifying and locating abnormal regions and cells in the heart, facilitating diagnosis and treatment and consequently reducing cardiovascular death rate. Semantic segmentation models have been widely used for this task in recent years. However, the task is challenging due to the complex anatomy of the heart and the presence of noise and variations in grayscale in cardiac medical images, which require high segmentation accuracy. To address this challenge, a feature-enhancement model based on Vision-Transformer-Unet (VIT-Unet) is proposed. Specifically, a parallel network that consists of Convolutional Neural Networks (CNNs) and Win-Transformer modules with windowed attention is constructed to extract comprehensive features. Then, a Fusion Convolution Transformer is added to enable the integration of global and local features from the parallel network. Finally, the Transformer module in the baseline model is replaced with our parallel fusion feature extraction network. The method proposed in the paper not only achieves higher segmentation accuracy but also presents more robust results compared to the baseline model, as evidenced by the experimental findings.
Computer Science,Medicine,Engineering
What problem does this paper attempt to address?