U-FNO -- An enhanced Fourier neural operator-based deep-learning model for multiphase flow
Not on the map. Why?
Abstract (arXiv)
Numerical simulation of multiphase flow in porous media is essential for many geoscience applications. Machine learning models trained with numerical simulation data can provide a faster alternative to traditional simulators. Here we present U-FNO, a novel neural network architecture for solving multiphase flow problems with superior accuracy, speed, and data efficiency. U-FNO is designed based on the newly proposed Fourier neural operator (FNO), which has shown excellent performance in single-phase flows. We extend the FNO-based architecture to a highly complex CO2-water multiphase problem with wide ranges of permeability and porosity heterogeneity, anisotropy, reservoir conditions, injection configurations, flow rates, and multiphase flow properties. The U-FNO architecture is more accurate in gas saturation and pressure buildup predictions than the original FNO and a state-of-the-art convolutional neural network (CNN) benchmark. Meanwhile, it has superior data utilization efficiency, requiring only a third of the training data to achieve the equivalent accuracy as CNN. U-FNO provides superior performance in highly heterogeneous geological formations and critically important applications such as gas saturation and pressure buildup "fronts" determination. The trained model can serve as a general-purpose alternative to routine numerical simulations of 2D-radial CO2 injection problems with significant speed-ups than traditional simulators.
Key facts. Cited by 292 papers in the map (data through 7 Oct 2026). Builds on 8 of its 45 references that are in the map. First posted on arXiv 3 Sep 2021.
Builds on
8 of 45 references.
References · 8
- 2020Fourier Neural Operator for Parametric Partial Differential Equationscited by 2,806
- 2020Neural Operator: Graph Kernel Network for Partial Differential Equationscited by 722
- 2020Multipole Graph Neural Operator for Parametric Partial Differential Equationscited by 343
- 2019Physics-Constrained Deep Learning for High-dimensional Surrogate Modeling and Uncertainty Quantification without Labeled Datacited by 261
- 2018Bayesian Deep Convolutional Encoder-Decoder Networks for Surrogate Modeling and Uncertainty Quantificationcited by 199
- 2019A deep-learning-based surrogate model for data assimilation in dynamic subsurface flow problemscited by 59
- 2018Deep convolutional encoder-decoder networks for uncertainty quantification of dynamic multiphase flow in heterogeneous mediacited by 56
- 2021CCSNet: a deep learning modeling suite for CO2 storagecited by 13
Built on by
Citing papers per year: 2021: 4; 2022: 24; 2023: 60; 2024: 52; 2025: 78; 2026: 74.
292 papers.
Sort:
By area:
- 2022Scientific Machine Learning through Physics-Informed Neural Networks: Where we are and What's nextcited by 603
- 2022Fourier Neural Operator with Learned Deformations for PDEs on General Geometriescited by 416
- 2023GNOT: A General Neural Operator Transformer for Operator Learningcited by 321
- 2024Transolver: A Fast Transformer Solver for PDEs on General Geometriescited by 299
- 2022Transformer for Partial Differential Equations' Operator Learningcited by 270
- 2021Adaptive Fourier Neural Operators: Efficient Token Mixers for Transformerscited by 237
- 2023Spherical Fourier Neural Operators: Learning Stable Dynamics on the Spherecited by 225
- 2022U-NO: U-shaped Neural Operatorscited by 219
- 2022Towards Multi-spatiotemporal-scale Generalized PDE Modelingcited by 190
- 2023Towards Foundation Models for Scientific Machine Learning: Characterizing Scaling and Transfer Behaviorcited by 128
- 2023Scalable Transformer for PDE Surrogate Modelingcited by 123
- 2024DiffusionPDE: Generative PDE-Solving Under Partial Observationcited by 100
38 more
- 2022Clifford Neural Layers for PDE Modelingcited by 91
- 2022Physics-Informed Machine Learning: A Survey on Problems, Methods and Applicationscited by 88
- 2023Solving High-Dimensional PDEs with Latent Spectral Modelscited by 84
- 2023Multi-Grid Tensorized Fourier Neural Operator for High-Resolution PDEscited by 66
- 2023Fourier-DeepONet: Fourier-enhanced deep operator networks for full waveform inversion with improved accuracy, generalizability, and robustnesscited by 60
- 2024Neural Operators with Localized Integral and Differential Kernelscited by 59
- 2023Improved Operator Learning by Orthogonal Attentioncited by 58
- 2023Towards Stability of Autoregressive Neural Operatorscited by 54
- 2022Generative Adversarial Neural Operatorscited by 51
- 2023Fourier-MIONet: Fourier-enhanced multiple-input neural operators for multiphase modeling of geological carbon sequestrationcited by 46
- 2024Towards a Foundation Model for Partial Differential Equations: Multi-Operator Learning and Extrapolationcited by 45
- 2023Nonlocality and Nonlinearity Implies Universality in Operator Learningcited by 43
- 2022Deep Learning Methods for Partial Differential Equations and Related Parameter Identification Problemscited by 34
- 2023Out-of-distributional risk bounds for neural operators with applications to the Helmholtz equationcited by 32
- 2023Neural Operator Learning for Long-Time Integration in Dynamical Systems with Recurrent Neural Networkscited by 32
- 2024Unisolver: PDE-Conditional Transformers Towards Universal Neural PDE Solverscited by 31
- 2024Conformalized-DeepONet: A Distribution-Free Framework for Uncertainty Quantification in Deep Operator Networkscited by 28
- 2023Physics informed WNOcited by 26
- 2022Partial Differential Equations Meet Deep Neural Networks: A Surveycited by 25
- 2022Incremental Spatial and Spectral Learning of Neural Operators for Solving Large-Scale PDEscited by 25
- 2023Rethinking materials simulations: Blending direct numerical simulations with neural operatorscited by 25
- 2022Physics-Guided, Physics-Informed, and Physics-Encoded Neural Networks in Scientific Computingcited by 24
- 2023Deep Operator Network Approximation Rates for Lipschitz Operatorscited by 24
- 2022Transformer Meets Boundary Value Inverse Problemscited by 22
- 2023Variational Autoencoding Neural Operatorscited by 22
- 2024VICON: Vision In-Context Operator Networks for Multi-Physics Fluid Dynamics Predictioncited by 22
- 2023Deep Learning in Deterministic Computational Mechanicscited by 21
- 2023Stacked networks improve physics-informed training: applications to neural networks and deep operator networkscited by 21
- 2022Transform Once: Efficient Operator Learning in Frequency Domaincited by 20
- 2023Beyond Regular Grids: Fourier-Based Neural Operators on Arbitrary Domainscited by 20
- 2023Improved generalization with deep neural operators for engineering systems: Path towards digital twincited by 19
- 2024PhyMPGN: Physics-encoded Message Passing Graph Network for spatiotemporal PDE systemscited by 18
- 2022NeurOLight: A Physics-Agnostic Neural Operator Enabling Parametric Photonic Device Simulationcited by 17
- 2022Bounding the Rademacher Complexity of Fourier neural operatorscited by 16
- 2023PROSE: Predicting Operators and Symbolic Expressions using Multimodal Transformerscited by 16
- 2023Multi-scale Time-stepping of Partial Differential Equations with Transformerscited by 14
- 2024MODNO: Multi Operator Learning With Distributed Neural Operatorscited by 14
- 2025A multi-agentic framework for real-time, autonomous freeform metasurface designcited by 14
50 most cited of 292 All 292 (JSON)
- 2026Learning PDE Dynamics between Submanifolds Using Green's Observation Operators1 Oct 2026
- 2026Cluster Attention Neural Operators for Solving Parametric Partial Differential Equations30 Sep 2026
- 2026Scale-Split Neural Operator for Memory- and Data-Efficient 3D Turbulence Prediction30 Sep 2026
- 2026PosteriorBench: From Point Estimates to Posterior Matching in Evaluating Generative Inverse Solvers17 Sep 2026
- 2026HiLNO: A Hierarchical Latent Neural Operator with Multi-Scale Supervision for PDEs on General Geometries16 Sep 2026
- 2026DU-NO: A Parameter-Efficient Double U-Shaped Neural Operator for Phase-Resolving Wave Modeling10 Sep 2026
- 2026Equation Recast for Canonical Operator Learning Across Parametric PDEs2 Sep 2026
- 2026Learning the Constitutive Behavior of Materials via Neural Operators and Causal Attention: Case Studies in Plasticity and Damage2 Sep 2026
- 2026Geometry-aware Latent Autoregressive Generative Model for PDEs in Complex Domains31 Aug 2026
- 2026Spectral-Embedded Operator Learning for Three-Phase Interfacial Flow: A Ternary Cahn-Hilliard-Navier-Stokes Benchmark29 Aug 2026
- 2026Quantum SEDONet: Spectrally-Embedded Quantum Deep Operator Networks for Partial Differential Equations27 Aug 2026
- 2026A Query-Time Framework for Transient 2D Pore-Scale Flow Prediction and Generative Design23 Aug 2026
38 more
- 2026The Kuramoto Neural Operator: Learning to Solve PDEs via Coupled Oscillator Dynamics10 Aug 2026
- 2026MoNo: Multiscale Optimal Transport Neural Operator for Solving PDEs on General Geometries10 Aug 2026
- 2026Neural Operators for Immersed-Boundary Soft Swimmers Locomotion7 Aug 2026
- 2026From Points to Edges: Edge-Conditioned Spectral Operators for Physics-Sensitive PDE Learning7 Aug 2026
- 2026A neural operator view on U-Nets for inverse imaging problems6 Aug 2026
- 2026Geometry-aware Incremental Neural Operator for Long-Horizon PDE prediction31 Jul 2026
- 2026Elliptic Regularity Theory in Barron Spaces and Applications to the Deep Ritz Method27 Jul 2026
- 2026Extending Fourier Neural Operators for Modeling Parameterized and Coupled PDEs26 Jul 2026
- 2026Adaptive Mamba Neural Operators20 Jul 2026
- 2026Physics-Audited Agentic Discovery in Scientific Machine Learning8 Jul 2026
- 2026CoFINN: Conservation Flux Informed Neural Networks for Physics Problems Governed by Conservation Laws5 Jul 2026
- 2026LLT: Local Linear Transformer for PDE Operator Learning4 Jul 2026
- 2026Mind the Residual Gap: Probabilistic Downscaling under Real-World Bias29 Jun 2026
- 2026Higher-Order Fourier Neural Operator: Explicit Mode Mixer for Nonlinear PDEs26 Jun 2026
- 2026Diffusion-corrected Autoregressive Fourier Neural Operator for Droplet Evolution Prediction25 Jun 2026
- 2026Geometry-Conditioned Fourier Neural Operators for Cubic Nonlinear Schrodinger Dynamics on Periodic Domains25 Jun 2026
- 2026A Physics-Informed Fourier-Wavelet Transformer for Multiscale Computational Fluid Dynamics Surrogate Modeling23 Jun 2026
- 2026Adaptive Distance-Aware Trunk Deep Operator Learning for Long-Span Roadway Bridges18 Jun 2026
- 2026A Validated LBM Dataset and Pipeline for Surrogate Modeling of Turbulent 3D Obstructed Channel Flows15 Jun 2026
- 2026HAMNO: A Hierarchical Adaptive Multi-scale Neural Operator with Physics-Informed Learning for Dynamical Systems10 Jun 2026
- 2026SirenFNO: Efficient and Full Frequency Learning of Fourier Neural Operators9 Jun 2026
- 2026Data assimilation for subsurface flow using latent diffusion model parameterization: performance of ensemble-Kalman and Monte Carlo techniques9 Jun 2026
- 2026On the training of physics-informed neural operators for solving parametric partial differential equations4 Jun 2026
- 2026Let There Be Light: Reflection, Refraction and Scattering for Neural Operators2 Jun 2026
- 2026FLAME: Physics-Guided Neural Operators for Onboard Satellite Methane Detection in Hyperspectral Imagery1 Jun 2026
- 2026Functional Attention: From Pairwise Affinities to Functional Correspondences29 May 2026
- 2026Physics-Informed Coarsening for Multigrid Graph Neural Surrogates29 May 2026
- 2026IV-Net: A neural network for elliptic PDEs with random and highly varying coefficients24 May 2026
- 2026WLNO: Wavelet-Laplace Neural Operator for Solving Partial Differential Equations23 May 2026
- 2026UFO: A Domain-Unification-Free Operator Framework for Generalized Operator Learning12 May 2026
- 2026Don't Fix the Basis -- Learn It: Spectral Representation with Adaptive Basis Learning for PDEs11 May 2026
- 2026Stable Long-Horizon PDE Forecasting via Latent Structured Spectral Propagators11 May 2026
- 2026Bridging Spectral Operator Learning and U-Net Hierarchies: SpectraNet for Stable Autoregressive PDE Surrogates9 May 2026
- 2026CATO: Charted Attention for Neural PDE Operators9 May 2026
- 2026Controlling Transient Amplification Improves Long-horizon Rollouts9 May 2026
- 2026Excluding the Target Domain Improves Extrapolation: Deconfounded Hierarchical Physics Constraints8 May 2026
- 2026Do Neural Operators Forget Geometry? The Forgetting Hypothesis in Deep Operator Learning7 May 2026
- 2026Conformalized Quantum DeepONet Ensembles: Towards Scalable Operator Learning with Distribution-Free Guarantees1 May 2026
50 newest of 292 All 292 (JSON)
- 2024DiffusionPDE: Generative PDE-Solving Under Partial Observationcited by 100
- 2024Data-Driven Stochastic Closure Modeling via Conditional Diffusion Model and Neural Operatorcited by 12
- 2023Inexpensive High Fidelity Melt Pool Models in Additive Manufacturing Using Generative Deep Diffusioncited by 11
- 2025Physics-Constrained Fine-Tuning of Flow-Matching Models for Generation and Inverse Problemscited by 8
- 2025VideoPDE: Unified Generative PDE Solving via Video Inpainting Diffusion Modelscited by 7
- 2024Generative prediction of flow fields around an obstacle using the diffusion modelcited by 6
- 2024DiffFluid: Plain Diffusion Models are Effective Predictors of Flow Dynamicscited by 4
- 2024Arbitrarily-Conditioned Multi-Functional Diffusion for Multi-Physics Emulationcited by 4
- 2026FlowRefiner: Flow Matching-Based Iterative Refinement for 3D Turbulent Flow Simulationcited by 4
- 2026Function-Space Decoupled Diffusion for Forward and Inverse Modeling in Carbon Capture and Storagecited by 2
- 2023DFU: scale-robust diffusion model for zero-shot super-resolution image generationcited by 1
- 2025GeoFunFlow: Geometric Function Flow Matching for Inverse Operator Learning over Complex Geometriescited by 0
12 most cited of 15 All 15 (JSON)
- 2026Geometry-aware Latent Autoregressive Generative Model for PDEs in Complex Domains31 Aug 2026
- 2026Diffusion-corrected Autoregressive Fourier Neural Operator for Droplet Evolution Prediction25 Jun 2026
- 2026Data assimilation for subsurface flow using latent diffusion model parameterization: performance of ensemble-Kalman and Monte Carlo techniques9 Jun 2026
- 2026FlowRefiner: Flow Matching-Based Iterative Refinement for 3D Turbulent Flow Simulation18 Apr 2026
- 2026Function-Space Decoupled Diffusion for Forward and Inverse Modeling in Carbon Capture and Storage12 Feb 2026
- 2025GeoFunFlow: Geometric Function Flow Matching for Inverse Operator Learning over Complex Geometries28 Sep 2025
- 2025Physics-Constrained Fine-Tuning of Flow-Matching Models for Generation and Inverse Problems5 Aug 2025
- 2025VideoPDE: Unified Generative PDE Solving via Video Inpainting Diffusion Models16 Jun 2025
- 2024Arbitrarily-Conditioned Multi-Functional Diffusion for Multi-Physics Emulation17 Oct 2024
- 2024DiffFluid: Plain Diffusion Models are Effective Predictors of Flow Dynamics20 Sep 2024
- 2024Data-Driven Stochastic Closure Modeling via Conditional Diffusion Model and Neural Operator6 Aug 2024
- 2024Generative prediction of flow fields around an obstacle using the diffusion model30 Jun 2024
12 newest of 15 All 15 (JSON)
- 2025Latent Mamba Operator for Partial Differential Equationscited by 8
- 2025GeoMaNO: Geometric Mamba Neural Operator for Partial Differential Equationscited by 2
- 2025Physics Consistency and Latent Dynamics in Spatiotemporal Physics Field Generationcited by 0
- 2026Adaptive Mamba Neural Operatorscited by 0
- 2026Adaptive Mamba Neural Operators20 Jul 2026
- 2025Latent Mamba Operator for Partial Differential Equations25 May 2025
- 2025GeoMaNO: Geometric Mamba Neural Operator for Partial Differential Equations17 May 2025
- 2025Physics Consistency and Latent Dynamics in Spatiotemporal Physics Field Generation16 May 2025
- 2022Scientific Machine Learning through Physics-Informed Neural Networks: Where we are and What's nextcited by 603
- 2022Fourier Neural Operator with Learned Deformations for PDEs on General Geometriescited by 416
- 2023GNOT: A General Neural Operator Transformer for Operator Learningcited by 321
- 2024Transolver: A Fast Transformer Solver for PDEs on General Geometriescited by 299
- 2022Transformer for Partial Differential Equations' Operator Learningcited by 270
- 2021Adaptive Fourier Neural Operators: Efficient Token Mixers for Transformerscited by 237
- 2023Spherical Fourier Neural Operators: Learning Stable Dynamics on the Spherecited by 225
- 2022U-NO: U-shaped Neural Operatorscited by 219
- 2022Towards Multi-spatiotemporal-scale Generalized PDE Modelingcited by 190
- 2023Towards Foundation Models for Scientific Machine Learning: Characterizing Scaling and Transfer Behaviorcited by 128
- 2023Scalable Transformer for PDE Surrogate Modelingcited by 123
- 2022Clifford Neural Layers for PDE Modelingcited by 91
12 most cited of 270 All 270 (JSON)
- 2026Learning PDE Dynamics between Submanifolds Using Green's Observation Operators1 Oct 2026
- 2026Cluster Attention Neural Operators for Solving Parametric Partial Differential Equations30 Sep 2026
- 2026Scale-Split Neural Operator for Memory- and Data-Efficient 3D Turbulence Prediction30 Sep 2026
- 2026PosteriorBench: From Point Estimates to Posterior Matching in Evaluating Generative Inverse Solvers17 Sep 2026
- 2026HiLNO: A Hierarchical Latent Neural Operator with Multi-Scale Supervision for PDEs on General Geometries16 Sep 2026
- 2026DU-NO: A Parameter-Efficient Double U-Shaped Neural Operator for Phase-Resolving Wave Modeling10 Sep 2026
- 2026Equation Recast for Canonical Operator Learning Across Parametric PDEs2 Sep 2026
- 2026Learning the Constitutive Behavior of Materials via Neural Operators and Causal Attention: Case Studies in Plasticity and Damage2 Sep 2026
- 2026Spectral-Embedded Operator Learning for Three-Phase Interfacial Flow: A Ternary Cahn-Hilliard-Navier-Stokes Benchmark29 Aug 2026
- 2026Quantum SEDONet: Spectrally-Embedded Quantum Deep Operator Networks for Partial Differential Equations27 Aug 2026
- 2026A Query-Time Framework for Transient 2D Pore-Scale Flow Prediction and Generative Design23 Aug 2026
- 2026The Kuramoto Neural Operator: Learning to Solve PDEs via Coupled Oscillator Dynamics10 Aug 2026
12 newest of 270 All 270 (JSON)