Hierarchical Bayesian inference with compositional score modeling for stellar streams
Giuseppe Viterbo, Jonas Arruda, Tobias Buck
astro-ph.GA
Submitted: 2026-07-22
Code: https://github.com/vepe99/diffusion-experiments
License: http://creativecommons.org/licenses/by/4.0/
The gist: Context: Stellar streams trace the gravitational potential of the Milky Way over a wide range of Galactocentric radii.
Terminology
Abstract
Context: Stellar streams trace the gravitational potential of the Milky Way over a wide range of Galactocentric radii. Since different streams sample different regions of the Galaxy, combining several of them can constrain the global mass distribution more tightly than modeling any single stream in isolation. Most of the existing multi-stream analyses rely on likelihood-based methods that require a new inference run whenever additional streams or kinematic measurements become available. Aims: We aim to infer the Milky Way potential from multiple stellar streams combined with an additional constraint through the Galactic circular velocity curve within a single hierarchical framework. Methods: We model the problem hierarchically, separating parameters that are common to all streams from parameters that are specific to each progenitor. We train score-based neural posterior estimators on a library of simulated streams and combine information from different streams through compositional score modeling as a post-training step. Results: Tests on independent simulations show that the inferred posteriors are well calibrated and accurate. Combining several streams reduces the uncertainties on the global potential parameters relative to single-stream analyses. Applied to Gaia data, the model favours a mildly oblate dark matter halo with axis ratio q NFW = 0.77, scale radius a NFW = 9.3 kpc, a disk mass of 4.3 times 10 10 M, and a local dark matter density rho NFW, = 0.01153 M pc-3 consistent with recent stream-based studies. Conclusions: This hierarchical framework provides a practical way to combine information from multiple stellar streams without repeating the full inference procedure for each new dataset. The method is adaptable to new datasets, like future Gaia DR4, or new spectroscopic surveys, with minimal computational cost.
Sources
- Diffusion Models in Simulation-Based Inference: A Tutorial Review
- Overcoming Selection Bias in Statistical Studies With Amortized Bayesian Inference
- FlowSN: Neural Simulation-Based Inference under Realistic Selection Effects applied to Supernova Cosmology
- Simulation-Based Inference: A Practical Guide
- Compositional Score Modeling for Simulation-based Inference
- A COMPASS to Model Comparison and Simulation-Based Inference in Galactic Chemical Evolution
- A New Catalog of Globular Clusters in the Milky Way
- Denoising Diffusion Probabilistic Models
- Gotta Go Fast When Generating Data with Score-Based Models
- Elucidating the Design Space of Diffusion-Based Generative Models
- Amortized Simulation-Based Inference of Colliding-Wind Binaries from Short, Noisy Image Time Series
- BayesFlow 2: Multi-Backend Amortized Bayesian Inference in Python
- Set Transformer: A Framework for Attention-based Permutation-Invariant Neural Networks
- Amortized Bayesian Workflow
- Decoupled Weight Decay Regularization
- Progressive Distillation for Fast Sampling of Diffusion Models
- Transfer learning for multifidelity simulation-based inference in cosmology
- Detecting Model Misspecification in Amortized Bayesian Inference with Neural Networks
- Validating Bayesian Inference Algorithms with Simulation-Based Calibration
- CASBI -- Chemical Abundance Simulation-Based Inference for Galactic Archeology
Related papers
- Apparent Stability in Self-Gravitating Turbulence and the Evolution of Molecular Clouds
- Two sets of potential-density basis pairs for the study of radial perturbations in collisionless spherical stellar systems
- Constraining reionization-era Ly alpha escape with JELS-MUSE: a highly complete H alpha-selected sample at z about6.1
- Deriving volume density profiles of filaments from observed surface densities
- Little Red Dots and Supermassive Black Hole Seed Formation in Ultralight Dark Matter Halos
- MEGATRON: how the first stars can create an iron metallicity plateau in the smallest dwarf galaxies