“智能健康与医疗”专辑+DU-Net:一种协同优化模型的小样本脑血管分割方法

DU-Net:A Collaborative Optimization Model for Small-Sample Cerebrovascular Segmentation

  • 摘要: 脑血管分割是脑血管疾病筛选、病理发展追踪和精准手术治疗的重要基础。受限于患者隐私保护和脑血管复杂的手工标注,现有的脑血管公开数据集样本少、人工标注信息不足,从而限制发展高性能的脑血管分割方法。为解决上述问题,本文提出了一种粗-精分割的Double U-Net协同优化模型(DU-Net),旨在提高小样本脑血管分割数据集下的性能表现。首先,DU-Net的两部分路径骨干由3D U-Net构成,其中粗分割路径采用低通道数量的轻量卷积核,从而减少协同训练负担,实现脑血管拓扑特征的初步学习。接着,将拓扑形态传递到精分割路径,从而为模型补偿泛化特征。此外,在精分割路径的跳跃连接部分引入了压缩激励(Squeeze and Excitation, SE)模块,为目标区域赋予更高的学习权重。实验结果显示,DU-Net在小样本脑血管数据集上的查准率、查全率、杰卡德系数和Dice系数分别是78.29%、67.91%、57.81%和72.59%,综合表现最优。结果表明,DU-Net具备更强的特征学习与识别能力,能够在小样本数据集上实现精确脑血管分割。

     

    Abstract: Cerebrovascular segmentation is a crucial foundation for cerebrovascular disease screening, pathological progression tracking, and precise surgical treatment. Owing to patient privacy protection concerns and complex structure, existing public cerebrovascular datasets have limited samples and insufficient manual annotation, which hinders the research and development of high-performance segmentation. To address the issues, this paper proposes a coarse-to-fine segmentation Double U-Net collaborative optimization model (DU-Net), aiming to improve performance on small-sample cerebrovascular segmentation. The DU-Net model consists of two paths, both based on 3D U-Net as the backbone. The coarse segmentation path utilizes lightweight convolutions with fewer channels to reduce the collaborative training burden and achieve preliminary learning of cerebrovascular topology. These features are then passed to the fine segmentation path to compensate for generalization features. In the fine segmentation path, convolutions with more channels contribute to enhancing the model ability to express detailed features, boundary information, and small targets. This improves segmentation accuracy and stability, particularly when refining local features and optimizing small target and boundary regions. In addition, a Squeeze and Excitation (SE) module is introduced in the skip connections of the fine segmentation path to assign higher learning weights to the target areas, significantly enhancing the model representation ability of fine segmentation path to learn and segment complex regions and detailed features. Based on existing public resources, this paper conducts comparative experiments with 10 typical medical segmentation models. Experimental results exhibit that the DU-Net achieves a precision of 78.29%, recall of 67.91%, Jaccard coefficient of 57.81%, and Dice coefficient of 72.59% on the small-sample cerebrovascular dataset, demonstrating superior performance across all metrics with the best overall performance. Visualization results of diverse segmentation show that the DU-Net model has significant advantages in vascular continuity, branch detail preservation, and complete reconstruction of vascular morphology, allowing for more accurate identification of vascular regions and reducing the occurrence of artifacts and mis-segmentation. To validate the effectiveness of the Squeeze and Excitation module and the collaborative optimization framework, an ablation study was conducted on the small-sample cerebrovascular dataset . Ablation experiments on the DU-Net modules demonstrate that the fine segmentation path using collaborative optimization effectively improves segmentation accuracy, particularly in handling complex, blurry boundaries and fine structures, thus significantly enhancing the overall precision and stability in medical image segmentation tasks. Furthermore, DV-Net and DCSR-Net were deployed based on V-Net and CSR-Net to validate diverse backbones. The results present that the DU-Net outperforms DV-Net and DCSR-Net, highlighting the ease of use and applicability of DU-Net on small-sample dataset. In conclusion, DU-Net possesses stronger feature learning and recognition capabilities, enabling accurate cerebrovascular segmentation on small-sample datasets and broad applications in multiple scenarios.

     

/

返回文章
返回