HES-UNet: A U-Net for Hepatic Echinococcosis Lesion Segmentation
Jiayan Chen1
Kai Li2
Zhanjin Wang1
Zhan Wang1 3
Jianqiang Huang1
1Qinghai University
2Tsinghua University
3Affiliated Hospital of Qinghai University
[arXiv ๐Ÿ“]
[code โš™๏ธ]

HES-UNet is an efficient model for hepatic echinococcosis lesion segmentation, effectively aggregating multi-scale feature information and incorporating global attention modules to achieve precise segmentation of lesion regions.


Abstract

Hepatic echinococcosis (HE) is a prevalent disease in economically underdeveloped pastoral areas, where adequate medical resources are usually lacking. Existing methods often ignore multi-scale feature fusion or focus only on feature fusion between adjacent levels, which may lead to insufficient feature fusion. To address these issues, we propose an efficient and accurate model for HE lesion segmentation: A U-Net for Hepatic Echinococcosis Lesion Segmentation (HES-UNet). HES-UNet combines convolutional layers and attention modules to capture local and global features. During downsampling, the multi-directional downsampling block (MDB) is employed to integrate high-frequency and low-frequency features, effectively extracting image details. The multi-scale aggregation block (MAB) aggregates multi-scale feature information. In contrast, the multi-scale upsampling Block (MUB) learns highly abstract features and supplies this information to the skip connection module to fuse multi-scale features. Due to the distinct regional characteristics of HE, there is currently no publicly available high-quality dataset for training our model. We collected CT slice data from 268 patients at a certain hospital to train and evaluate the model. The experimental results show that HES-UNet achieves state-of-the-art performance on our dataset, achieving an overall Dice Similarity Coefficient (DSC) of 89.21%, which is 1.09% higher than that of TransUNet.

Overall pipeline of the model architecture of HES-UNet and its modules.

Results on HE Lesion Segmentation

Performance comparisons of our HES-UNet and other existing segmentation models. Bold indicates optimal performance.

Ablation study of the different modules in the HES-UNet. Bold indicates optimal performance.

HE Lesion Segmentation Demos

The following demos illustrate a comparison between HES-UNet and other existing segmentation models on our dataset. In the visualization, true positives (TP) are highlighted in green, false positives (FP) in blue, and false negatives (FN) in hotpink.

Cystic Echinococcosis Lesion Segmentation Sample

Alveolar Echinococcosis Lesion Segmentation Sample


Acknowledgements

Website template was borrowed from Colorful Image Colorization and Nerfies; the code can be found here and here. Thank you (.โ› แด— โ›.).