SECRET: Stochasticity Emulator for Cosmic Ray Electrons

TL;DR

This paper introduces SECRET, a machine learning-based emulator that efficiently models the probabilistic distribution of cosmic-ray electron spectra, enabling faster parameter studies and statistical analysis of cosmic-ray data.

Contribution

We develop SECRET, a novel machine learning emulator using MADE to represent the joint probability distribution of cosmic-ray electron fluxes, allowing interpolation over transport parameters.

Findings

Accelerates cosmic-ray electron spectrum simulations.

Enables interpolation over transport parameters.

Provides a probabilistic framework for cosmic-ray flux analysis.

Abstract

The spectrum of cosmic-ray electrons depends sensitively on the history and spatial distribution of nearby sources. Given our limited observational handle on cosmic-ray sources, any model remains necessarily probabilistic. Previously, predictions were performed in a Monte Carlo fashion, summing the contributions from individual, simulated sources to generate samples from the statistical ensemble of possible electron spectra. Such simulations need to be re-run if the cosmic-ray transport parameters (e.g. diffusion coefficient, maximum energy) are changed, rendering any parameter study computationally expensive. In addition, a proper statistical analysis of observations and comparison with such probabilistic models requires the joint probability distribution of the full spectrum instead of only samples. Note that parametrising this joint distribution is rendered difficult by the non-Gaussian statistics of the cosmic-ray fluxes. Here, we employ machine learning to compute the joint probability distribution of cosmic-ray electron fluxes. Specifically, we employ masked autoregressive density estimation (MADE) for a representation of the high-dimensional joint probability distribution. In a first step, we train the network on a Monte Carlo simulation for a fixed set of transport parameters, thus significantly accelerating the generation of samples. In a second step, we extend this setup to SECRET (Stochasticity Emulator for Cosmic Ray Electrons), allowing to reliably interpolate over the space of transport parameters. We make the MADE and SECRET codes available at https://git.rwth-aachen.de/pmertsch/secret .