Universal Functional Regression with Neural Operator Flows
Not on the map. Why?
Abstract (arXiv)
Regression on function spaces is typically limited to models with Gaussian process priors. We introduce the notion of universal functional regression, in which we aim to learn a prior distribution over non-Gaussian function spaces that remains mathematically tractable for functional regression. To do this, we develop Neural Operator Flows (OpFlow), an infinite-dimensional extension of normalizing flows. OpFlow is an invertible operator that maps the (potentially unknown) data function space into a Gaussian process, allowing for exact likelihood estimation of functional point evaluations. OpFlow enables robust and accurate uncertainty quantification via drawing posterior samples of the Gaussian process and subsequently mapping them into the data function space. We empirically study the performance of OpFlow on regression and generation tasks with data generated from Gaussian processes with known posterior forms and non-Gaussian processes, as well as real-world earthquake seismograms with an unknown closed-form distribution.
Key facts. Cited by 9 papers in the map (data through 14 Oct 2026). Builds on 48 of its 79 references that are in the map. First posted on arXiv 3 Apr 2024.
Builds on
48 of 79 references.
References · 48
- 2020An Image is Worth 16x16 Words: Transformers for Image Recognition at Scalecited by 30,186
- 2020Score-Based Generative Modeling through Stochastic Differential Equationscited by 9,122
- 2022Generative Adversarial Networkscited by 5,627
- 2018Neural Ordinary Differential Equationscited by 3,779
- 2020Fourier Neural Operator for Parametric Partial Differential Equationscited by 2,908
- 2018Glow: Generative Flow with Invertible 1x1 Convolutionscited by 1,545
- 2019Normalizing Flows for Probabilistic Modeling and Inferencecited by 1,323
- 2022FourCastNet: A Global Data-driven High-resolution Weather Model using Adaptive Fourier Neural Operatorscited by 776
- 2020Neural Operator: Graph Kernel Network for Partial Differential Equationscited by 734
- 2018GPyTorch: Blackbox Matrix-Matrix Gaussian Process Inference with GPU Accelerationcited by 609
- 2021Neural Operator: Learning Maps Between Function Spacescited by 414
- 2023Neural Operators for Accelerating Scientific Simulations and Designcited by 221
36 more
- 2020SRFlow: Learning the Super-Resolution Space with Normalizing Flowcited by 209
- 2018When Gaussian Process Meets Big Data: A Review of Scalable GPscited by 206
- 2017Doubly Stochastic Variational Inference for Deep Gaussian Processescited by 143
- 2020The Random Feature Model for Input-Output Maps between Banach Spacescited by 90
- 2021Generative Models as Distributions of Functionscited by 89
- 2023Score-based Diffusion Models in Function Spacecited by 82
- 2020Augmented Normalizing Flows: Bridging the Gap Between Generative Flows and Latent Variable Modelscited by 66
- 2024Neural Operators with Localized Integral and Differential Kernelscited by 62
- 2020Deep Probabilistic Imaging: Uncertainty Quantification and Multi-modal Solution Characterization for Computational Imagingcited by 57
- 2020A Survey of Constrained Gaussian Process Regression: Approaches and Implementation Challengescited by 52
- 2022Generative Adversarial Neural Operatorscited by 52
- 2019WaveFlow: A Compact Flow-based Model for Raw Audiocited by 51
- 2021Seismic wave propagation and inversion with Neural Operatorscited by 51
- 2022Neural Diffusion Processescited by 50
- 2022Diffusion Generative Models in Infinite Dimensionscited by 49
- 2020Composing Normalizing Flows for Inverse Problemscited by 48
- 2019Leveraging exploration in off-policy algorithms via normalizing flowscited by 46
- 2023Infinite-Dimensional Diffusion Modelscited by 39
- 2023Continuous-Time Functional Diffusion Processescited by 39
- 2023Conditional score-based diffusion models for Bayesian inference in infinite dimensionscited by 38
- 2023Multilevel Diffusion: Infinite Dimensional Score-Based Diffusion Models for Image Generationcited by 37
- 2020The Expressive Power of a Class of Normalizing Flow Modelscited by 35
- 2018Inference in Deep Gaussian Processes using Stochastic Gradient Hamiltonian Monte Carlocited by 32
- 2020Bayesian Image Reconstruction using Deep Generative Modelscited by 29
- 2020Transforming Gaussian Processes With Normalizing Flowscited by 28
- 2020Optimal Thinning of MCMC Outputcited by 26
- 2020Fast Matrix Square Roots with Applications to Gaussian Processes and Bayesian Optimizationcited by 26
- 2023Variational Autoencoding Neural Operatorscited by 22
- 2020Deep Sigma Point Processescited by 17
- 2021Blending Anti-Aliasing into Vision Transformercited by 16
- 2019Compositionally-Warped Gaussian Processescited by 14
- 2023Globally injective and bijective neural operatorscited by 13
- 2023Broadband Ground Motion Synthesis via Generative Adversarial Neural Operators: Development and Validationcited by 12
- 2019Compositional uncertainty in deep Gaussian processescited by 8
- 2022PaCMO: Partner Dependent Human Motion Generation in Dyadic Human Activity using Neural Operatorscited by 8
- 2023Thin and Deep Gaussian Processescited by 3
Built on by
Citing papers per year: 2024: 1; 2025: 5; 2026: 3.
9 papers.
- 2025Guided Diffusion Sampling on Function Spaces with Applications to PDEscited by 36
- 2025Stochastic Process Learning via Operator Flow Matchingcited by 14
- 2025Mesh-Informed Neural Operator : A Transformer Generative Approachcited by 11
- 2024FUSE: Fast Unified Simulation and Estimation for PDEscited by 8
- 2025Reframing Generative Models for Physical Systems using Stochastic Interpolantscited by 2
- 2025FuncGenFoil: Airfoil Generation and Editing Model in Function Spacecited by 0
- 2026Large-Scale 3D Ground-Motion Synthesis with Physics-Inspired Latent Operator Flow Matchingcited by 0
- 2026Flow Annealing Posterior Sampling for Function-Space Regression and Inverse Problemscited by 0
- 2026Orca: Neural Operators for Causal Reasoning in Continuous Timecited by 0
- 2026Orca: Neural Operators for Causal Reasoning in Continuous Time30 Jul 2026
- 2026Flow Annealing Posterior Sampling for Function-Space Regression and Inverse Problems21 Jun 2026
- 2026Large-Scale 3D Ground-Motion Synthesis with Physics-Inspired Latent Operator Flow Matching18 Mar 2026
- 2025Reframing Generative Models for Physical Systems using Stochastic Interpolants30 Sep 2025
- 2025Mesh-Informed Neural Operator : A Transformer Generative Approach20 Jun 2025
- 2025Guided Diffusion Sampling on Function Spaces with Applications to PDEs22 May 2025
- 2025FuncGenFoil: Airfoil Generation and Editing Model in Function Space15 Feb 2025
- 2025Stochastic Process Learning via Operator Flow Matching7 Jan 2025
- 2024FUSE: Fast Unified Simulation and Estimation for PDEs23 May 2024