Data driven adaptive Gaussian mixture model for solving Fokker–Planck equation

The Fokker-Planck (FP) equation provides a powerful tool for describing the state transition probability density function of complex dynamical systems governed by stochastic differential equations (SDEs). Unfortunately, the analytical solution of the FP equation can be found in very few special cases. Therefore, it has become an interest to find a numerical approximation method of the FP equation suitable for a wider range of nonlinear systems. In this paper, a machine learning method based on an adaptive Gaussian mixture model (AGMM) is proposed to deal with the general FP equations. Compared with previous numerical discretization methods, the proposed method seamlessly integrates data and mathematical models. The prior knowledge generated by the assumed mathematical model can improve the performance of the learning algorithm. Also, it yields more interpretability for machine learning methods. Numerical examples for one-dimensional and two-dimensional SDEs with one and/or two noises are given. The simulation results show the effectiveness and robustness of the AGMM technique for solving the FP equation. In addition, the computational complexity and the optimization algorithm of the model are also discussed.