Stochastic Process Learning via Operator Flow Matching
Yaozhong Shi, Zachary E. Ross, Domniki Asimaki, Kamyar Azizzadenesheli
Abstract
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.
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.
Your agent calls
Luneget_paper_fulltext
Free to start. No credit card required.
Terminal
Install the CLIlune papers fulltext d86a1e8d-a715-438f-89d4-dadf973fff16Cited by top-tier papers3
- Guided Diffusion Sampling on Function Spaces with Applications to PDEsJiachen Yao, Abbas Mammadov, Julius Berner, Gavin Kerrigan et al.NeurIPS 2025 · 37 citations
- Generative Neural Operators through Diffusion Last LayerSungwon Park, Anthony Zhou, Hongjoong Kim, Amir Barati FarimaniICML 2026 · 1 citation
- FuncGenFoil: Airfoil Generation and Editing Model in Function SpaceJinouwen Zhang, Junjie Ren, Qianhong Ma, Jianyu Wu et al.NeurIPS 2025 · 1 citation
Builds on14
- Denoising Diffusion Probabilistic ModelsJonathan Ho, Ajay Jain, Pieter AbbeelNeurIPS 2020 · 35,902 citations
- Fourier Neural Operator for Parametric Partial Differential EquationsZongyi Li, Nikola Borislavov Kovachki, Kamyar Azizzadenesheli, Burigede Liu et al.ICLR 2021 · 3,911 citations
- DPM-Solver: A Fast ODE Solver for Diffusion Probabilistic Model Sampling in Around 10 StepsCheng Lu, Yuhao Zhou, Fan Bao, Jianfei Chen et al.NeurIPS 2022 · 2,653 citations
- Score-Based Generative Modeling through Stochastic Differential EquationsYang Song, Jascha Sohl-Dickstein, Diederik P. Kingma, Abhishek Kumar et al.ICLR 2021 · 1,270 citations
- Geometry-Informed Neural Operator for Large-Scale 3D PDEsZongyi Li, Nikola B. Kovachki, Christopher B. Choy, Boyi Li et al.NeurIPS 2023 · 461 citations
Related papers
- Functional Mean Flow in Hilbert SpaceZhiqi Li, Yuchen Sun, Greg Turk, Bo ZhuCVPR 2026 · 5 citations
- Flow Matching Neural ProcessesHussen Abu Hamad, Dan RosenbaumNeurIPS 2025 · 8 citations
- Multitask Learning with Stochastic InterpolantsHugo Negrel, Florentin Coeurdoux, Michael S. Albergo, Eric Vanden-EijndenNeurIPS 2025 · 3 citations
- Infinite Neural Operators: Gaussian processes on functionsDaniel Augusto de Souza, Yuchen Zhu, Jake Cunningham, Yuri F. Saporito et al.NeurIPS 2025 · 1 citation
- MARS: Meta-learning as Score Matching in the Function SpaceKrunoslav Lehman Pavasovic, Jonas Rothfuss, Andreas KrauseICLR 2023 · 1 citation
