We prove first-order convergence of semi-discrete monotone finite difference schemes for Hamilton--Jacobi equations on the Wasserstein space over a finite graph. A central challenge is the boundary degeneracy of the Wasserstein simplex, which prevents the direct use of the standard $L^1$ adjoint method and limits doubling-of-variables arguments to the suboptimal rate $\mathcal O(h^{\frac 12})$ \cite{CDM25}. We address this issue by introducing a weighted $L^1$ framework with a boundary-vanishing weight and by analyzing the corresponding weighted adjoint equation for the linearized operator of the scheme, featuring a new geometric drift term. Our proof relies on uniform bounds for the weighted adjoint variable and the mesh-parameter derivative of the numerical solution. These estimates are derived from discrete gradient and semi-concavity bounds, obtained through a bootstrap argument for two classes of monotone Hamiltonians.
In 1994, Nessyahu, Tadmor and Tassa studied convergence rates of monotone finite volume approximations of conservation laws. For compactly supported, $\Lip^+$-bounded initial data they showed a first-order convergence rate in the Wasserstein distance. Our main result is to prove that this rate is optimal. We further provide numerical evidence indicating that the rate in the case of $\Lip^+$-unbounded initial data is worse than first-order.
We study discretizations of Hamiltonian systems on the probability density manifold equipped with the $L^2$-Wasserstein metric. Based on discrete optimal transport theory, several Hamiltonian systems on graph (lattice) with different weights are derived, which can be viewed as spatial discretizations to the original Hamiltonian systems. We prove the consistency and provide the approximate orders for those discretizations. By regularizing the system using Fisher information, we deduce an explicit lower bound for the density function, which guarantees that symplectic schemes can be used to discr…
In this paper, we propose a numerical method to solve the classic $L^2$-optimal transport problem. Our algorithm is based on use of multiple shooting, in combination with a continuation procedure, to solve the boundary value problem associated to the transport problem. We exploit the viewpoint of Wasserstein Hamiltonian flow with initial and target densities, and our method is designed to retain the underlying Hamiltonian structure. Several numerical examples are presented to illustrate the performance of the method.
We study a class of regularized proximal operators in Wasserstein-2 space. We derive their solutions by kernel integration formulas. We obtain the Wasserstein proximal operator using a pair of forward-backward partial differential equations consisting of a continuity equation and a Hamilton-Jacobi equation with a terminal time potential function and an initial time density function. We regularize the PDE pair by adding forward and backward Laplacian operators. We apply Hopf-Cole type transformations to rewrite these regularized PDE pairs into forward-backward heat equations. We then use the fu…
Numerical moment stabilization of central difference approximations for linear stationary reaction-convection-diffusion equations with applications to stationary Hamilton-Jacobi equations
In this article, we consider the preconditioned Hamiltonian Monte Carlo (pHMC) algorithm defined directly on an infinite-dimensional Hilbert space. In this context, and under a condition reminiscent of strong log-concavity of the target measure, we prove convergence bounds for adjusted pHMC in the standard 1-Wasserstein distance. The arguments rely on a synchronous coupling of two copies of pHMC, which is controlled by adapting elements from arXiv:1805.00452.
Non-asymptotic convergence analysis of the stochastic gradient Hamiltonian Monte Carlo algorithm with discontinuous stochastic gradient with applications to training of ReLU neural networks
School of Mathematics, Renmin University of China, Beijing, China（中国人民大学数学学院）；Division of Mathematical Sciences, Nanyang Technological University, 21 Nanyang Link, 637371 Singapore（南洋理工大学数学科学学院）；Financial Technology Thrust, Society Hub, The Hong Kong University of Science and Technology (Guangzhou), Guangzhou, China（香港科技大学（广州）金融科技中心）
