Stop using root-mean-square error as a precipitation target!
Abstract
Root-mean-square error (RMSE) remains the default training loss for data-driven precipitation models, despite precipitation being semi-continuous, zero-inflated, strictly non-negative, and heavy-tailed. This Gaussian-implied objective misspecifies the data-generating process because it tolerates negative predictions, underpenalises rare heavy events, and ignores the mass at zero. We propose replacing RMSE with the Tweedie deviance, a likelihood-based and differentiable loss from the exponential--dispersion family with variance function . For it yields a compound Poisson--Gamma distribution with a point mass at zero and a continuous density for , matching observed precipitation characteristics. We (i) estimate from the variance--mean power law and show that precipitation across temporal aggregations is far from Gaussian, with the Tweedie power increasing with accumulation length towards a Gamma limit; and (ii) demonstrate consistent skill gains when training deep data-driven models with Tweedie deviance in place of RMSE. In diffusion-model downscaling over Beijing, Tweedie loss improves wet-pixel MAE and extreme recall ( vs at the 99th percentile). In ConvLSTM nowcasting over Kolkata, Tweedie loss yields improved wet-pixel MAE and dry-pixel hit rates, with improvements that compound autoregressively with lead time (for MAE, at growing to at ). Because the Tweedie deviance is continuous in , it adapts smoothly across scales, offering a statistically justified, practical replacement for RMSE in precipitation-based learning tasks.
Keywords
Cite
@article{arxiv.2509.08369,
title = {Stop using root-mean-square error as a precipitation target!},
author = {Kieran M. R. Hunt},
journal= {arXiv preprint arXiv:2509.08369},
year = {2025}
}
Comments
Submitted to AIES