Search papers, labs, and topics across Lattice.
This paper introduces a conditional normalizing flow-based physics-informed neural network (PINN) framework designed to efficiently approximate the solution operator of the Fokker-Planck equation (FPE) across various initial conditions. By reformulating the problem using the Chapman-Kolmogorov equation, the authors successfully approximate transition probability density functions (PDFs) from Dirac mass initial conditions, addressing challenges associated with singularities. The proposed method, which incorporates a time-weighted loss function, demonstrates significant robustness and effectiveness in numerical experiments, particularly for small time intervals.
Transitioning from Dirac mass initial conditions, this framework achieves robust approximations of Fokker-Planck solutions, even in the face of numerical instabilities.
The Fokker-Planck equation (FPE) plays a pivotal role in describing the time evolution of probability density functions (PDFs) for systems governed by stochastic dynamics. In this work, we propose a conditional normalizing flow-based physics-informed neural network (PINN) framework for efficiently approximating the solution operator of the FPE for a whole range of initial conditions. Leveraging the Chapman-Kolmogorov equation for Markovian stochastic processes, the problem is reformulated into approximating a transition PDF starting at initial time from a Dirac mass centered at an arbitrary point. The PDF of an associated linearized stochastic differential equation (SDE) is employed as the base distribution for the normalizing flow, providing a good approximation of the target PDF, especially for small times, and thereby avoiding the singularity of the map associated with the Dirac delta initial distribution. Furthermore, a time-weighted loss function is introduced to mitigate numerical instabilities arising at small times, achieving a balance between causality and training difficulty as time progresses. A variety of numerical experiments are presented to illustrate the effectiveness and robustness of the proposed method.
