Share your thoughts, 1 month free Claude Pro on usSee more
WorkDL logo mark

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

Enrique Hern\'andez Noguera, Md Meftahul Ferdaus, Elias Ioup, Mahdi Abdelguerfi, Julian Simeonov• 2026

Related benchmarks

TaskDatasetResultRank
Fluid Dynamics SimulationNavier-Stokes (NS) nu=10^-5 at 64x64 unified-protocol (test)
Relative L2 Error (Test)7.87
31
Trajectory PredictionNavier-Stokes ν = 10⁻⁵, turbulent
Relative L2 Error0.0822
2
Trajectory PredictionNavier-Stokes ν = 10⁻³, smooth
Relative L2 Error0.0011
2
Trajectory PredictionNavier-Stokes ν = 10⁻⁴ intermediate
Relative L2 Error1.521
2
Trajectory PredictionPDEBench Shallow Water (2D)
Relative L2 Error0.0012
2
Trajectory PredictionPDEBench Diffusion-Reaction
Relative L2 Error2.01
2
Trajectory PredictionThe Well Active-Matter
Relative L2 Error0.0017
2
Showing 7 of 7 rows

Other info

Follow for update