Base Models for Parabolic Partial Differential Equations

Xingzi Xu, Ali Hasan, Jie Ding, Vahid Tarokh

Research output: Contribution to journalConference articlepeer-review

Abstract

Parabolic partial differential equations (PDEs) appear in many disciplines to model the evolution of various mathematical objects, such as probability flows, value functions in control theory, and derivative prices in finance. It is often necessary to compute the solutions or a function of the solutions to a parametric PDE in multiple scenarios corresponding to different parameters of this PDE. This process often requires resolving the PDEs from scratch, which is time-consuming. To better employ existing simulations for the PDEs, we propose a framework for finding solutions to parabolic PDEs across different scenarios by meta-learning an underlying base distribution. We build upon this base distribution to propose a method for computing solutions to parametric PDEs under different parameter settings. Finally, we illustrate the application of the proposed methods through extensive experiments in generative modeling, stochastic control, and finance. The empirical results suggest that the proposed approach improves generalization to solving new PDEs.

Original languageEnglish (US)
Pages (from-to)3853-3878
Number of pages26
JournalProceedings of Machine Learning Research
Volume244
StatePublished - 2024
Event40th Conference on Uncertainty in Artificial Intelligence, UAI 2024 - Barcelona, Spain
Duration: Jul 15 2024Jul 19 2024

Bibliographical note

Publisher Copyright:
© 2024 Proceedings of Machine Learning Research. All rights reserved.

Fingerprint

Dive into the research topics of 'Base Models for Parabolic Partial Differential Equations'. Together they form a unique fingerprint.

Cite this