PnP-Corrector: A Universal Correction Framework for Coupled Spatiotemporal Forecasting
For researchers in climate modeling and coupled dynamical systems, this framework addresses the critical bottleneck of error accumulation in long-range predictions.
Coupled spatiotemporal forecasting suffers from compounding errors due to Reciprocal Error Amplification. The proposed PnP-Corrector framework, using DSLCast, reduces prediction error by 29% in a 300-day global ocean-atmosphere coupled forecast, surpassing state-of-the-art models.
Coupled spatiotemporal forecasting is important for predicting the future evolution of multiple interacting dynamical systems, such as in climate models. However, existing methods are severely constrained by the persistent bottleneck of compounding errors. In coupled systems, errors from each subsystem simulator propagate and amplify one another, a phenomenon we term Reciprocal Error Amplification, leading to a rapid collapse of long-range predictions. To address this challenge, we propose a universal framework called PnP-Corrector (Plug-and-Play Corrector). The core idea of our framework is to decouple the physical simulation from the error correction process: it freezes pre-trained physics simulation engines and exclusively trains a correction agent to proactively counteract the systematic biases emerging from the coupled system. Furthermore, we design an efficient predictive model architecture, DSLCast, to serve as the backbone of this framework. Extensive experiments demonstrate that our method significantly enhances the long-term stability and accuracy of coupled forecasting systems. For instance, in the challenging task of a 300-day global ocean-atmosphere coupled forecast, our PnP-Corrector framework reduces the prediction error of the baseline model by 29% and surpasses state-of-the-art models on several key metrics.