提出拓扑感知模块,让视网膜血管分割更连贯,适合临床分析。
TFFM: Topology-Aware Feature Fusion Module via Latent Graph Reasoning for Retinal Vessel Segmentation
- 将局部特征映射到潜在图空间,用图注意力捕捉全局结构依赖
- 在Fundus-AVSeg上达90.97%的联合Dice,Hausdorff距离仅3.50像素
- 显著减少血管断裂(降38%),结果可直接用于自动生物标志物量化
精确分割视网膜动脉和静脉对系统性心血管疾病的诊断具有重要意义。然而,标准卷积架构常产生拓扑不连贯的分割结果,存在缝隙与断点,尽管像素级精度高,但无法支持可靠的图结构临床分析。为此,我们提出一种拓扑感知框架,旨在保持血管连通性。该框架引入拓扑特征融合模块(TFFM),将局部特征表示映射至潜在图空间,利用图注意力网络捕获固定感受野易遗漏的全局结构依赖。同时,采用混合损失函数,结合Tversky损失处理类别不平衡,并使用软clDice损失显式惩罚拓扑断开。在Fundus-AVSeg数据集上的评估显示,本方法达到90.97%的联合Dice分数,95% Hausdorff距离为3.50像素。显著相比基线减少约38%的血管碎片化,生成拓扑一致的血管树,适用于自动化生物标志物量化。代码已开源:https://tffm-module.github.io/。
原文摘要 · Abstract (English)
Precise segmentation of retinal arteries and veins carries the diagnosis of systemic cardiovascular conditions. However, standard convolutional architectures often yield topologically disjointed segmentations, characterized by gaps and discontinuities that render reliable graph-based clinical analysis impossible despite high pixel-level accuracy. To address this, we introduce a topology-aware framework engineered to maintain vascular connectivity. Our architecture fuses a Topological Feature Fusion Module (TFFM) that maps local feature representations into a latent graph space, deploying Graph Attention Networks to capture global structural dependencies often missed by fixed receptive fields. Furthermore, we drive the learning process with a hybrid objective function, coupling Tversky loss for class imbalance with soft clDice loss to explicitly penalize topological disconnects. Evaluation on the Fundus-AVSeg dataset reveals state-of-the-art performance, achieving a combined Dice score of 90.97% and a 95% Hausdorff Distance of 3.50 pixels. Notably, our method decreases vessel fragmentation by approximately 38% relative to baselines, yielding topologically coherent vascular trees viable for automated biomarker quantification. We open-source our code at https://tffm-module.github.io/.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。