Skip to main navigation Skip to search Skip to main content

Taming the Loss Landscape of PINNs with Noisy Feynman–Kac Supervision: Operator Preconditioning and Non-Asymptotic Error Bounds

Research output: Conference PapersRGC 31A - Invited conference paper (refereed items)Yespeer-review

Abstract

Physics-Informed Neural Networks (PINNs) often train slowly or fail to converge on challenging partial differential equations (PDEs), a behavior recently linked to severely ill-conditioned loss landscapes inherited from the underlying differential operator. We propose FK-PINNs, a simple modification of the PINN objective that provably improves this conditioning: at a few points in the domain we compute Feynman--Kac estimates of the solution by Monte Carlo averaging, and add the resulting data-fidelity term to the standard residual and boundary losses. For a broad class of linear second-order PDEs admitting a Feynman--Kac representation, we show that this term acts as an operator-level preconditioner: for suitable weights, our comparison bounds guarantee a substantially smaller condition number than under the standard PINN loss, even for modest Monte Carlo sample budgets. Leveraging learning-theoretic tools, we derive non-asymptotic L2-error bounds for the FK-PINNs with tanh activation by decomposing the excess risk into approximation, statistical, and optimization error terms and tightly controlling the Monte Carlo error tails. Along the way, we establish pseudo-dimension bounds for first- and second-order derivatives of tanh neural networks, which are of independent interest and, to the best of our knowledge, new. Numerical experiments on Poisson, Schrödinger, mean exit time, and committor problems corroborate the theory, and show that FK-PINNs can successfully solve PDEs for which vanilla PINNs exhibit severe failure modes.
Original languageEnglish
Number of pages56
Publication statusPublished - Jul 2026
Event43rd International Conference on Machine Learning (ICML 2026) - Seoul, Korea, Republic of
Duration: 6 Jul 202611 Jul 2026
https://icml.cc/Conferences/2026

Conference

Conference43rd International Conference on Machine Learning (ICML 2026)
Abbreviated titleICML 2026
PlaceKorea, Republic of
CitySeoul
Period6/07/2611/07/26
Internet address

Bibliographical note

Since this conference is yet to commence, the information for this record is subject to revision.

Funding

N.T. acknowledges support from the Hong Kong PhD Fellowship Scheme. X.Z. acknowledges support from the Hong Kong General Research Funds (Grants No. 11304525, No. 11318522, and No. 11308323).

RGC Funding Information

  • RGC-funded

Fingerprint

Dive into the research topics of 'Taming the Loss Landscape of PINNs with Noisy Feynman–Kac Supervision: Operator Preconditioning and Non-Asymptotic Error Bounds'. Together they form a unique fingerprint.

Cite this