On the mean-field limit for Stein variational gradient descent: stability and multilevel approximation

Fuente: arXiv
Salvato in:
Dettagli Bibliografici
Autori principali: Weissmann, Simon, Zech, Jakob
Natura: Preprint
Pubblicazione: 2024
Soggetti:
Accesso online:
Tags: Aggiungi Tag
Nessun Tag, puoi essere il primo ad aggiungerne!!
_version_ 1866913221195071488
author Weissmann, Simon
Zech, Jakob
author_facet Weissmann, Simon
Zech, Jakob
contents In this paper we propose and analyze a novel multilevel version of Stein variational gradient descent (SVGD). SVGD is a recent particle based variational inference method. For Bayesian inverse problems with computationally expensive likelihood evaluations, the method can become prohibitive as it requires to evolve a discrete dynamical system over many time steps, each of which requires likelihood evaluations at all particle locations. To address this, we introduce a multilevel variant that involves running several interacting particle dynamics in parallel corresponding to different approximation levels of the likelihood. By carefully tuning the number of particles at each level, we prove that a significant reduction in computational complexity can be achieved. As an application we provide a numerical experiment for a PDE driven inverse problem, which confirms the speed up suggested by our theoretical results.
format Preprint
id arxiv_https___arxiv_org_abs_2402_01320
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle On the mean-field limit for Stein variational gradient descent: stability and multilevel approximation
Weissmann, Simon
Zech, Jakob
Numerical Analysis
Methodology
In this paper we propose and analyze a novel multilevel version of Stein variational gradient descent (SVGD). SVGD is a recent particle based variational inference method. For Bayesian inverse problems with computationally expensive likelihood evaluations, the method can become prohibitive as it requires to evolve a discrete dynamical system over many time steps, each of which requires likelihood evaluations at all particle locations. To address this, we introduce a multilevel variant that involves running several interacting particle dynamics in parallel corresponding to different approximation levels of the likelihood. By carefully tuning the number of particles at each level, we prove that a significant reduction in computational complexity can be achieved. As an application we provide a numerical experiment for a PDE driven inverse problem, which confirms the speed up suggested by our theoretical results.
title On the mean-field limit for Stein variational gradient descent: stability and multilevel approximation
topic Numerical Analysis
Methodology
url https://arxiv.org/abs/2402.01320