Abstract:Segmentation networks are not explicitly imposed to learn global invariants of an image, such as the shape of an object and the geometry between multiple objects, when they are trained with a standard loss function. On the other hand, incorporating such invariants into network training may help improve performance for various segmentation tasks when they are the intrinsic characteristics of the objects to be segmented. One example is segmentation of aorta and great vessels in computed tomography (CT) images where vessels are found in a particular geometry in the body due to the human anatomy and they mostly seem as round objects on a 2D CT image. This paper addresses this issue by introducing a new topology-aware loss function that penalizes topology dissimilarities between the ground truth and prediction through persistent homology. Different from the previously suggested segmentation network designs, which apply the threshold filtration on a likelihood function of the prediction map and the Betti numbers of the ground truth, this paper proposes to apply the Vietoris-Rips filtration to obtain persistence diagrams of both ground truth and prediction maps and calculate the dissimilarity with the Wasserstein distance between the corresponding persistence diagrams. The use of this filtration has advantage of modeling shape and geometry at the same time, which may not happen when the threshold filtration is applied. Our experiments on 4327 CT images of 24 subjects reveal that the proposed topology-aware loss function leads to better results than its counterparts, indicating the effectiveness of this use.

Loss Tomography from Tree Topologies to General Topologies

Fast Parameter Estimation in Loss Tomography for Networks of General Topology

Inference of Link Loss Rates by Explicit Estimation

Simultaneous Topology and Loss Tomography Via a Modified Theme Dictionary Model

A bottom-up inference of loss rate

A Fast Method To Estimate Loss Rate

Lossy link identification based on maximum pseudo likelihood estimation

Temporal Dependence Network Loss Tomography Using Maximum Pseudo Likelihood Method

Evaluating Loss Landscapes from a Topology Perspective

Adaptive Path Selection for Link Loss Inference in Network Tomography Applications

Improving Maximum-Likelihood-based Topology Inference by Sequentially Inserting Leaf Nodes

An Improved Topology Inference Algorithm Based on End-to-End Measurements

Topology-Aware Loss for Aorta and Great Vessel Segmentation in Computed Tomography Images

Improved Fast Topology Estimation Method Based on Maximum Likelihood

Loss-based prior for tree topologies in BART models

Reviews: Topological Distances and Losses for Brain Networks

Data aggregation based topology inference for wireless sensor networks

Optimal topological design for distributed estimation over sensor networks

A New Topology Optimization Methodology Based on Constraint Maximum-Weight Connected Graph Theorem

APPLICATION OF MINIMUM INFORMATION LOSS BASED STATE ESTIMATION TO TOPOLOGY ERROR IDENTIFICATION

Accurate and Effective Inference of Network Link Loss from Unicast End-to-end Measurements