Mengke Li, Flora Wang, Jonathan Engel, Matthew Mumpowe, Rebecca Surman, Nicole Vassh
Accurately modeling $β$-decay half-lives of highly neutron-rich nuclei is a critical challenge for understanding $r$-process nucleosynthesis and the resulting kilonova light curves. We introduce a data-driven approach to model $β$ decay that uses Mixture Density Networks (MDNs) trained on the latest experimental measurements to extrapolate the half-lives of neutron rich nuclei. Unlike standard deterministic regression, the MDN directly parameterizes the probability distribution of the half-lives, providing intrinsic (aleatoric) uncertainty. We contrast the intrinsic uncertainty of a single model with the range of models produced by using random variations of the input training data samples. While all models show excellent agreement with experimental data, we find that varying the training data sets produces a wide range of extrapolations, particularly along data-poor regions such as the $N=126$ isotonic chain. We then propagate a selection of model results into $r$-process network calculations to evaluate their impact on isotopic abundances. Finally, by coupling the resulting isotopic abundances with thermalization efficiencies, we translate the effective nuclear heating rates into bolometric light curves. We compare the ranges of outcomes produced from the intrinsic uncertainty of a single model to those produced by multiple independent models trained on different training data sets.