Stochastic Process Learning via Operator Flow Matching
About
Expanding on neural operators, we propose a novel framework for stochastic process learning across arbitrary domains. In particular, we develop operator flow matching (OFM) for learning stochastic process priors on function spaces. OFM provides the probability density of the values of any collection of points and enables mathematically tractable functional regression at new points with mean and density estimation. Our method outperforms state-of-the-art models in stochastic process learning, functional regression, and prior learning.
Yaozhong Shi, Zachary E. Ross, Domniki Asimaki, Kamyar Azizzadenesheli• 2025
Related benchmarks
| Task | Dataset | Result | Rank | |
|---|---|---|---|---|
| 1D Matérn GP regression | 1D Matérn GP (test) | SWD2.17 | 14 | |
| Functional Regression | 1D Gibbs GP Query size 128 | SWD2.87 | 7 | |
| Functional Regression | 1D Gibbs GP Query size 512 | SWD3.69 | 7 | |
| Non-Gaussian functional regression | Black hole | CRPS1.33 | 6 | |
| Non-Gaussian functional regression | Navier-Stokes | CRPS2.97 | 6 |
Showing 5 of 5 rows