Robust SDE-Based Variational Formulations for Solving Linear PDEs via Deep Learning
Lorenz Richter, Julius Berner
Abstract
The combination of Monte Carlo methods and deep learning has recently led to efficient algorithms for solving partial differential equations (PDEs) in high dimensions. Related learning problems are often stated as variational formulations based on associated stochastic differential equations (SDEs), which allow the minimization of corresponding losses using gradient-based optimization methods. In respective numerical implementations it is therefore crucial to rely on adequate gradient estimators that exhibit low variance in order to reach convergence accurately and swiftly. In this article, we rigorously investigate corresponding numerical aspects that appear in the context of linear Kolmogorov PDEs. In particular, we systematically compare existing deep learning approaches and provide theoretical explanations for their performances. Subsequently, we suggest novel methods that can be shown to be more robust both theoretically and numerically, leading to substantial performance improvements. * Equal contribution (the author order was determined by numpy.random.rand(1)) 1 dida Datenschmiede GmbH, Germany
Ask about this paper
Your agent reads all of it.
Lune indexed this paper to the last equation, along with the top-tier papers that cite it. Ask a question and the answer quotes them.
Cited by top-tier papers5
- Improved sampling via learned diffusionsLorenz Richter, Julius BernerICLR 2024 · 103 citations
- Trust Region Constrained Measure Transport in Path Space for Stochastic Optimal Control and InferenceDenis Blessing, Julius Berner, Lorenz Richter, Carles Domingo-Enrich et al.NeurIPS 2025 · 24 citations
- Solving Poisson Equations using Neural Walk-on-SpheresHong Chul Nam, Julius Berner, Anima AnandkumarICML 2024 · 12 citations
- Learning ReLU networks to high uniform accuracy is intractableJulius Berner, Philipp Grohs, Felix VoigtländerICLR 2023 · 2 citations
- Amortized Control of Continuous State Space Feynman-Kac Model for Irregular Time SeriesByoungwoo Park, Hyungi Lee, Juho LeeICLR 2025
Builds on2
- A Variational Perspective on Diffusion-Based Generative Models and Score MatchingChin-Wei Huang, Jae Hyun Lim, Aaron C. CourvilleNeurIPS 2021 · 246 citations
- Numerically Solving Parametric Families of High-Dimensional Kolmogorov Partial Differential Equations via Deep LearningJulius Berner, Markus Dablander, Philipp GrohsNeurIPS 2020 · 58 citations
Related papers
- Deep Stochastic MechanicsElena Orlova, Aleksei Ustimenko, Ruoxi Jiang, Peter Y. Lu et al.ICML 2024 · 2 citations
- An Interpretable Approach to the Solutions of High-Dimensional Partial Differential EquationsLulu Cao, Yufei Liu, Zhenzhong Wang, Dejun Xu et al.AAAI 2024 · 15 citations
- Scalable Inference in SDEs by Direct Matching of the Fokker-Planck-Kolmogorov EquationArno Solin, Ella Tamir, Prakhar VermaNeurIPS 2021 · 23 citations
- Gold-standard solutions to the Schrödinger equation using deep learning: How much physics do we need?Leon Gerard, Michael Scherbela, Philipp Marquetand, Philipp GrohsNeurIPS 2022 · 51 citations
- Deep Latent Regularity Network for Modeling Stochastic Partial Differential EquationsShiqi Gong, Peiyan Hu, Qi Meng, Yue Wang et al.AAAI 2023 · 7 citations
