Preserving instance continuity and length in segmentation through connectivity-aware loss computation
- URL: http://arxiv.org/abs/2509.03154v1
- Date: Wed, 03 Sep 2025 09:08:47 GMT
- Title: Preserving instance continuity and length in segmentation through connectivity-aware loss computation
- Authors: Karol Szustakowski, Luk Frank, Julia Esser, Jan GrĂ¼ndemann, Marie Piraud,
- Abstract summary: We propose two novel loss functions, Negative Centerline Loss and Simplified Topology Loss, that help preserve connectivity of output instances.<n>We evaluate our approach on a 3D light-sheet fluorescence microscopy dataset of axon initial segments (AIS), a task prone to discontinuity due to signal dropout.
- Score: 0.29656637520758655
- License: http://arxiv.org/licenses/nonexclusive-distrib/1.0/
- Abstract: In many biomedical segmentation tasks, the preservation of elongated structure continuity and length is more important than voxel-wise accuracy. We propose two novel loss functions, Negative Centerline Loss and Simplified Topology Loss, that, applied to Convolutional Neural Networks (CNNs), help preserve connectivity of output instances. Moreover, we discuss characteristics of experiment design, such as downscaling and spacing correction, that help obtain continuous segmentation masks. We evaluate our approach on a 3D light-sheet fluorescence microscopy dataset of axon initial segments (AIS), a task prone to discontinuity due to signal dropout. Compared to standard CNNs and existing topology-aware losses, our methods reduce the number of segmentation discontinuities per instance, particularly in regions with missing input signal, resulting in improved instance length calculation in downstream applications. Our findings demonstrate that structural priors embedded in the loss design can significantly enhance the reliability of segmentation for biological applications.
Related papers
- SDF-TopoNet: A Two-Stage Framework for Tubular Structure Segmentation via SDF Pre-training and Topology-Aware Fine-Tuning [2.3436632098950456]
Key challenge is ensuring topological correctness while maintaining computational efficiency.<n>We propose textbfSDF-TopoNet, an improved topology-aware segmentation framework.<n>We show that SDF-TopoNet outperforms existing methods in both topological accuracy and quantitative segmentation metrics.
arXiv Detail & Related papers (2025-03-14T23:54:38Z) - Efficient Connectivity-Preserving Instance Segmentation with Supervoxel-Based Loss Function [2.501693072047969]
We propose a topology-aware neural network segmentation method with minimal computational overhead.<n>We demonstrate its effectiveness on a new public dataset of 3-d light microscopy images of mouse brains.
arXiv Detail & Related papers (2025-01-02T02:49:13Z) - Skeleton Recall Loss for Connectivity Conserving and Resource Efficient Segmentation of Thin Tubular Structures [2.0710692923459804]
Accurately segmenting thin tubular structures, such as vessels, nerves, roads or concrete cracks, is a crucial task in computer vision.
Standard deep learning-based segmentation loss functions, such as Dice or Cross-Entropy, focus on overlap, often at the expense of preserving structural connectivity or topology.
We propose a novel Skeleton Recall Loss, which effectively addresses these challenges by circumventing intensive GPU-based calculations with inexpensive CPU operations.
arXiv Detail & Related papers (2024-04-03T18:42:19Z) - Leveraging Frequency Domain Learning in 3D Vessel Segmentation [50.54833091336862]
In this study, we leverage Fourier domain learning as a substitute for multi-scale convolutional kernels in 3D hierarchical segmentation models.
We show that our novel network achieves remarkable dice performance (84.37% on ASACA500 and 80.32% on ImageCAS) in tubular vessel segmentation tasks.
arXiv Detail & Related papers (2024-01-11T19:07:58Z) - Fuzzy Attention Neural Network to Tackle Discontinuity in Airway
Segmentation [67.19443246236048]
Airway segmentation is crucial for the examination, diagnosis, and prognosis of lung diseases.
Some small-sized airway branches (e.g., bronchus and terminaloles) significantly aggravate the difficulty of automatic segmentation.
This paper presents an efficient method for airway segmentation, comprising a novel fuzzy attention neural network and a comprehensive loss function.
arXiv Detail & Related papers (2022-09-05T16:38:13Z) - InverseForm: A Loss Function for Structured Boundary-Aware Segmentation [80.39674800972182]
We present a novel boundary-aware loss term for semantic segmentation using an inverse-transformation network.
This plug-in loss term complements the cross-entropy loss in capturing boundary transformations.
We analyze the quantitative and qualitative effects of our loss function on three indoor and outdoor segmentation benchmarks.
arXiv Detail & Related papers (2021-04-06T18:52:45Z) - Topological obstructions in neural networks learning [67.8848058842671]
We study global properties of the loss gradient function flow.
We use topological data analysis of the loss function and its Morse complex to relate local behavior along gradient trajectories with global properties of the loss surface.
arXiv Detail & Related papers (2020-12-31T18:53:25Z) - Spectral Analysis for Semantic Segmentation with Applications on Feature
Truncation and Weak Annotation [19.967870811543737]
A striking balance between the accuracy and the training cost of the SSNNs such as U-Net exists.
We propose a spectral analysis to investigate the correlations among the resolution of the down sampled grid, the loss function and the accuracy of the SSNNs.
arXiv Detail & Related papers (2020-12-28T07:18:25Z) - An Uncertainty-Driven GCN Refinement Strategy for Organ Segmentation [53.425900196763756]
We propose a segmentation refinement method based on uncertainty analysis and graph convolutional networks.
We employ the uncertainty levels of the convolutional network in a particular input volume to formulate a semi-supervised graph learning problem.
We show that our method outperforms the state-of-the-art CRF refinement method by improving the dice score by 1% for the pancreas and 2% for spleen.
arXiv Detail & Related papers (2020-12-06T18:55:07Z) - A persistent homology-based topological loss function for multi-class
CNN segmentation of cardiac MRI [7.993897173085253]
We build a richer description of segmentation topology by considering all possible labels and label pairs.
These topological priors allow us to resolve all topological errors in a subset of 150 examples from the ACDC short axis CMR training data set.
arXiv Detail & Related papers (2020-08-21T17:09:13Z) - clDice -- A Novel Topology-Preserving Loss Function for Tubular
Structure Segmentation [57.20783326661043]
We introduce a novel similarity measure termed centerlineDice (short clDice)
We theoretically prove that clDice guarantees topology preservation up to homotopy equivalence for binary 2D and 3D segmentation.
We benchmark the soft-clDice loss on five public datasets, including vessels, roads and neurons (2D and 3D)
arXiv Detail & Related papers (2020-03-16T16:27:49Z)
This list is automatically generated from the titles and abstracts of the papers in this site.
This site does not guarantee the quality of this site (including all information) and is not responsible for any consequences.