mlsanityData to 14 Oct 2026

PnP-Corrector: A Universal Correction Framework for Coupled Spatiotemporal Forecasting

arXiv 2605.08935 ↗First posted 9 May 2026cs.AI cs.LGcited by 2

Not on the map. Why?

Abstract (arXiv)

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 28% and surpasses state-of-the-art models on several key metrics.

Key facts. Cited by 2 papers in the map (data through 14 Oct 2026). Builds on 38 of its 59 references that are in the map. First posted on arXiv 9 May 2026.

Builds on

38 of 59 references.

References · 38

  1. 2017Attention Is All You Needcited by 60,361
  2. 2020An Image is Worth 16x16 Words: Transformers for Image Recognition at Scalecited by 30,186
  3. 2015U-Net: Convolutional Networks for Biomedical Image Segmentationcited by 19,713
  4. 2021High-Resolution Image Synthesis with Latent Diffusion Modelscited by 18,863
  5. 2022Generative Adversarial Networkscited by 5,627
  6. 2020Taming Transformers for High-Resolution Image Synthesiscited by 3,253
  7. 2015Convolutional LSTM Network: A Machine Learning Approach for Precipitation Nowcastingcited by 1,436
  8. 2020Learning Mesh-Based Simulation with Graph Networkscited by 868
  9. 2021Physics-informed neural networks (PINNs) for fluid mechanics: A reviewcited by 473
  10. 2023FuXi: A cascade machine learning forecasting system for 15-day global weather forecastcited by 241
  11. 2023Neural General Circulation Models for Weather and Climatecited by 239
  12. 2023PDE-Refiner: Achieving Accurate Long Rollouts with Neural PDE Solverscited by 216
26 more
  1. 2022FourCastNet: Accelerating Global High-Resolution Weather Forecasting using Adaptive Fourier Neural Operatorscited by 188
  2. 2022SimVP: Simpler yet Better Video Predictioncited by 184
  3. 2021PredRNN: A Recurrent Neural Network for Spatiotemporal Predictive Learningcited by 182
  4. 2022Earthformer: Exploring Space-Time Transformers for Earth System Forecastingcited by 152
  5. 2022MultiScale MeshGraphNetscited by 98
  6. 2022Predicting Physics in Mesh-reduced Space with Temporal Attentioncited by 84
  7. 2021Physics-Guided Deep Learning for Dynamical Systems: A Surveycited by 44
  8. 2023PastNet: Introducing Physical Inductive Biases for Spatio-temporal Video Predictioncited by 44
  9. 2020Generative Adversarial Networks for Spatio-temporal Data: A Surveycited by 35
  10. 2025OneForecast: A Universal Framework for Global and Regional Weather Forecastingcited by 34
  11. 2022Physically Constrained Generative Adversarial Networks for Improving Precipitation Fields from Earth System Modelscited by 29
  12. 2022Contact-aware Human Motion Forecastingcited by 27
  13. 2023AI-GOMS: Large AI-Driven Global Ocean Modeling Systemcited by 27
  14. 2018Multi-Step Prediction of Dynamic Systems with Recurrent Neural Networkscited by 25
  15. 2022SimVPv2: Towards Simple yet Powerful Spatiotemporal Predictive Learningcited by 22
  16. 2024Coupled Ocean-Atmosphere Dynamics in a Machine Learning Earth System Modelcited by 21
  17. 2021Simple Baseline for Single Human Motion Forecastingcited by 18
  18. 2023Earthfarseer: Versatile Spatio-Temporal Dynamical Systems Modeling in One Modelcited by 17
  19. 2025CirT: Global Subseasonal-to-Seasonal Forecasting with Geometry-inspired Transformercited by 14
  20. 2025NeuralOM: Neural Ocean Model for Subseasonal-to-Seasonal Simulationcited by 12
  21. 2025Turb-L1: Achieving Long-term Turbulence Tracing By Tackling Spectral Biascited by 9
  22. 2025Differential-Integral Neural Operator for Long-Term Turbulence Forecastingcited by 4
  23. 2025Spatiotemporal Forecasting as Planning: A Model-Based Reinforcement Learning Approach with Generative World Modelscited by 4
  24. 2026OMNIFLOW: A Physics-Grounded Multimodal Agent for Generalized Scientific Reasoningcited by 2
  25. 2025Learning from History: A Retrieval-Augmented Framework for Spatiotemporal Predictioncited by 1
  26. 2025NeuralOGCM: Differentiable Ocean Modeling with Learnable Physicscited by 1

Built on by

Cited by year (2026 to 14 Oct)

Citing papers per year: 2026: 2.

2 papers.