We gratefully acknowledge support from
the Simons Foundation and member institutions.
Full-text links:

Download:

Current browse context:

cs.CV

Change to browse by:

References & Citations

DBLP - CS Bibliography

Bookmark

(what is this?)
CiteULike logo BibSonomy logo Mendeley logo del.icio.us logo Digg logo Reddit logo

Computer Science > Computer Vision and Pattern Recognition

Title: clDice -- A Novel Topology-Preserving Loss Function for Tubular Structure Segmentation

Abstract: Accurate segmentation of tubular, network-like structures, such as vessels, neurons, or roads, is relevant to many fields of research. For such structures, the topology is their most important characteristic; particularly preserving connectedness: in the case of vascular networks, missing a connected vessel entirely alters the blood-flow dynamics. We introduce a novel similarity measure termed centerlineDice (short clDice), which is calculated on the intersection of the segmentation masks and their (morphological) skeleta. We theoretically prove that clDice guarantees topology preservation up to homotopy equivalence for binary 2D and 3D segmentation. Extending this, we propose a computationally efficient, differentiable loss function (soft-clDice) for training arbitrary neural segmentation networks. We benchmark the soft-clDice loss on five public datasets, including vessels, roads and neurons (2D and 3D). Training on soft-clDice leads to segmentation with more accurate connectivity information, higher graph similarity, and better volumetric scores.
Comments: * The authors Suprosanna Shit and Johannes C. Paetzold contributed equally to the work
Subjects: Computer Vision and Pattern Recognition (cs.CV); Machine Learning (cs.LG); Image and Video Processing (eess.IV)
DOI: 10.1109/CVPR46437.2021.01629
Report number: CVPR 2021
Cite as: arXiv:2003.07311 [cs.CV]
  (or arXiv:2003.07311v7 [cs.CV] for this version)

Submission history

From: Johannes C. Paetzold [view email]
[v1] Mon, 16 Mar 2020 16:27:49 GMT (7206kb,D)
[v2] Mon, 23 Mar 2020 20:45:16 GMT (7206kb,D)
[v3] Sun, 29 Mar 2020 22:46:43 GMT (7206kb,D)
[v4] Thu, 3 Dec 2020 19:53:43 GMT (7241kb,D)
[v5] Mon, 29 Mar 2021 13:36:28 GMT (14293kb,D)
[v6] Tue, 30 Mar 2021 11:51:21 GMT (14293kb,D)
[v7] Fri, 15 Jul 2022 10:39:38 GMT (7146kb,D)

Link back to: arXiv, form interface, contact.