Weak collocation regression method: Fast reveal hidden stochastic dynamics from high-dimensional aggregate data
Liwei Lu,Zhijun Zeng,Yan Jiang,Yi Zhu,Pipi Hu
DOI: https://doi.org/10.1016/j.jcp.2024.112799
IF: 4.645
2024-02-01
Journal of Computational Physics
Abstract:Revealing hidden dynamics from the stochastic data is a challenging problem as the randomness takes part in the evolution of the data. The problem becomes exceedingly hard if the trajectories of the stochastic data are absent in many scenarios. In this work, we propose the Weak Collocation Regression (WCR) method to learn the dynamics from the stochastic data without the labels of trajectories. This method utilize the governing equation of the probability distribution function–the Fokker-Planck (FP) equation. Using its weak form and integration by parts, we move all the spacial derivatives of the distribution function to the test functions which can be computed explicitly. Since the data is a sampling of the corresponding distribution function, we can compute the integrations in the weak form, which has no spacial derivatives on the distribution functions, by simply adding the values of the integrands at the data points. We further assume the unknown drift and diffusion terms can be expanded by the base functions in a dictionary with the coefficients to be determined. Cooperating the collocation treatment and linear multi-step methods, we transfer the revealing process to a linear algebraic system. Using the sparse regression, we eventually obtain the unknown coefficients and hence the hidden stochastic dynamics. The numerical experiments show that our method is flexible and fast, which reveals the dynamics within seconds in the multi-dimensional problems and can be extended to high dimensional data. The complex tasks with variable-dependent diffusion and coupled drift can be correctly identified by WCR and the performance is robust, achieving high accuracy in the cases of noisy data. The rigorous error estimate is also included to support our various numerical experiments.
physics, mathematical,computer science, interdisciplinary applications