Weak adversarial tensor neural network for high-dimensional Fokker-Planck equations
Abstract
Solving high-dimensional stationary Fokker-Planck (SFP) equations remains challenging. Recently, deep learning methods have shown great potential for solving high-dimensional SFP equations due to their universal approximation and mesh-free training. However, the high-dimensional integrals involved in loss computation and normalization can be expensive to approximate accurately using Monte Carlo integration. To address this issue, we propose a weak adversarial tensor neural network (WATNN) method for solving high-dimensional SFP equations, which combines a weak adversarial (WA) formulation with the tensor neural network (TNN). The WA formulation avoids the cross terms introduced by squaring the strong residual, reducing the number of integrals to be computed. When both the primal network and the adversarial test network are set as TNNs, the TNN quadrature scheme provides accurate and efficient normalization and loss computation with relatively few fixed quadrature points, avoiding the statistical sampling error that arises when using Monte Carlo integration. We demonstrate the effectiveness numerically on four high-dimensional SFP equations. For the 6D unimodal, 6D multi-modal, 10D multi-modal and 20D multi-modal problems, WATNN achieves the mean relative errors of 1.50%, 1.13%, 1.20% and 7.93% respectively, in only a few minutes.
Related
- Boundary-preserving randomized neural network methods for Poisson equations on general two- and three-dimensional domains
- Asymptotic-Preserving Tensor Neural Networks for 3D High-Field Kinetic Equations with Applications to Uncertainty Quantification
- A Tensor Neural Network Method for High-Order Homogenization of Locally Periodic Elliptic Problems
- A diffusion-free multi-layer neural-network method for multidimensional nonlinear hyperbolic equations
- A SAV Neural Network Method for Allen--Cahn and Cahn--Hilliard Equations
BibTeX
@misc{shao2026weak,
title = {{Weak adversarial tensor neural network for high-dimensional Fokker-Planck equations}},
author = {Sihong Shao and Zhongwei Xu},
howpublished = {arXiv},
year = {2026},
month = oct,
doi = {10.48550/arxiv.2610.32169},
eprint = {2610.32169},
archivePrefix = {arXiv},
url = {https://arxiv.org/abs/2610.32169},
}
RIS
TY - UNPB
TI - Weak adversarial tensor neural network for high-dimensional Fokker-Planck equations
AU - Sihong Shao
AU - Zhongwei Xu
PY - 2026
DA - 2026/10/23
PB - arXiv
DO - 10.48550/arxiv.2610.32169
UR - https://arxiv.org/abs/2610.32169
AB - Solving high-dimensional stationary Fokker-Planck (SFP) equations remains challenging. Recently, deep learning methods have shown great potential for solving high-dimensional SFP equations due to their universal approximation and mesh-free training. However, the high-dimensional integrals involved in loss computation and normalization can be expensive to approximate accurately using Monte Carlo integration. To address this issue, we propose a weak adversarial tensor neural network (WATNN) method for solving high-dimensional SFP equations, which combines a weak adversarial (WA) formulation with the tensor neural network (TNN). The WA formulation avoids the cross terms introduced by squaring the strong residual, reducing the number of integrals to be computed. When both the primal network and the adversarial test network are set as TNNs, the TNN quadrature scheme provides accurate and efficient normalization and loss computation with relatively few fixed quadrature points, avoiding the statistical sampling error that arises when using Monte Carlo integration. We demonstrate the effectiveness numerically on four high-dimensional SFP equations. For the 6D unimodal, 6D multi-modal, 10D multi-modal and 20D multi-modal problems, WATNN achieves the mean relative errors of 1.50%, 1.13%, 1.20% and 7.93% respectively, in only a few minutes.
ER -