English

Stop using root-mean-square error as a precipitation target!

Atmospheric and Oceanic Physics 2025-09-11 v1

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 V(μ)=μpV(\mu)=\mu^p. For 1<p<21<p<2 it yields a compound Poisson--Gamma distribution with a point mass at zero and a continuous density for y>0y>0, matching observed precipitation characteristics. We (i) estimate pp from the variance--mean power law and show that precipitation across temporal aggregations is far from Gaussian, with the Tweedie power pp 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 (0.60\sim0.60 vs 0.500.50 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, 2\sim2% at t+1t{+}1 growing to 16\sim16% at t+4t{+}4). Because the Tweedie deviance is continuous in pp, 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

R2 v1 2026-07-01T05:29:41.800Z