Our new X account is live! Follow @wizwand_team for updates
WorkDL logo mark

Hierarchical Implicit Neural Emulators

About

Neural PDE solvers offer a powerful tool for modeling complex dynamical systems, but often struggle with error accumulation over long time horizons and maintaining stability and physical consistency. We introduce a multiscale implicit neural emulator that enhances long-term prediction accuracy by conditioning on a hierarchy of lower-dimensional future state representations. Drawing inspiration from the stability properties of numerical implicit time-stepping methods, our approach leverages predictions several steps ahead in time at increasing compression rates for next-timestep refinements. By actively adjusting the temporal downsampling ratios, our design enables the model to capture dynamics across multiple granularities and enforce long-range temporal coherence. Experiments on turbulent fluid dynamics show that our method achieves high short-term accuracy and produces long-term stable forecasts, significantly outperforming autoregressive baselines while adding minimal computational overhead.

Ruoxi Jiang, Xiao Zhang, Karan Jakhar, Peter Y. Lu, Pedram Hassanzadeh, Michael Maire, Rebecca Willett• 2025

Related benchmarks

TaskDatasetResultRank
Autoregressive Rollout ForecastingL96 (test)
RMSE0.0712
15
Long-horizon forecastingKS (test)
∆RMSE (%)34.7
3
Long-horizon forecastingL96 (test)
Delta RMSE (%)-5.6
3
Showing 3 of 3 rows

Other info

Follow for update