TY - JOUR

T1 - Predictive uncertainty on astrophysics recovery from multifield cosmology

AU - Andrianomena, Sambatra

AU - Hassan, Sultan

N1 - Publisher Copyright:
© 2023 IOP Publishing Ltd and Sissa Medialab.

PY - 2023/6/1

Y1 - 2023/6/1

N2 - We investigate how the constraints on the density parameter (Ωm), the power spectrum amplitude (σ 8) and the supernova feedback parameters (A SN1 and A SN2) vary when exploiting information from multiple fields in cosmology. We make use of a convolutional neural network to retrieve the salient features from different combinations of field maps from IllustrisTNG in the CAMELS project. The fields considered are neutral hydrogen (HI), gas density (Mgas), magnetic fields (B) and gas metallicity (Z). We estimate the predictive uncertainty — sum of the squares of aleatoric and epistemic uncertainties — of the parameters inferred by our model by using Monte Carlo dropout, a Bayesian approximation. Results show that in general, the performance of the model improves as the number of channels of its input is increased. In the best setup which includes all fields (four channel input, Mgas-HI-B-Z) the model achieves R 2 > 0.96 on all parameters. Similarly, we find that the predictive uncertainty, which is dominated by the aleatoric uncertainty, decreases as more fields are used to train the model in general. The uncertainties obtained by dropout variational inference are overestimated on all parameters in our case, in that the predictive uncertainty is much larger than the actual squared error, which is the square of the difference between the ground truth and prediction. After calibration, which consists of a simple σ scaling method, the average deviation of the predictive uncertainty from the actual error goes down to 25% at most (on A SN1).

AB - We investigate how the constraints on the density parameter (Ωm), the power spectrum amplitude (σ 8) and the supernova feedback parameters (A SN1 and A SN2) vary when exploiting information from multiple fields in cosmology. We make use of a convolutional neural network to retrieve the salient features from different combinations of field maps from IllustrisTNG in the CAMELS project. The fields considered are neutral hydrogen (HI), gas density (Mgas), magnetic fields (B) and gas metallicity (Z). We estimate the predictive uncertainty — sum of the squares of aleatoric and epistemic uncertainties — of the parameters inferred by our model by using Monte Carlo dropout, a Bayesian approximation. Results show that in general, the performance of the model improves as the number of channels of its input is increased. In the best setup which includes all fields (four channel input, Mgas-HI-B-Z) the model achieves R 2 > 0.96 on all parameters. Similarly, we find that the predictive uncertainty, which is dominated by the aleatoric uncertainty, decreases as more fields are used to train the model in general. The uncertainties obtained by dropout variational inference are overestimated on all parameters in our case, in that the predictive uncertainty is much larger than the actual squared error, which is the square of the difference between the ground truth and prediction. After calibration, which consists of a simple σ scaling method, the average deviation of the predictive uncertainty from the actual error goes down to 25% at most (on A SN1).

KW - Bayesian reasoning

KW - cosmological simulations

KW - hydrodynamical simulations

KW - Machine learning

UR - http://www.scopus.com/inward/record.url?scp=85163816783&partnerID=8YFLogxK

UR - http://www.scopus.com/inward/citedby.url?scp=85163816783&partnerID=8YFLogxK

U2 - 10.1088/1475-7516/2023/06/051

DO - 10.1088/1475-7516/2023/06/051

M3 - Article

AN - SCOPUS:85163816783

SN - 1475-7516

VL - 2023

JO - Journal of Cosmology and Astroparticle Physics

JF - Journal of Cosmology and Astroparticle Physics

IS - 6

M1 - 051

ER -