Nested Slice Sampling: Vectorized Nested Sampling for GPU-Accelerated Inference
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.
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