English

Nested Slice Sampling: Vectorized Nested Sampling for GPU-Accelerated Inference

Computation 2026-05-12 v2 Machine Learning Machine Learning

Abstract

Model comparison and calibrated uncertainty quantification often require integrating over parameters, but scalable inference can be challenging for complex, multimodal targets. Nested Sampling is a robust alternative to standard MCMC, yet its typically sequential structure and hard constraints make efficient accelerator implementations difficult. This paper introduces Nested Slice Sampling (NSS), a GPU-friendly, vectorized formulation of Nested Sampling that uses Hit-and-Run Slice Sampling for constrained updates. A tuning analysis yields a simple near-optimal rule for setting the slice width, improving high-dimensional behavior and making per-step compute more predictable for parallel execution. Experiments on challenging synthetic targets, high dimensional Bayesian inference, and Gaussian process hyperparameter marginalization show that NSS maintains accurate evidence estimates and high-quality posterior samples, and is particularly robust on difficult multimodal problems where current state-of-the-art methods such as tempered SMC baselines can struggle. An open-source implementation is released to facilitate adoption and reproducibility.

Keywords

Cite

@article{arxiv.2601.23252,
  title  = {Nested Slice Sampling: Vectorized Nested Sampling for GPU-Accelerated Inference},
  author = {David Yallup and Namu Kroupa and Will Handley},
  journal= {arXiv preprint arXiv:2601.23252},
  year   = {2026}
}

Comments

58 pages, 13 figures, Accepted to Transactions on Machine Learning Research