提出可微分的连续图像简点检测,实现拓扑保持的端到端分割。
Continuous-tone Simple Points: An $\ell_0$-Norm of Cyclic Gradient for Topology-Preserving Data-Driven Image Segmentation

- 基于循环梯度的ℓ₀范数,首次在连续图像上定义可微简点
- 在多个数据集上提升分割拓扑正确率,骨架结构更准确
- 适用于任意带softmax/sigmoid输出的分割网络,易集成
拓扑特征在图像分割、骨架提取等任务中对几何合理性与结构一致性至关重要。然而,现有简点检测方法仅限于二值图像且不可微,难以融入现代深度学习的梯度优化。同时,形态学与纯数据驱动方法常无法保证拓扑一致性。为此,本文提出一种新方法,直接在连续值图像上计算简点,实现可微拓扑推理。基于此,我们设计了一种高效骨架提取算法,可保持二值与连续图像的拓扑结构。进一步,构建了一个变分模型,通过保留拓扑不可移除(即非简点)来施加拓扑约束,可无缝集成至任何具有softmax或sigmoid输出的深度神经网络分割模型中。实验表明,该方法在多个基准测试中显著提升拓扑完整性与结构准确性。代码已开源:https://github.com/levnsio/CSP。
原文摘要 · Abstract (English)
Topological features play an essential role in ensuring geometric plausibility and structural consistency in image analysis tasks such as segmentation and skeletonization. However, integrating topology-preserving learning based on simple points into deep learning tasks remains challenging, as existing simple point detection methods are confined to binary images and are non-differentiable, rendering them incompatible with gradient-based optimization in modern deep learning. Moreover, morphological and purely data-driven approaches often fail to guaranty topological consistency. To address these limitations, we propose a novel method that directly computes simple points on continuous-valued images, enabling differentiable topological inference. Building on this theory, we develop an efficient skeleton extraction algorithm that preserves topological structures in binary and continuous-valued images. Furthermore, we design a variational model that enforces topological constraints by preserving topologically non-removable (i.e., non-simple) points, which can be seamlessly integrated into any deep neural network segmentation with softmax or sigmoid outputs. Experimental results demonstrate that the proposed approach effectively improves topological integrity and structural accuracy across multiple benchmarks. The codes are available in https://github.com/levnsio/CSP.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。