PuSH - Publikationsserver des Helmholtz Zentrums München

Capturing shape information with multi-scale topological loss terms for 3D reconstruction.

Lect. Notes Comput. Sc. 13434 LNCS, 150-159 (2022)
Postprint DOI
Open Access Green
Reconstructing 3D objects from 2D images is both challenging for our brains and machine learning algorithms. To support this spatial reasoning task, contextual information about the overall shape of an object is critical. However, such information is not captured by established loss terms (e.g. Dice loss). We propose to complement geometrical shape information by including multi-scale topological features, such as connected components, cycles, and voids, in the reconstruction loss. Our method uses cubical complexes to calculate topological features of 3D volume data and employs an optimal transport distance to guide the reconstruction process. This topology-aware loss is fully differentiable, computationally efficient, and can be added to any neural network. We demonstrate the utility of our loss by incorporating it into SHAPR, a model for predicting the 3D cell shape of individual cells based on 2D microscopy images. Using a hybrid loss that leverages both geometrical and topological information of single objects to assess their shape, we find that topological information substantially improves the quality of reconstructions, thus highlighting its ability to extract more relevant features from image datasets.
Altmetric
Weitere Metriken?
Zusatzinfos bearbeiten [➜Einloggen]
Publikationstyp Artikel: Journalartikel
Dokumenttyp Wissenschaftlicher Artikel
Korrespondenzautor
Schlagwörter 3d Shape Prediction ; Cubical Complex ; Topological Loss
ISSN (print) / ISBN 0302-9743
e-ISSN 1611-3349
Quellenangaben Band: 13434 LNCS, Heft: , Seiten: 150-159 Artikelnummer: , Supplement: ,
Verlag Springer
Verlagsort Berlin [u.a.]
Nichtpatentliteratur Publikationen
Institut(e) Institute of AI for Health (AIH)
Helmholtz Pioneer Campus (HPC)