Fixing the Double Penalty in Data-Driven Weather Forecasting Through a Modified Spherical Harmonic Loss Function
Abstract
Recent advancements in data-driven weather forecasting models have delivered deterministic models that outperform the leading operational forecast systems based on traditional, physics-based models. However, these data-driven models are typically trained with a mean squared error loss function, which causes smoothing of fine scales through a “double penalty” effect. We develop a simple, parameter-free modification to this loss function that avoids this problem by separating the loss attributable to decorrelation from the loss attributable to spectral amplitude errors. Fine-tuning the GraphCast model with this new loss function results in sharp deterministic weather forecasts, an increase of the model’s effective resolution from 1,250km to 160km, improvements to ensemble spread, and improvements to predictions of tropical cyclone strength and surface wind extremes.
Cite
Text
Subich et al. "Fixing the Double Penalty in Data-Driven Weather Forecasting Through a Modified Spherical Harmonic Loss Function." Proceedings of the 42nd International Conference on Machine Learning, 2025.Markdown
[Subich et al. "Fixing the Double Penalty in Data-Driven Weather Forecasting Through a Modified Spherical Harmonic Loss Function." Proceedings of the 42nd International Conference on Machine Learning, 2025.](https://mlanthology.org/icml/2025/subich2025icml-fixing/)BibTeX
@inproceedings{subich2025icml-fixing,
title = {{Fixing the Double Penalty in Data-Driven Weather Forecasting Through a Modified Spherical Harmonic Loss Function}},
author = {Subich, Christopher and Husain, Syed Zahid and Separovic, Leo and Yang, Jing},
booktitle = {Proceedings of the 42nd International Conference on Machine Learning},
year = {2025},
pages = {57191-57211},
volume = {267},
url = {https://mlanthology.org/icml/2025/subich2025icml-fixing/}
}