OptFormer: Optical Flow-Guided Attention and Phase Space Reconstruction for SST Forecasting
Yin Wang, Chunlin Gong, Zhuozhen Xu, Lehan Zhang, Xiang Wu
TL;DR
OptFormer tackles SST forecasting by unifying phase-space reconstruction with optical-flow-guided attention in an encoder–decoder framework, enabling data-efficient learning of nonlinear spatiotemporal dynamics. By embedding delayed attractors via Takens-like phase-space reconstruction and guiding attention with dense optical flow, the model concentrates on dynamically evolving regions and captures long-range dependencies. Empirical results on NOAA OISST data show significant improvements in RMSE and MAPE across spatial scales, horizons, and seasons, with ablations confirming the contributions of optical attention, multi-scale feature encoding, and temporal auto-correlation modeling. The work offers a practical, GPU-friendly approach for ocean monitoring and climate forecasting, while outlining future directions toward multivariate inputs and PDE-aware modeling.
Abstract
Sea Surface Temperature (SST) prediction plays a vital role in climate modeling and disaster forecasting. However, it remains challenging due to its nonlinear spatiotemporal dynamics and extended prediction horizons. To address this, we propose OptFormer, a novel encoder-decoder model that integrates phase-space reconstruction with a motion-aware attention mechanism guided by optical flow. Unlike conventional attention, our approach leverages inter-frame motion cues to highlight relative changes in the spatial field, allowing the model to focus on dynamic regions and capture long-range temporal dependencies more effectively. Experiments on NOAA SST datasets across multiple spatial scales demonstrate that OptFormer achieves superior performance under a 1:1 training-to-prediction setting, significantly outperforming existing baselines in accuracy and robustness.
