Weak Generative Sampler to Efficiently Sample Invariant Distribution of Stochastic Differential Equation

Fuente: arXiv
Saved in:
Bibliographic Details
Main Authors: Cai, Zhiqiang, Cao, Yu, Huang, Yuanfei, Zhou, Xiang
Format: Preprint
Published: 2024
Subjects:
Online Access:
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866918045694296064
author Cai, Zhiqiang
Cao, Yu
Huang, Yuanfei
Zhou, Xiang
author_facet Cai, Zhiqiang
Cao, Yu
Huang, Yuanfei
Zhou, Xiang
contents Sampling invariant distributions from an Itô diffusion process presents a significant challenge in stochastic simulation. Traditional numerical solvers for stochastic differential equations require both a fine step size and a lengthy simulation period, resulting in biased and correlated samples. The current deep learning-based method solves the stationary Fokker--Planck equation to determine the invariant probability density function in the form of deep neural networks, but they generally do not directly address the problem of sampling from the computed density function. In this work, we introduce a framework that employs a weak generative sampler (WGS) to directly generate independent and identically distributed (iid) samples induced by a transformation map derived from the stationary Fokker--Planck equation. Our proposed loss function is based on the weak form of the Fokker--Planck equation, integrating normalizing flows to characterize the invariant distribution and facilitate sample generation from a base distribution. Our randomized test function circumvents the need for min-max optimization in the traditional weak formulation. Our method necessitates neither the computationally intensive calculation of the Jacobian determinant nor the invertibility of the transformation map. A crucial component of our framework is the adaptively chosen family of test functions in the form of Gaussian kernel functions with centers related to the generated data samples. Experimental results on several benchmark examples demonstrate the effectiveness and scalability of our method, which offers both low computational costs and excellent capability in exploring multiple metastable states.
format Preprint
id arxiv_https___arxiv_org_abs_2405_19256
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle Weak Generative Sampler to Efficiently Sample Invariant Distribution of Stochastic Differential Equation
Cai, Zhiqiang
Cao, Yu
Huang, Yuanfei
Zhou, Xiang
Machine Learning
Numerical Analysis
Mathematical Physics
Dynamical Systems
Probability
Sampling invariant distributions from an Itô diffusion process presents a significant challenge in stochastic simulation. Traditional numerical solvers for stochastic differential equations require both a fine step size and a lengthy simulation period, resulting in biased and correlated samples. The current deep learning-based method solves the stationary Fokker--Planck equation to determine the invariant probability density function in the form of deep neural networks, but they generally do not directly address the problem of sampling from the computed density function. In this work, we introduce a framework that employs a weak generative sampler (WGS) to directly generate independent and identically distributed (iid) samples induced by a transformation map derived from the stationary Fokker--Planck equation. Our proposed loss function is based on the weak form of the Fokker--Planck equation, integrating normalizing flows to characterize the invariant distribution and facilitate sample generation from a base distribution. Our randomized test function circumvents the need for min-max optimization in the traditional weak formulation. Our method necessitates neither the computationally intensive calculation of the Jacobian determinant nor the invertibility of the transformation map. A crucial component of our framework is the adaptively chosen family of test functions in the form of Gaussian kernel functions with centers related to the generated data samples. Experimental results on several benchmark examples demonstrate the effectiveness and scalability of our method, which offers both low computational costs and excellent capability in exploring multiple metastable states.
title Weak Generative Sampler to Efficiently Sample Invariant Distribution of Stochastic Differential Equation
topic Machine Learning
Numerical Analysis
Mathematical Physics
Dynamical Systems
Probability
url https://arxiv.org/abs/2405.19256