A graph-based Neural Network surrogate model for accelerating semi-analytical model of galaxy formation and evolution

TL;DR

This paper introduces a graph neural network surrogate model that efficiently predicts galaxy properties from dark matter merger trees, significantly accelerating semi-analytic galaxy formation simulations.

Contribution

A novel graph-based neural network model that accurately predicts galaxy properties across cosmic time, enabling faster exploration of galaxy formation models.

Findings

Predicts stellar mass with R^2 of 0.946-0.973 across redshifts 0-3.

Achieves a scatter of 0.19-0.28 dex in stellar mass predictions.

Reproduces galaxy properties accurately over multiple merger trees and redshifts.

Abstract

Understanding how galaxy populations emerge and evolve from the growth of dark matter structure is a central challenge in galaxy formation theory. Semi-analytic models (SAMs) provide an efficient framework to address this problem, but exploring large ensembles of merger trees across broad parameter spaces remains computationally demanding. We develop a conditional graph neural network surrogate model that combines merger tree information with SAM parameters to predict galaxy properties across cosmic time. Using merger trees of dark matter halos from the Uchuu simulation and the Galacticus SAM, the model predicts stellar mass, luminosity, angular momentum, gas metal mass, and specific star formation rate across the wide redshift range of 0 <= z <= 5. For instance, the model can predict stellar mass at 0 <= z <= 3 with a scatter of 0.19-0.28 dex and coefficient of determination R^2 of 0.946-0.973 (R^2 close to 1 indicates prediction closely matching the truth). The results show that a single graph based model can reproduce these galaxy properties with good accuracy over multiple SAM realizations, merger trees and redshifts. This catalog-level model provides a practical route for accelerating SAM based studies of galaxy formation to enable a more detailed investigation of the model parameter space. The inference code, trained models, and example data products are publicly available at https://github.com/MutongCat/sam2galaxy-gnn.