SINO: Scale-Invariant Neural Operator
Organizations: Westlake University · University of Science and Technology of China
Abstract
In scientific machine learning, physical fields governed by partial differential equations exhibit low-rank structure and scale invariance. When solving equations on coarse grids, missing information leads to the closure problem: modeling unresolved physics to recover lost dynamics. Although closure terms depend on grid resolution, they represent scale-invariant physical laws. A model truly learning physics should capture these mechanisms with low-rank parameterization rather than memorizing grid-specific patterns. Inspired by this, we propose the Scale-Invariant Neural Operator (SINO), which learns on normalized physical scales via a dual-branch architecture operating in spectral and spatial domains. SINO uses bottleneck MLPs to generate continuous convolution kernels, embedding an explicit low-rank inductive bias that concentrates more than 95 percent of variance in 2-3 modes, as validated by PCA across benchmarks, while drastically reducing parameters. This principled design yields 38 times steeper scaling law exponents than FNO, demonstrating superior parameter efficiency. We compare SINO with traditional models (U-Net, DeepONet), Transformer models (Transolver, Oformer, GK-Transformer), and frequency-domain models (FNO, AMFNO, UFNO) on closure problems spanning externally forced Burgers turbulence, decaying Burgers turbulence, KS turbulence, Kolmogorov-forced NS turbulence, and decaying NS turbulence. Experiments show SINO achieves 1.5-38 times error reduction and 2-23 times parameter efficiency over baselines, with superior scaling laws reflecting exceptional data efficiency from principled low-rank design. Code is available at https://github.com/AI4Science-WestlakeU/SINO.
Figures & tables
| Problem | Dim | Resolution | Spatial Discretization | Temporal Scheme |
|---|---|---|---|---|
| Forcing Burgers | 1D | 512 32 | WENO5 Finite Volume Method | SSP-RK3 |
| Decaying Burgers | 1D | 2048 256 | Pseudo-Spectral | TVD RK3 |
| KS | 1D | 256 64 | Pseudo-Spectral | ETDRK4 |
| Forcing NS (Re=1000) | 2D | 512 64 | Finite Volume Method (Van Leer) | Semi-Implicit |
| Forcing NS (Re=4000) | 2D | 512 64 | Finite Volume Method (Van Leer) | Semi-Implicit |
| Decaying NS | 2D | 512 64 | Finite Volume Method (Van Leer) | Semi-Implicit |
| Method | 1D Benchmarks | Params | |||||
| Forcing Burgers | KS | Decaying Burgers | |||||
| AVE | Best | AVE | Best | AVE | Best | ||
| U-Net | 4.07E-03 | 3.31E-03 | 3.33E-03 | 1.48E-03 | 1.65E-01 | 1.65E-01 | 7,077 |
| DeepONet | 3.22E-03 | 2.78E-03 | 1.05E+00 | 1.39E+00 | 1.65E-01 | 1.65E-01 | 14,913 |
| Transolver | 2.94E-03 | 2.80E-03 | 1.14E-02 | 1.09E-02 | 1.65E-01 | 1.64E-01 | 8,601 |
| Oformer | 2.92E-03 | 2.92E-03 | 5.45E-03 | 5.45E-03 | 1.65E-01 | 1.65E-01 | 5,444 |
| Method | 2D Benchmarks | Params | |||||
| Forcing NS (Re=1000) | Forcing NS (Re=4000) | Decaying NS | |||||
| AVE | Best | AVE | Best | AVE | Best | ||
| U-Net | 1.34E-01 | 1.16E-01 | 1.91E-01 | 1.84E-01 | 6.67E-02 | 5.64E-02 | 61,006 |
| DeepONet | 4.62E-01 | 4.60E-01 | 5.25E-01 | 5.18E-01 | 2.95E-01 | 2.79E-01 | 269,153 |
| Transolver | 3.65E-01 | 3.48E-01 | 4.27E-01 | 4.23E-01 | 2.24E-01 | 2.17E-01 | 54,306 |
| Oformer | 5.07E-01 | 5.07E-01 | 5.33E-01 | 5.33E-01 | 4.03E-01 | 4.03E-01 | 38,087 |
Appendix figures & tables6 assets
Supplementary material from the paper’s appendix.
Appendix
| Method | Supplementary 1D Benchmarks | Params | |||||
| Forcing Burgers (128) | KS (32) | Decaying Burgers (128) | |||||
| AVE | Best | AVE | Best | AVE | Best | ||
| U-Net | 1.15E-04 | 7.76E-05 | 4.38E-01 | 5.92E-01 | 1.87E-01 | 1.87E-01 | 7,077 |
| DeepONet | 2.00E-04 | 1.72E-04 | 1.70E+00 | 1.39E+00 | 1.87E-01 | 1.87E-01 | 14,913 |
| Transolver | 2.60E-04 | 2.55E-04 | 7.47E-01 | 7.36E-01 | 1.87E-01 | 1.87E-01 | 8,601 |
| Oformer | 1.18E-04 | 1.18E-04 | 1.89E+01 | 1.89E+01 | 1.87E-01 | 1.87E-01 | 5,444 |
| Method | Supplementary 2D Benchmarks | Params | |||||
| Forcing NS (Re=2000) | NS (Taylor-Green, Re=1000) | Forcing NS (Re= ) | |||||
| AVE | Best | AVE | Best | AVE | Best | ||
| U-Net | 1.51E-01 | 1.48E-01 | 1.41E-02 | 1.09E-02 | 2.37E-01 | 2.31E-01 | 61,006 |
| DeepONet | 5.02E-01 | 4.96E-01 | 1.59E-01 | 1.56E-01 | 6.04E-01 | 5.91E-01 | 269,153 |
| Transolver | 4.72E-01 | 4.14E-01 | 6.61E-02 | 6.01E-02 | 5.59E-01 | 5.37E-01 | 54,306 |
| Oformer | 5.64E-01 | 5.64E-01 | 1.61E-01 | 1.61E-01 | 6.45E-01 | 6.43E-01 | 38,087 |
| SINO (64×64) | Coarse DNS | ||||||
|---|---|---|---|---|---|---|---|
| Benchmark | SINO-2 | SINO-8 | SINO-16 | SINO-32 | 64×64 | 128×128 | 256×256 |
| Decaying NS | 5.99E-02 | 4.98E-02 | 4.49E-02 | 4.08E-02 | 4.04E-01 | 1.88E-01 | 5.64E-02 |
| Forcing NS (Re=1000) | 9.17E-02 | 6.89E-02 | 6.67E-02 | 6.06E-02 | 5.07E-01 | 2.58E-01 | 6.47E-02 |
Neural Operators
Neural operators learn mappings between functions, enabling fast surrogates for PDEs and other continuous physical systems.
See all 313 papers