改进随机微分方程求解器,让高维偏微分方程计算更准更快
Integration Matters for Learning PDEs with Backward SDEs
- 用斯特拉托诺维奇形式与随机海恩积分替代传统欧拉-马鲁雅玛方法
- 在多个高维测试集上超越原版BSDE方法,媲美物理信息神经网络
- 揭示积分方案对求解器性能的关键影响,适合研究高维PDE的学者
基于反向随机微分方程(BSDE)的深度学习方法为求解高维偏微分方程(PDEs)提供了不同于物理信息神经网络(PINNs)的路径,尤其在与动力系统相关的随机最优控制场景中具有潜在优势。然而,已有研究表明,标准的BSDE求解器在实际表现上逊于PINNs。本文指出这一性能差距的根本原因在于:标准欧拉-马鲁雅玛(EM)积分方案应用于单步自洽性BSDE损失时引入了离散化偏差,导致优化目标偏离真实解。该偏差无法通过减小步长或采用多步自洽性损失有效缓解。为此,我们提出基于斯特拉托诺维奇形式的BSDE公式,并结合随机海恩积分实现。实验表明,新方法完全消除了EM积分带来的偏差,且在多个高维基准测试中持续优于基于EM的方法,性能可与PINNs相媲美。研究强调了积分方案在基于BSDE的PDE求解器中的关键作用,这一算法细节此前在文献中几乎未受关注。
原文摘要 · Abstract (English)
Backward stochastic differential equation (BSDE)-based deep learning methods provide an alternative to Physics-Informed Neural Networks (PINNs) for solving high-dimensional partial differential equations (PDEs), offering potential algorithmic advantages in settings such as stochastic optimal control, where the PDEs of interest are tied to an underlying dynamical system. However, standard BSDE-based solvers have empirically been shown to underperform relative to PINNs in the literature. In this paper, we identify the root cause of this performance gap as a discretization bias introduced by the standard Euler-Maruyama (EM) integration scheme applied to one-step self-consistency BSDE losses, which shifts the optimization landscape off target. We find that this bias cannot be satisfactorily addressed through finer step-sizes or multi-step self-consistency losses. To properly handle this issue, we propose a Stratonovich-based BSDE formulation, which we implement with stochastic Heun integration. We show that our proposed approach completely eliminates the bias issues faced by EM integration. Furthermore, our empirical results show that our Heun-based BSDE method consistently outperforms EM-based variants and achieves competitive results with PINNs across multiple high-dimensional benchmarks. Our findings highlight the critical role of integration schemes in BSDE-based PDE solvers, an algorithmic detail that has received little attention thus far in the literature.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。