Bridging Spectral Operator Learning and U-Net Hierarchies: SpectraNet for Stable Autoregressive PDE Surrogates
About
Neural operators for time-dependent PDEs face a structural tension: spectral architectures (FNO and descendants) inherit exponential rollout-error growth from their one-step Lipschitz constant, while hierarchical U-Net operators trade resolution invariance for multi-scale detail. We introduce SpectraNet, an autoregressive neural operator that composes truncated spectral convolutions inside a U-Net hierarchy with a Residual-Target Spectral Block trained under a Semigroup-Consistency Loss. The residual-target parametrization replaces L^T stability blow-up with linear T*delta drift, and the spectral path's parameter count is Theta(L w^2 M^2), independent of grid N. Under a single unified protocol against 16 published neural-operator baselines on Navier-Stokes nu=1e-5 at 64x64, SpectraNet reaches test relative L2 = 0.0822 at 2.04M parameters -- 2.33x fewer than canonical FNO at ~20% lower error -- and wins five of six rows in a cross-PDE comparison against FNO (NS at nu in {1e-4, 1e-3}, PDEBench Shallow-Water 2D and Diffusion-Reaction, with the Active-Matter row going to FNO inside its seed spread). Trained from scratch at native 128^2 under the same protocol, SpectraNet improves to 0.0724 while FNO regresses to 0.3080. Free rollout stays bounded for T=100 where FNO diverges across all 200 test trajectories. On consumer CPU at B=1, SpectraNet runs sub-200ms while the full-attention Transformer that wins raw L2 pays ~60x latency; we do not claim to beat that Transformer on raw L2, only to dominate the lightweight (<=5M parameter, sub-200ms CPU) Pareto frontier. Source code: https://github.com/Enrikkk/spectranet
Related benchmarks
| Task | Dataset | Result | Rank | |
|---|---|---|---|---|
| Fluid Dynamics Simulation | Navier-Stokes (NS) nu=10^-5 at 64x64 unified-protocol (test) | Relative L2 Error (Test)7.87 | 31 | |
| Trajectory Prediction | Navier-Stokes ν = 10⁻⁵, turbulent | Relative L2 Error0.0822 | 2 | |
| Trajectory Prediction | Navier-Stokes ν = 10⁻³, smooth | Relative L2 Error0.0011 | 2 | |
| Trajectory Prediction | Navier-Stokes ν = 10⁻⁴ intermediate | Relative L2 Error1.521 | 2 | |
| Trajectory Prediction | PDEBench Shallow Water (2D) | Relative L2 Error0.0012 | 2 | |
| Trajectory Prediction | PDEBench Diffusion-Reaction | Relative L2 Error2.01 | 2 | |
| Trajectory Prediction | The Well Active-Matter | Relative L2 Error0.0017 | 2 |