提出HES-UNet模型,提升肝包虫病病灶分割精度。
HES-UNet: A U-Net for Hepatic Echinococcosis Lesion Segmentation
- 融合多方向下采样与多尺度聚合,增强特征表达
- 在自建数据集上达89.21% Dice系数,优于TransUNet
- 适合医疗影像分割研究者与医学图像分析应用
肝包虫病(HE)在经济欠发达牧区高发,医疗资源匮乏。现有方法常忽略多尺度特征融合或仅关注相邻层间融合,导致特征利用不充分。为此,本文提出HES-UNet,一种高效精准的肝包虫病灶分割模型。该模型结合卷积层与注意力模块,捕获局部与全局特征。下采样阶段采用多方向下采样块(MDB)整合高频与低频特征,有效提取图像细节;多尺度聚合块(MAB)融合多尺度特征信息;上采样阶段的多尺度上采样块(MUB)学习高度抽象特征,并通过跳跃连接模块实现多尺度特征融合。由于肝包虫病区域特性显著,目前尚无公开高质量训练数据集。本研究收集268例患者CT切片数据用于模型训练与评估。实验结果表明,HES-UNet在自建数据集上达到89.21%的总体Dice相似系数(DSC),较TransUNet提升1.09%,性能处于领先水平。
原文摘要 · Abstract (English)
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 HES-UNet, an efficient and accurate model for HE lesion segmentation. This model 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. The project page is available at https://chenjiayan-qhu.github.io/HES-UNet-page.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。