# NeuralOM: Neural Ocean Model for Subseasonal-to-Seasonal Simulation

arXiv: https://arxiv.org/abs/2505.21020 · HTML: https://mlsanity.com/paper/2505.21020/

## Key facts

- Cited by 12 papers in the map (data through 14 Oct 2026).
- Builds on 18 of its 64 references that are in the map.
- First posted on arXiv 27 May 2025.

## Builds on (most cited first, 18 of 18)

- [Swin Transformer: Hierarchical Vision Transformer using Shifted Windows](https://mlsanity.com/paper/2103.14030/) (2021)
- [Masked Autoencoders Are Scalable Vision Learners](https://mlsanity.com/paper/2111.06377/) (2021)
- [Fourier Neural Operator for Parametric Partial Differential Equations](https://mlsanity.com/paper/2010.08895/) (2020)
- [Learning Mesh-Based Simulation with Graph Networks](https://mlsanity.com/paper/2010.03409/) (2020)
- [FourCastNet: A Global Data-driven High-resolution Weather Model using Adaptive Fourier Neural Operators](https://mlsanity.com/paper/2202.11214/) (2022)
- [FuXi: A cascade machine learning forecasting system for 15-day global weather forecast](https://mlsanity.com/paper/2306.12873/) (2023)
- [FengWu: Pushing the Skillful Global Medium-range Weather Forecast beyond 10 Days Lead](https://mlsanity.com/paper/2304.02948/) (2023)
- [Scaling transformer neural networks for skillful and reliable medium-range weather forecasting](https://mlsanity.com/paper/2312.03876/) (2023)
- [PastNet: Introducing Physical Inductive Biases for Spatio-temporal Video Prediction](https://mlsanity.com/paper/2305.11421/) (2023)
- [OneForecast: A Universal Framework for Global and Regional Weather Forecasting](https://mlsanity.com/paper/2502.00338/) (2025)
- [AI-GOMS: Large AI-Driven Global Ocean Modeling System](https://mlsanity.com/paper/2308.03152/) (2023)
- [Earthfarseer: Versatile Spatio-Temporal Dynamical Systems Modeling in One Model](https://mlsanity.com/paper/2312.08403/) (2023)
- [Data-driven Global Ocean Modeling for Seasonal to Decadal Prediction](https://mlsanity.com/paper/2405.15412/) (2024)
- [CirT: Global Subseasonal-to-Seasonal Forecasting with Geometry-inspired Transformer](https://mlsanity.com/paper/2502.19750/) (2025)
- [Modeling Spatio-temporal Dynamical Systems with Neural Discrete Learning and Levels-of-Experts](https://mlsanity.com/paper/2402.05970/) (2024)
- [Turb-L1: Achieving Long-term Turbulence Tracing By Tackling Spectral Bias](https://mlsanity.com/paper/2505.19038/) (2025)
- [LangYa: Revolutionizing Cross-Spatiotemporal Ocean Forecasting](https://mlsanity.com/paper/2412.18097/) (2024)
- [ResoNet: Robust and Explainable ENSO Forecasts with Hybrid Convolution and Transformer Networks](https://mlsanity.com/paper/2312.10429/) (2023)

## Built on by (most cited first, 12 of 12)

- [Frequency-Aligned Knowledge Distillation for Lightweight Spatiotemporal Forecasting](https://mlsanity.com/paper/2507.02939/) (2025)
- [Differential-Integral Neural Operator for Long-Term Turbulence Forecasting](https://mlsanity.com/paper/2509.21196/) (2025)
- [Spatiotemporal Forecasting as Planning: A Model-Based Reinforcement Learning Approach with Generative World Models](https://mlsanity.com/paper/2510.04020/) (2025)
- [PnP-Corrector: A Universal Correction Framework for Coupled Spatiotemporal Forecasting](https://mlsanity.com/paper/2605.08935/) (2026)
- [OceanGym: A Benchmark Environment for Underwater Embodied Agents](https://mlsanity.com/paper/2509.26536/) (2025)
- [Advanced Global Wildfire Activity Modeling with Hierarchical Graph ODE](https://mlsanity.com/paper/2601.01501/) (2026)
- [VISION: Prompting Ocean Vertical Velocity Reconstruction from Incomplete Observations](https://mlsanity.com/paper/2509.21477/) (2025)
- [Unlocking Out-of-Distribution Generalization in Dynamics through Physics-Guided Augmentation](https://mlsanity.com/paper/2510.24216/) (2025)
- [Symmetric Composition of Anisotropic Operators for Global Subseasonal-to-Seasonal Climate Forecasting](https://mlsanity.com/paper/2602.15040/) (2026)
- [OceanDepths: A Global Dataset of Paired Subsurface and Surface Ocean Observations](https://mlsanity.com/paper/2608.16373/) (2026)
- [Decision-Oriented Uncertainty Quantification for Risk Control in Earth System Spatiotemporal Foundation Models](https://mlsanity.com/paper/2609.14821/) (2026)
- [OceanFlow: Coherent Flow Matching for Probabilistic Ocean Modeling](https://mlsanity.com/paper/2610.15518/) (2026)

## Abstract (arXiv)

Long-term, high-fidelity simulation of slow-changing physical systems, such as the ocean and climate, presents a fundamental challenge in scientific computing. Traditional autoregressive machine learning models often fail in these tasks as minor errors accumulate and lead to rapid forecast degradation. To address this problem, we propose NeuralOM, a general neural operator framework designed for simulating complex, slow-changing dynamics. NeuralOM's core consists of two key innovations: (1) a Progressive Residual Correction Framework that decomposes the forecasting task into a series of fine-grained refinement steps, effectively suppressing long-term error accumulation; and (2) a Physics-Guided Graph Network whose built-in adaptive messaging mechanism explicitly models multi-scale physical interactions, such as gradient-driven flows and multiplicative couplings, thereby enhancing physical consistency while maintaining computational efficiency. We validate NeuralOM on the challenging task of global Subseasonal-to-Seasonal (S2S) ocean simulation. Extensive experiments demonstrate that NeuralOM not only surpasses state-of-the-art models in forecast accuracy and long-term stability, but also excels in simulating extreme events. For instance, at a 60-day lead time, NeuralOM achieves a 13.3% lower RMSE compared to the best-performing baseline, offering a stable, efficient, and physically-aware paradigm for data-driven scientific computing. Code link: https://github.com/YuanGao-YG/NeuralOM.
