Estimating Dark Matter Halo Masses in Simulated Galaxy Clusters with Graph Neural Networks

Nikhil Garuda,John F. Wu,Dylan Nelson,Annalisa Pillepich
2024-11-20
Abstract:Galaxies grow and evolve in dark matter halos. Because dark matter is not visible, galaxies' halo masses ($\rm{M}_{\rm{halo}}$) must be inferred indirectly. We present a graph neural network (GNN) model for predicting $\rm{M}_{\rm{halo}}$ from stellar mass ($\rm{M}_{*}$) in simulated galaxy clusters using data from the IllustrisTNG simulation suite. Unlike traditional machine learning models like random forests, our GNN captures the information-rich substructure of galaxy clusters by using spatial and kinematic relationships between galaxy neighbour. A GNN model trained on the TNG-Cluster dataset and independently tested on the TNG300 simulation achieves superior predictive performance compared to other baseline models we tested. Future work will extend this approach to different simulations and real observational datasets to further validate the GNN model's ability to generalise.
Astrophysics of Galaxies,Cosmology and Nongalactic Astrophysics,Instrumentation and Methods for Astrophysics,Artificial Intelligence
What problem does this paper attempt to address?