NUNO: A General Framework for Learning Parametric PDEs with Non-Uniform Data
Not on the map. Why?
Abstract (arXiv)
The neural operator has emerged as a powerful tool in learning mappings between function spaces in PDEs. However, when faced with real-world physical data, which are often highly non-uniformly distributed, it is challenging to use mesh-based techniques such as the FFT. To address this, we introduce the Non-Uniform Neural Operator (NUNO), a comprehensive framework designed for efficient operator learning with non-uniform data. Leveraging a K-D tree-based domain decomposition, we transform non-uniform data into uniform grids while effectively controlling interpolation error, thereby paralleling the speed and accuracy of learning from non-uniform data. We conduct extensive experiments on 2D elasticity, (2+1)D channel flow, and a 3D multi-physics heatsink, which, to our knowledge, marks a novel exploration into 3D PDE problems with complex geometries. Our framework has reduced error rates by up to 60% and enhanced training speeds by 2x to 30x. The code is now available at https://github.com/thu-ml/NUNO.
Key facts. Cited by 15 papers in the map (data through 7 Oct 2026). Builds on 20 of its 39 references that are in the map. First posted on arXiv 30 May 2023.
Builds on
20 of 39 references.
References · 20
- 2017Attention Is All You Needcited by 59,932
- 2015U-Net: Convolutional Networks for Biomedical Image Segmentationcited by 19,604
- 2016PointNet: Deep Learning on Point Sets for 3D Classification and Segmentationcited by 5,223
- 2020Fourier Neural Operator for Parametric Partial Differential Equationscited by 2,806
- 2020Implicit Neural Representations with Periodic Activation Functionscited by 2,214
- 2019DeepONet: Learning nonlinear operators for identifying differential equations based on the universal approximation theorem of operatorscited by 1,965
- 2020Learning to Simulate Complex Physics with Graph Networkscited by 867
- 2020Neural Operator: Graph Kernel Network for Partial Differential Equationscited by 722
- 2014Spatial Pyramid Pooling in Deep Convolutional Networks for Visual Recognitioncited by 719
- 2022Fourier Neural Operator with Learned Deformations for PDEs on General Geometriescited by 416
- 2022Message Passing Neural PDE Solverscited by 368
- 2021Choose a Transformer: Fourier or Galerkincited by 333
8 more
- 2023GNOT: A General Neural Operator Transformer for Operator Learningcited by 321
- 2017PDE-Net: Learning PDEs from Datacited by 304
- 2021Multiwavelet-based Operator Learning for Differential Equationscited by 229
- 2020PhyGeoNet: Physics-Informed Geometry-Adaptive Convolutional Neural Networks for Solving Parameterized Steady-State PDEs on Irregular Domaincited by 173
- 2022Physics-Informed Machine Learning: A Survey on Problems, Methods and Applicationscited by 88
- 2020DeepGreen: Deep Learning of Green's Functions for Nonlinear Boundary Value Problemscited by 61
- 2022Multi-resolution partial differential equations preserved learning framework for spatiotemporal dynamicscited by 44
- 2020nPINNs: nonlocal Physics-Informed Neural Networks for a parametrized nonlocal universal Laplacian operator. Algorithms and Applicationscited by 36
Built on by
Citing papers per year: 2023: 1; 2024: 3; 2025: 5; 2026: 6.
15 papers.
Sort:
By area:
- 2024DPOT: Auto-Regressive Denoising Operator Transformer for Large-Scale PDE Pre-Trainingcited by 137
- 2025Machine learning for modelling unstructured grid data in computational physics: a reviewcited by 12
- 2025Mixture-of-Experts Operator Transformer for Large-Scale PDE Pre-Trainingcited by 10
- 2024Towards General Neural Surrogate Solvers with Specialized Neural Acceleratorscited by 8
- 2025Operator Learning with Domain Decomposition for Geometry Generalization in PDE Solvingcited by 6
- 2024Reference Neural Operators: Learning the Smooth Dependence of Solutions of PDEs on Geometric Deformationscited by 5
- 2025Physics-informed Temporal Alignment for Auto-regressive PDE Foundation Modelscited by 5
- 2025GFocal: A Global-Focal Neural Operator for Solving PDEs on Arbitrary Geometriescited by 2
- 2023Enhancing Solutions for Complex PDEs: Introducing Complementary Convolution and Equivariant Attention in Fourier Neural Operatorscited by 1
- 2026Benchmarking Long Roll-outs of Auto-regressive Neural Operators for the Compressible Navier-Stokes Equations with Conserved Quantity Correctioncited by 1
- 2026Stabilizing autoregressive forecasts in chaotic systems via multi-rate latent recurrencecited by 0
- 2026UniFluids: Unified Neural Operator Learning with Conditional Flow-matchingcited by 0
- 2026Beyond Arbitrary Geometry: Topology Generalization In neural PDE Operators5 Sep 2026
- 2026Geometry-aware Latent Autoregressive Generative Model for PDEs in Complex Domains31 Aug 2026
- 2026Linear-Nonlinear Fusion Neural Operator for Partial Differential Equations25 Mar 2026
- 2026UniFluids: Unified Neural Operator Learning with Conditional Flow-matching19 Mar 2026
- 2026Benchmarking Long Roll-outs of Auto-regressive Neural Operators for the Compressible Navier-Stokes Equations with Conserved Quantity Correction30 Jan 2026
- 2026Stabilizing autoregressive forecasts in chaotic systems via multi-rate latent recurrence20 Jan 2026
- 2025Mixture-of-Experts Operator Transformer for Large-Scale PDE Pre-Training29 Oct 2025
- 2025GFocal: A Global-Focal Neural Operator for Solving PDEs on Arbitrary Geometries6 Aug 2025
- 2025Physics-informed Temporal Alignment for Auto-regressive PDE Foundation Models16 May 2025
- 2025Operator Learning with Domain Decomposition for Geometry Generalization in PDE Solving1 Apr 2025
- 2025Machine learning for modelling unstructured grid data in computational physics: a review13 Feb 2025
- 2024Reference Neural Operators: Learning the Smooth Dependence of Solutions of PDEs on Geometric Deformations27 May 2024
3 more
- 2024Towards General Neural Surrogate Solvers with Specialized Neural Accelerators2 May 2024
- 2024DPOT: Auto-Regressive Denoising Operator Transformer for Large-Scale PDE Pre-Training6 Mar 2024
- 2023Enhancing Solutions for Complex PDEs: Introducing Complementary Convolution and Equivariant Attention in Fourier Neural Operators21 Nov 2023
- 2024DPOT: Auto-Regressive Denoising Operator Transformer for Large-Scale PDE Pre-Trainingcited by 137
- 2025Machine learning for modelling unstructured grid data in computational physics: a reviewcited by 12
- 2024Towards General Neural Surrogate Solvers with Specialized Neural Acceleratorscited by 8
- 2025Operator Learning with Domain Decomposition for Geometry Generalization in PDE Solvingcited by 6
- 2024Reference Neural Operators: Learning the Smooth Dependence of Solutions of PDEs on Geometric Deformationscited by 5
- 2025Physics-informed Temporal Alignment for Auto-regressive PDE Foundation Modelscited by 5
- 2025GFocal: A Global-Focal Neural Operator for Solving PDEs on Arbitrary Geometriescited by 2
- 2023Enhancing Solutions for Complex PDEs: Introducing Complementary Convolution and Equivariant Attention in Fourier Neural Operatorscited by 1
- 2026Benchmarking Long Roll-outs of Auto-regressive Neural Operators for the Compressible Navier-Stokes Equations with Conserved Quantity Correctioncited by 1
- 2026Stabilizing autoregressive forecasts in chaotic systems via multi-rate latent recurrencecited by 0
- 2026Linear-Nonlinear Fusion Neural Operator for Partial Differential Equationscited by 0
- 2026Beyond Arbitrary Geometry: Topology Generalization In neural PDE Operatorscited by 0
- 2026Beyond Arbitrary Geometry: Topology Generalization In neural PDE Operators5 Sep 2026
- 2026Linear-Nonlinear Fusion Neural Operator for Partial Differential Equations25 Mar 2026
- 2026Benchmarking Long Roll-outs of Auto-regressive Neural Operators for the Compressible Navier-Stokes Equations with Conserved Quantity Correction30 Jan 2026
- 2026Stabilizing autoregressive forecasts in chaotic systems via multi-rate latent recurrence20 Jan 2026
- 2025GFocal: A Global-Focal Neural Operator for Solving PDEs on Arbitrary Geometries6 Aug 2025
- 2025Physics-informed Temporal Alignment for Auto-regressive PDE Foundation Models16 May 2025
- 2025Operator Learning with Domain Decomposition for Geometry Generalization in PDE Solving1 Apr 2025
- 2025Machine learning for modelling unstructured grid data in computational physics: a review13 Feb 2025
- 2024Reference Neural Operators: Learning the Smooth Dependence of Solutions of PDEs on Geometric Deformations27 May 2024
- 2024Towards General Neural Surrogate Solvers with Specialized Neural Accelerators2 May 2024
- 2024DPOT: Auto-Regressive Denoising Operator Transformer for Large-Scale PDE Pre-Training6 Mar 2024
- 2023Enhancing Solutions for Complex PDEs: Introducing Complementary Convolution and Equivariant Attention in Fourier Neural Operators21 Nov 2023