.png){kind=icon}
.png){kind=icon}
Return
Deep JKO: Time-implicit particle methods for general nonlinear gradient flows
.png){kind=icon}
OA
AI
DOI：10.1016/j.jcp.2024.113187.png)
.png){kind=icon}
.png){kind=icon}
Share.png){kind=icon}
Save
Abstract
En 中文
We develop novel neural network -based implicit particle methods to compute high -dimensional Wasserstein -type gradient flows with linear and nonlinear mobility functions. The main idea is to use the Lagrangian formulation in the Jordan-Kinderlehrer-Otto (JKO) framework, where the velocity field is approximated using a neural network. We leverage the formulations from the neural ordinary differential equation (neural ODE) in the context of continuous normalizing flow for efficient density computation. Additionally, we make use of an explicit recurrence relation for computing derivatives, which greatly streamlines the backpropagation process. Our methodology demonstrates versatility in handling a wide range of gradient flows, accommodating various potential functions and nonlinear mobility scenarios. Extensive experiments demonstrate the efficacy of our approach, including an illustrative example from Bayesian inverse problems. This underscores that our scheme provides a viable alternative solver for the Kalman -Wasserstein gradient flow.
Keywords:
Nonlinear gradient flows
JKO scheme
Neural ODE
Particle methods
Density estimation
Nonlinear mobility
AI Summary
Key information extracted from the uploaded paper, including a brief overview, abstract, background, key highlights, visual analysis, and future outlook.
Journal
IF:
3.8
Papers:
1.5W
Citations:
7.4W