Wasserstein Auto-Encoders of Merge Trees (and Persistence Diagrams) - Archive ouverte HAL
Article Dans Une Revue IEEE Transactions on Visualization and Computer Graphics Année : 2024

Wasserstein Auto-Encoders of Merge Trees (and Persistence Diagrams)

Résumé

This paper presents a computational framework for the Wasserstein auto-encoding of merge trees (MT-WAE), a novel extension of the classical auto-encoder neural network architecture to the Wasserstein metric space of merge trees. In contrast to traditional auto-encoders which operate on vectorized data, our formulation explicitly manipulates merge trees on their associated metric space at each layer of the network, resulting in superior accuracy and interpretability. Our novel neural network approach can be interpreted as a non-linear generalization of previous linear attempts [72] at merge tree encoding. It also trivially extends to persistence diagrams. Extensive experiments on public ensembles demonstrate the efficiency of our algorithms, with MT-WAE computations in the orders of minutes on average. We show the utility of our contributions in two applications adapted from previous work on merge tree encoding [72]. First, we apply MT-WAE to merge tree compression, by concisely representing them with their coordinates in the final layer of our auto-encoder. Second, we document an application to dimensionality reduction, by exploiting the latent space of our auto-encoder, for the visual analysis of ensemble data. We illustrate the versatility of our framework by introducing two penalty terms, to help preserve in the latent space both the Wasserstein distances between merge trees, as well as their clusters. In both applications, quantitative experiments assess the relevance of our framework. Finally, we provide a C++ implementation that can be used for reproducibility.
Fichier principal
Vignette du fichier
2307.02509 (4).pdf (25.72 Mo) Télécharger le fichier
Origine Fichiers produits par l'(les) auteur(s)

Dates et versions

hal-04552330 , version 1 (19-04-2024)

Identifiants

Citer

Mathieu Pont, Julien Tierny. Wasserstein Auto-Encoders of Merge Trees (and Persistence Diagrams). IEEE Transactions on Visualization and Computer Graphics, In press, pp.1-16. ⟨10.1109/TVCG.2023.3334755⟩. ⟨hal-04552330⟩
222 Consultations
23 Téléchargements

Altmetric

Partager

More