首页|A deep learning method based on prior knowledge with dual training for solving FPK equation

A deep learning method based on prior knowledge with dual training for solving FPK equation

扫码查看
The evolution of the probability density function of a stochastic dynamical system over time can be described by a Fokker-Planck-Kolmogorov(FPK)equation,the solution of which determines the distribution of macroscopic variables in the stochastic dynamic system.Traditional methods for solving these equations often struggle with computational efficiency and scalability,particularly in high-dimensional contexts.To address these challenges,this paper proposes a novel deep learning method based on prior knowledge with dual training to solve the stationary FPK equations.Initially,the neural network is pre-trained through the prior knowledge obtained by Monte Carlo simulation(MCS).Subsequently,the second training phase incorporates the FPK differential operator into the loss function,while a supervisory term consisting of local maximum points is specifically included to mitigate the generation of zero solutions.This dual-training strategy not only expedites convergence but also enhances computational efficiency,making the method well-suited for high-dimensional systems.Numerical examples,including two different two-dimensional(2D),six-dimensional(6D),and eight-dimensional(8D)systems,are conducted to assess the efficacy of the proposed method.The results demonstrate robust performance in terms of both computational speed and accuracy for solving FPK equations in the first three systems.While the method is also applicable to high-dimensional systems,such as 8D,it should be noted that computational efficiency may be marginally compromised due to data volume constraints.

deep learningprior knowledgeFPK equationprobability density function

彭登辉、王神龙、黄元辰

展开 >

School of Mechanical Engineering,University of Shanghai for Science and Technology,Shanghai 200093,China

National Natural Science Foundation of China

12172226

2024

中国物理B(英文版)
中国物理学会和中国科学院物理研究所

中国物理B(英文版)

CSTPCDEI
影响因子:0.995
ISSN:1674-1056
年,卷(期):2024.33(1)
  • 29