用SDF预训练+拓扑感知微调,提升管状结构分割精度与效率
SDF-TopoNet: A Two-Stage Framework for Tubular Structure Segmentation via SDF Pre-training and Topology-Aware Fine-Tuning
- 分两阶段训练:先用SDF编码拓扑信息,再动态调整拓扑损失
- 在5个数据集上均优于现有方法,拓扑准确率和分割指标双提升
- 适合医学影像、神经网络等需保持连通性与连续性的场景
准确分割血管、神经元、道路网络等管状与曲线结构在诸多应用中至关重要。关键挑战在于保证拓扑正确性的同时兼顾计算效率。现有方法常依赖基于持久同调的拓扑损失(如贝蒂误差),但计算开销大且对像素级精度不敏感,通常还需额外使用Dice或MSE损失补偿。为此,我们提出SDF-TopoNet,一种改进的拓扑感知分割框架,显著提升分割精度与训练效率。该方法采用新颖的两阶段训练策略:预训练阶段利用符号距离函数(SDF)作为辅助目标,使模型在不直接使用高成本拓扑损失的情况下编码拓扑信息;微调阶段引入动态适配器与优化后的拓扑损失,确保拓扑正确性,同时缓解过拟合并降低计算负担。我们在五个基准数据集上进行评估,结果表明SDF-TopoNet在拓扑准确率和定量分割指标上均优于现有方法,且大幅降低训练复杂度。
原文摘要 · Abstract (English)
Accurate segmentation of tubular and curvilinear structures, such as blood vessels, neurons, and road networks, is crucial in various applications. A key challenge is ensuring topological correctness while maintaining computational efficiency. Existing approaches often employ topological loss functions based on persistent homology, such as Betti error, to enforce structural consistency. However, these methods suffer from high computational costs and are insensitive to pixel-level accuracy, often requiring additional loss terms like Dice or MSE to compensate. To address these limitations, we propose \textbf{SDF-TopoNet}, an improved topology-aware segmentation framework that enhances both segmentation accuracy and training efficiency. Our approach introduces a novel two-stage training strategy. In the pre-training phase, we utilize the signed distance function (SDF) as an auxiliary learning target, allowing the model to encode topological information without directly relying on computationally expensive topological loss functions. In the fine-tuning phase, we incorporate a dynamic adapter alongside a refined topological loss to ensure topological correctness while mitigating overfitting and computational overhead. We evaluate our method on five benchmark datasets. Experimental results demonstrate that SDF-TopoNet outperforms existing methods in both topological accuracy and quantitative segmentation metrics, while significantly reducing training complexity.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。