HaloFlow II: Robust Galaxy Halo Mass Inference with Domain Adaptation

TL;DR

HaloFlow II enhances galaxy halo mass inference by integrating domain adaptation techniques, significantly reducing bias and improving robustness across different galaxy formation simulations, thus enabling more reliable application to real observational data.

Contribution

The paper introduces HaloFlow$^{ m DA}$, which combines simulation-based inference with domain adaptation to improve generalization across different galaxy formation models.

Findings

MMD domain adaptation reduces bias by up to 57%.

HaloFlow$^{ m DA}$ achieves more robust halo mass estimates.

Improved calibration and reduced residuals across simulations.

Abstract

Precise halo mass ($M_h$) measurements are crucial for cosmology and galaxy formation. HaloFlow introduced a simulation-based inference (SBI) framework that uses state-of-the-art simulated galaxy images to precisely infer $M_h$. However, for HaloFlow to be applied to observations, it must be generalizable even when the underlying galaxy formation physics differ from those in the simulations on which it was trained. Without this generalization, HaloFlow produces biased and overconfident $M_h$ posteriors when applied to simulations with different physics. We introduce HaloFlow$^{\rm DA}$, an extension of HaloFlow that integrates domain adaptation (DA) with SBI to mitigate these cross-simulation shifts. Using synthetic galaxy images forward-modeled from the IllustrisTNG, EAGLE, and SIMBA simulations, we test two DA methods: Domain-Adversarial Neural Networks (DANN) and Maximum Mean Discrepancy (MMD). Incorporating DA significantly reduces bias and improves calibration, with MMD achieving the most stable performance, lowering the normalized residual metric, $\beta$, by an average of 31% and up to 57% when trained and tested on different simulations. Overall, HaloFlow$^{\rm DA}$ produces more robust, less biased with similar precision, $M_h$ constraints than the standard approach using the stellar-to-halo mass relation. HaloFlow$^{\rm DA}$ enables consistent, simulation-trained inference models to generalize across domains, establishing a foundation for robust $M_h$ inference from real HSC-SSP observations.