DU-Net:一种协同优化模型的小样本脑血管分割方法

DU-Net:A collaborative optimization model for small-sample cerebrovascular segmentation

  • 摘要: 脑血管分割是脑血管疾病筛选、病理发展追踪和精准手术治疗的重要基础. 受限于患者隐私保护和脑血管复杂的手工标注,现有的脑血管公开数据集样本少、人工标注信息不足,从而限制发展高性能的脑血管分割方法. 为解决上述问题,本文提出了一种粗–精分割的Double U-Net协同优化模型(DU-Net),旨在提高小样本相位对比磁共振血管成像(Phase contrast magnetic resonance angiography, PC-MRA)脑血管分割数据集下的性能表现. 首先,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 screening cerebrovascular diseases, monitoring pathological progression, and planning precise surgical interventions. Compared to other magnetic resonance angiography (MRA) modalities, phase-contrast MRA offers significant advantages due to its non-invasive nature, high spatial resolution, and ability to capture detailed flow information, making it particularly valuable for cerebrovascular analysis. However, due to concerns over patient privacy and the inherent complexity of cerebrovascular structures, publicly available cerebrovascular datasets are generally limited in terms of sample size and the extent of manual annotations, restricting the development of high-performance segmentation algorithms. To address these issues, this study proposes a coarse-to-fine segmentation Double U-Net collaborative optimization model (DU-Net), aiming to improve performance in small-sample cerebrovascular segmentation tasks. The DU-Net model consists of two paths, both built upon a three-dimensional U-Net backbone. The coarse segmentation path utilizes lightweight convolutions with fewer channels to minimize the collaborative training burden and achieve preliminary learning of cerebrovascular topology. The extracted features are subsequently transferred to the fine segmentation path, which enhances the ability of the model to generalize global features. In the fine segmentation path, convolutions with a higher number of channels enhance the ability of the model to capture detailed features, boundary information, and small targets. This improves segmentation accuracy and stability, particularly in refining local features and optimizing small target and boundary regions. Additionally, a Squeeze and Excitation (SE) module was incorporated into the skip connections of the fine segmentation path to assign higher learning weights to target areas, significantly enhancing the model representation ability of the fine segmentation path to learn and segment complex regions and detailed features. Based on publicly available resources, this study conducted comparative experiments with ten representative medical segmentation models. Experimental results revealed that the DU-Net achieved a precision of 78.29%, recall of 67.91%, Jaccard coefficient of 57.81%, and Dice coefficient of 72.59% on a small-sample cerebrovascular dataset PCA22, outperforming all baseline models across all metrics. Visualization results of diverse segmentation demonstrated that the DU-Net model offers significant advantages in maintaining vascular continuity, preserving branch details, and reconstructing complete vascular morphology, allowing for more accurate identification of vascular regions and reduction in artifacts and mis-segmentation. To verify the superior adaptability of the SE-Residual module and the effectiveness of the collaborative optimization framework, ablation experiments were conducted on PCA22. The results showed that the collaborative optimization mechanism enabled the fine segmentation path and significantly improved segmentation accuracy and model stability, particularly in handling complex, blurred boundaries and fine structures. Furthermore, double volumetric network (DV-Net) and Double Cross-scale residual network (DCSR-Net) models were constructed based on volumetric network (V-Net) and Cross-scale residual network (CSR-Net), respectively, to assess the performance gains of the collaborative optimization framework across different backbone networks. Results indicated that DU-Net consistently outperformed both DV-Net and DCSR-Net in terms of overall metrics, further underscoring its ease of deployment and strong adaptability to small-sample cerebrovascular datasets. Moreover, cross-dataset validation experiments showed that DU-Net, despite domain shifts in imaging modality and resolution, outperformed most comparative methods on the Time-of-flight MRA (TOF-MRA)-based subset of the public IXI dataset (IXI-Sub), validating its strong generalization capability. In conclusion, DU-Net exhibits enhanced feature learning and recognition capabilities, enabling accurate cerebrovascular segmentation on small-sample datasets and offering broad applicability across diverse scenarios.

     

/

返回文章
返回