Improving Generalization by Learning Geometry-Dependent and Physics-Based Reconstruction of Image Sequences | IEEE Journals & Magazine | IEEE Xplore

Improving Generalization by Learning Geometry-Dependent and Physics-Based Reconstruction of Image Sequences


Abstract:

Deep neural networks have shown promise in image reconstruction tasks, although often on the premise of large amounts of training data. In this paper, we present a new ap...Show More

Abstract:

Deep neural networks have shown promise in image reconstruction tasks, although often on the premise of large amounts of training data. In this paper, we present a new approach to exploit the geometry and physics underlying electrocardiographic imaging (ECGI) to learn efficiently with a relatively small dataset. We first introduce a non-Euclidean encoding-decoding network that allows us to describe the unknown and measurement variables over their respective geometrical domains. We then explicitly model the geometry-dependent physics in between the two domains via a bipartite graph over their graphical embeddings. We applied the resulting network to reconstruct electrical activity on the heart surface from body-surface potentials. In a series of generalization tasks with increasing difficulty, we demonstrated the improved ability of the network to generalize across geometrical changes underlying the data using less than 10% of training data and fewer variations of training geometry in comparison to its Euclidean alternatives. In both simulation and real-data experiments, we further demonstrated its ability to be quickly fine-tuned to new geometry using a modest amount of data.
Published in: IEEE Transactions on Medical Imaging ( Volume: 42, Issue: 2, February 2023)
Page(s): 403 - 415
Date of Publication: 28 October 2022

ISSN Information:

PubMed ID: 36306312

Funding Agency:


Contact IEEE to Subscribe

References

References is not available for this document.