Efficient Bayesian analysis of kilonovae and gamma ray burst afterglows with fiesta

Fuente: arXiv
Saved in:
Bibliographic Details
Main Authors: Koehn, Hauke, Wouters, Thibeau, Pang, Peter T. H., Bulla, Mattia, Rose, Henrik, Wichern, Hannah, Dietrich, Tim
Format: Preprint
Published: 2025
Subjects:
Online Access:
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866914253062012928
author Koehn, Hauke
Wouters, Thibeau
Pang, Peter T. H.
Bulla, Mattia
Rose, Henrik
Wichern, Hannah
Dietrich, Tim
author_facet Koehn, Hauke
Wouters, Thibeau
Pang, Peter T. H.
Bulla, Mattia
Rose, Henrik
Wichern, Hannah
Dietrich, Tim
contents Gamma-ray burst (GRB) afterglows and kilonovae (KNe) are electromagnetic transients that can accompany binary neutron star (BNS) mergers. Therefore, studying their emission processes is of general interest for constraining cosmological parameters or the behavior of ultra-dense matter. One common method to analyze electromagnetic data from BNS mergers is to sample a Bayesian posterior over the parameters of a physical model for the transient. However, sampling the posterior is computationally costly and because of the many likelihood evaluations required in this process, detailed models are too expensive to be used directly in Bayesian inference. In this paper, we address the problem by introducing fiesta, a python package to train machine learning (ML) surrogates for GRB afterglow and kilonova models that have the capacity to accelerate likelihood evaluations. Specifically, we introduce extensive ML surrogates for the state-of-the-art GRB afterglow models afterglowpy and pyblastafterglow, along with a new surrogate for KN emission based on the possis code. Our surrogates enable evaluation of the light-curve posterior within minutes. We also provide built-in posterior sampling capabilities in fiesta that rely on the flowMC package, which efficiently scale to higher dimensions when adding up to tens of nuisance sampling parameters. Because of its use of the JAX framework, fiesta also allows for GPU acceleration during both surrogate training and posterior sampling. We applied our framework to reanalyze AT2017gfo/GRB170817A and GRB211211A with our surrogates, thus employing the new pyblastafterglow model for the first time in Bayesian inference.
format Preprint
id arxiv_https___arxiv_org_abs_2507_13807
institution arXiv
publishDate 2025
record_format arxiv
spellingShingle Efficient Bayesian analysis of kilonovae and gamma ray burst afterglows with fiesta
Koehn, Hauke
Wouters, Thibeau
Pang, Peter T. H.
Bulla, Mattia
Rose, Henrik
Wichern, Hannah
Dietrich, Tim
High Energy Astrophysical Phenomena
Gamma-ray burst (GRB) afterglows and kilonovae (KNe) are electromagnetic transients that can accompany binary neutron star (BNS) mergers. Therefore, studying their emission processes is of general interest for constraining cosmological parameters or the behavior of ultra-dense matter. One common method to analyze electromagnetic data from BNS mergers is to sample a Bayesian posterior over the parameters of a physical model for the transient. However, sampling the posterior is computationally costly and because of the many likelihood evaluations required in this process, detailed models are too expensive to be used directly in Bayesian inference. In this paper, we address the problem by introducing fiesta, a python package to train machine learning (ML) surrogates for GRB afterglow and kilonova models that have the capacity to accelerate likelihood evaluations. Specifically, we introduce extensive ML surrogates for the state-of-the-art GRB afterglow models afterglowpy and pyblastafterglow, along with a new surrogate for KN emission based on the possis code. Our surrogates enable evaluation of the light-curve posterior within minutes. We also provide built-in posterior sampling capabilities in fiesta that rely on the flowMC package, which efficiently scale to higher dimensions when adding up to tens of nuisance sampling parameters. Because of its use of the JAX framework, fiesta also allows for GPU acceleration during both surrogate training and posterior sampling. We applied our framework to reanalyze AT2017gfo/GRB170817A and GRB211211A with our surrogates, thus employing the new pyblastafterglow model for the first time in Bayesian inference.
title Efficient Bayesian analysis of kilonovae and gamma ray burst afterglows with fiesta
topic High Energy Astrophysical Phenomena
url https://arxiv.org/abs/2507.13807