Cosmology with Topological Deep Learning

Fuente: arXiv
Guardado en:
Detalles Bibliográficos
Autores principales: Lee, Jun-Young, Villaescusa-Navarro, Francisco
Formato: Preprint
Publicado: 2025
Materias:
Acceso en línea:
Etiquetas: Agregar Etiqueta
Sin Etiquetas, Sea el primero en etiquetar este registro!
_version_ 1866911092778729472
author Lee, Jun-Young
Villaescusa-Navarro, Francisco
author_facet Lee, Jun-Young
Villaescusa-Navarro, Francisco
contents The standard cosmological model with cold dark matter posits a hierarchical formation of structures. We introduce topological neural networks (TNNs), implemented as message-passing neural networks on higher-order structures, to effectively capture the topological information inherent in these hierarchies that traditional graph neural networks (GNNs) fail to account for. Our approach not only considers the vertices and edges that comprise a graph but also extends to higher-order cells such as tetrahedra, clusters, and hyperedges. This enables message-passing between these heterogeneous structures within a combinatorial complex. Furthermore, our TNNs are designed to conserve the $E(3)$-invariance, which refers to the symmetry arising from invariance against translations, reflections, and rotations. When applied to the Quijote suite, our TNNs achieve a significant reduction in the mean squared error. Compared to our GNNs, which lack higher-order message-passing, ClusterTNNs show improvements of up to 22% in $Ω_{\rm m}$ and 34% in $σ_8$ jointly, while the best FullTNN achieves an improvement of up to 60% in $σ_8$. In the context of the CAMELS suite, our models yield results comparable to the current GNN benchmark, albeit with a slight decrease in performance. We emphasize that our topology and symmetry-aware neural networks provide enhanced expressive power in modeling the large-scale structures of our universe.
format Preprint
id arxiv_https___arxiv_org_abs_2505_23904
institution arXiv
publishDate 2025
record_format arxiv
spellingShingle Cosmology with Topological Deep Learning
Lee, Jun-Young
Villaescusa-Navarro, Francisco
Cosmology and Nongalactic Astrophysics
The standard cosmological model with cold dark matter posits a hierarchical formation of structures. We introduce topological neural networks (TNNs), implemented as message-passing neural networks on higher-order structures, to effectively capture the topological information inherent in these hierarchies that traditional graph neural networks (GNNs) fail to account for. Our approach not only considers the vertices and edges that comprise a graph but also extends to higher-order cells such as tetrahedra, clusters, and hyperedges. This enables message-passing between these heterogeneous structures within a combinatorial complex. Furthermore, our TNNs are designed to conserve the $E(3)$-invariance, which refers to the symmetry arising from invariance against translations, reflections, and rotations. When applied to the Quijote suite, our TNNs achieve a significant reduction in the mean squared error. Compared to our GNNs, which lack higher-order message-passing, ClusterTNNs show improvements of up to 22% in $Ω_{\rm m}$ and 34% in $σ_8$ jointly, while the best FullTNN achieves an improvement of up to 60% in $σ_8$. In the context of the CAMELS suite, our models yield results comparable to the current GNN benchmark, albeit with a slight decrease in performance. We emphasize that our topology and symmetry-aware neural networks provide enhanced expressive power in modeling the large-scale structures of our universe.
title Cosmology with Topological Deep Learning
topic Cosmology and Nongalactic Astrophysics
url https://arxiv.org/abs/2505.23904