arXiv · 2502.20482
R-ParVI: Particle-based variational inference through lens of rewards
Abstract
A reward-guided, gradient-free ParVI method, \textit{R-ParVI}, is proposed for sampling partially known densities (e.g. up to a constant). R-ParVI formulates the sampling problem as particle flow driven by rewards: particles are drawn from a prior distribution, navigate through parameter space with movements determined by a reward mechanism blending assessments from the target density, with the steady state particle configuration approximating the target geometry. Particle-environment interactions are simulated by stochastic perturbations and the reward mechanism, which drive particles towards high density regions while maintaining diversity (e.g. preventing from collapsing into clusters). R-ParVI offers fast, flexible, scalable and stochastic sampling and inference for a class of probabilistic models such as those encountered in Bayesian inference and generative modelling.
Explore related subjects
Keep this discovery
Explore connections, maps & timelines
Yongchao Huang. 2025-02-27. R-ParVI: Particle-based variational inference through lens of rewards. https://arxiv.org/abs/2502.20482
Cite the original work for its findings. Save a collection to share your selection of sources.