arXiv:2506.02312eess.IVcs.CV2025-06被引 2

提出DEFFA-Unet,提升视网膜血管分割精度与泛化能力

Dual encoding feature filtering generalized attention UNET for retinal vessel segmentation

  • 双编码器处理域不变输入,增强特征提取
  • 跨数据集验证中,准确率超现有方法1.8%以上
  • 适合医学图像分割、尤其对误报敏感场景

视网膜血管分割对眼病和心血管疾病诊断至关重要。尽管2015年Olaf Ronneberger提出的U-Net显著推动了该领域发展,但训练数据有限、分布不均及特征提取不足等问题仍限制分割性能与模型泛化能力。为此,本文提出DEFFA-Unet,引入额外编码器处理域不变预处理输入,提升特征表达与泛化能力;设计特征过滤融合模块,实现精准特征筛选与鲁棒融合;针对高精度需求场景(假阳性代价高),以注意力引导的特征重构融合模块替代传统跳跃连接;同时提出创新的数据增强与平衡策略,缓解数据稀缺与分布偏移问题。在四个基准数据集(DRIVE、CHASEDB1、STARE、HRF)及一个SLO数据集(IOSTAR)上的全面评估显示,所提方法在多项指标上优于基线与主流模型,尤其在跨验证泛化性能上表现突出。

原文摘要 · Abstract (English)

Retinal blood vessel segmentation is crucial for diagnosing ocular and cardiovascular diseases. Although the introduction of U-Net in 2015 by Olaf Ronneberger significantly advanced this field, yet issues like limited training data, imbalance data distribution, and inadequate feature extraction persist, hindering both the segmentation performance and optimal model generalization. Addressing these critical issues, the DEFFA-Unet is proposed featuring an additional encoder to process domain-invariant pre-processed inputs, thereby improving both richer feature encoding and enhanced model generalization. A feature filtering fusion module is developed to ensure the precise feature filtering and robust hybrid feature fusion. In response to the task-specific need for higher precision where false positives are very costly, traditional skip connections are replaced with the attention-guided feature reconstructing fusion module. Additionally, innovative data augmentation and balancing methods are proposed to counter data scarcity and distribution imbalance, further boosting the robustness and generalization of the model. With a comprehensive suite of evaluation metrics, extensive validations on four benchmark datasets (DRIVE, CHASEDB1, STARE, and HRF) and an SLO dataset (IOSTAR), demonstrate the proposed method's superiority over both baseline and state-of-the-art models. Particularly the proposed method significantly outperforms the compared methods in cross-validation model generalization.

血管分割医学图像U-Net改进注意力机制

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。