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

Read full text
Authors: David Paul Yallup, Namu Kroupa, Will J. Handley
Cited by: 0

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.

Fields of Study & Subjects

Markov Chains and Monte Carlo Methods Gaussian Processes and Bayesian Inference Statistical Methods and Bayesian Inference

Publication details

Journal / source
arXiv (Cornell University)
Publication date
30 January 2026
Document type
Preprint