Learning to Advect: A Neural Semi-Lagrangian Architecture for Weather Forecasting
Abstract
Recent machine-learning approaches to weather forecasting often employ a monolithic architecture in which distinct physical mechanisms-advection (long-range transport), diffusion-like mixing, thermodynamic processes, and forcing-are represented implicitly within a single large network. This is particularly problematic for advection, where long-range transport typically requires expensive global interaction mechanisms or deep stacks of local convolutional layers. To mitigate this, we present PARADIS, a physics-inspired global weather prediction model that enforces inductive biases on network behavior through a functional decomposition into advection, diffusion, and reaction blocks acting on latent variables. We implement advection through a Neural Semi-Lagrangian operator that performs trajectory-based transport via differentiable interpolation on the sphere, enabling end-to-end learning of both the latent modes to be transported and their characteristic trajectories. Diffusion-like processes are modeled by depthwise-separable spatial mixing, whereas local source terms and vertical interactions are handled via pointwise channel interactions, yielding a physically structured operator decomposition. Evaluated on ERA5 benchmarks, PARADIS achieves competitive deterministic forecast skill, with particularly strong short-lead performance, while preserving substantially better spectral fidelity and forecast activity during medium-range rollouts.
Cite
@article{arxiv.2601.21151,
title = {Learning to Advect: A Neural Semi-Lagrangian Architecture for Weather Forecasting},
author = {Carlos A. Pereira and Stéphane Gaudreault and Valentin Dallerit and Christopher Subich and Shoyon Panday and Siqi Wei and Sasa Zhang and Siddharth Rout and Eldad Haber and Raymond J. Spiteri and David Millard and Emilia Diaconescu},
journal= {arXiv preprint arXiv:2601.21151},
year = {2026}
}