Search papers, labs, and topics across Lattice.
This paper introduces a novel, differentiable method for computing simple points on continuous-valued images by leveraging the $\ell_0$-norm of cyclic gradients, enabling topology-preserving learning in deep segmentation tasks. The approach facilitates efficient skeleton extraction and enforces topological constraints by preserving non-simple points within a variational model. Experiments demonstrate improved topological integrity and structural accuracy on various benchmarks when integrated with deep neural network segmentation.
Guaranteeing topological consistency in image segmentation is now possible within deep learning frameworks thanks to a novel differentiable simple point computation method applicable to continuous-valued images.
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.
