English

Unbiased and Second-Order-Free Training for High-Dimensional PDEs

Machine Learning 2026-05-15 v1 Numerical Analysis Numerical Analysis Optimization and Control

Abstract

Deep learning methods based on backward stochastic differential equations (BSDEs) have emerged as competitive alternatives to physics-informed neural networks (PINNs) for solving high-dimensional partial differential equations (PDEs). By leveraging probabilistic representations, BSDE approaches can avoid the curse of dimensionality and often admit second-order-free training objectives that do not require explicit Hessian evaluations. It has recently been established that the commonly used Euler-Maruyama (EM) time discretization induces an intrinsic bias in BSDE training losses. While high-order schemes such as Heun can fully eliminate this bias, such schemes re-introduce second-order spatial derivatives and incur substantial computational overhead. In this work, we provide a principled analysis of EM-induced loss bias and propose an unbiased, second-order-free training framework that preserves the computational advantages of BSDE methods. Our code is available at https://github.com/seojaemin22/Un-EM-BSDE.

Keywords

Cite

@article{arxiv.2605.14643,
  title  = {Unbiased and Second-Order-Free Training for High-Dimensional PDEs},
  author = {Jaemin Seo and Surin Lee and Jae Yong Lee},
  journal= {arXiv preprint arXiv:2605.14643},
  year   = {2026}
}

Comments

Accepted at ICML 2026

R2 v1 2026-07-22T07:12:03.452Z