Optimal Diffusion Processes
Saber Jafarizadeh
TL;DR
The paper addresses designing diffusion processes with a fixed stationary distribution $\pi(x)$ and fixed average variance $\widehat{\sigma}^{2}/2$ that converge most rapidly to equilibrium. It casts the problem as a variational spectral-gap optimization, solving a Rayleigh quotient via a min-max framework to show the optimum occurs with a linear drift and a nontrivial variance function, yielding $\lambda_1 = \widehat{\sigma}^{2}/[2 (m_2 - m_1^{2})]$ and $\phi_1(x)=(x-m_1)/\sqrt{m_2-m_1^{2}}$. The results imply a concavity of the optimal relaxation time in the stationary distribution and establish that all Pearson hypergeometric-type diffusions with variance degree up to 2 are optimal; the paper also provides explicit optimal parameters for higher-degree polynomial variance and non-polynomial variance (hyperexponential) cases. Collectively, these findings offer a principled method for designing diffusion samplers with maximal convergence speed under distributional and variance constraints, and illustrate the theory with concrete Pearson, polynomial, and non-polynomial examples.
Abstract
Of stochastic differential equations, diffusion processes have been adopted in numerous applications, as more relevant and flexible models. This paper studies diffusion processes in a different setting, where for a given stationary distribution and average variance, it seeks the diffusion process with optimal convergence rate. It is shown that the optimal drift function is a linear function and the convergence rate of the stochastic process is bounded by the ratio of the average variance to the variance of the stationary distribution. Furthermore, the concavity of the optimal relaxation time as a function of the stationary distribution has been proven, and it is shown that all Pearson diffusion processes of the Hypergeometric type with polynomial functions of at most degree two as the variance functions are optimal.
