ICML 2026poster0 citations

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

Jaemin Seo, Su Rin Lee, JaeYong Lee

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.

FairnessRetrievalBenchmark
BibTeX
@inproceedings{
seo2026unbiased,
title={Unbiased and Second-Order-Free Training for High-Dimensional {PDE}s},
author={Jaemin Seo and Su Rin Lee and Jae Yong Lee},
booktitle={Forty-third International Conference on Machine Learning},
year={2026},
url={https://openreview.net/forum?id=ysBZhSRtCM}
}