Topology-Aware Segmentation Using Discrete Morse Theory

Abstract

In the segmentation of fine-scale structures from natural and biomedical images, per-pixel accuracy is not the only metric of concern. Topological correctness, such as vessel connectivity and membrane closure, is crucial for downstream analysis tasks. In this paper, we propose a new approach to train deep image segmentation networks for better topological accuracy. In particular, leveraging the power of discrete Morse theory (DMT), we identify global structures, including 1D skeletons and 2D patches, which are important for topological accuracy. Trained with a novel loss based on these global structures, the network performance is significantly improved especially near topologically challenging locations (such as weak spots of connections and membranes). On diverse datasets, our method achieves superior performance on both the DICE score and topological metrics.

19 pages, 11 figures

References in corpus (6)

• Rethinking Atrous Convolution for Semantic Image Segmentation
• A Topological Loss Function for Deep-Learning based Image Segmentation using Persistent Homology
• Sliced Wasserstein Kernel for Persistence Diagrams
• A Topological Regularizer for Classifiers via Persistent Homology
• Connectivity-Optimized Representation Learning via Persistent Homology
• Road Network Reconstruction from Satellite Images with Machine Learning Supported by Topological Methods

Cited by in corpus (3)

• Trigger Hunting with a Topological Prior for Trojan Detection
• TopoTxR: A Topological Biomarker for Predicting Treatment Response in Breast Cancer
• Quantifying Topology In Pancreatic Tubular Networks From Live Imaging 3D Microscopy