Table of Contents
Fetching ...

Ensemble-size-dependence of deep-learning post-processing methods that minimize an (un)fair score: motivating examples and a proof-of-concept solution

Christopher David Roberts

TL;DR

This work analyzes how distribution-aware ensemble post-processing that induces inter-member dependencies can undermine the fairness of the adjusted CRPS ($aCRPS$) when ensembles are finite. It first demonstrates, via a Gaussian signal-plus-noise idealization, that linear calibrations exploiting ensemble dependencies can yield apparent improvements in $aCRPS$ at the cost of reliability, highlighting the non-fairness of $aCRPS$ under dependency. The authors then compare two transformer-based post-processing approaches: the PoET ensemble transformer, which shares information across ensemble members and can improve $aCRPS$ but introduces ensemble-size sensitivity and over-dispersion; and the trajectory transformer, which applies attention across lead times per member to preserve conditional independence and achieve ensemble-size independence. Applied to ECMWF subseasonal weekly-mean $T_{2m}$ forecasts, the trajectory transformer reduces systematic biases while maintaining or improving forecast reliability across different training and real-time ensemble sizes, offering a robust proof-of-concept pathway for fair, ensemble-size–agnostic post-processing in practice.

Abstract

Fair scores reward ensemble forecast members that behave like samples from the same distribution as the verifying observations. They are therefore an attractive choice as loss functions to train data-driven ensemble forecasts or post-processing methods when large training ensembles are either unavailable or computationally prohibitive. The adjusted continuous ranked probability score (aCRPS) is fair and unbiased with respect to ensemble size, provided forecast members are exchangeable and interpretable as conditionally independent draws from an underlying predictive distribution. However, distribution-aware post-processing methods that introduce structural dependency between members can violate this assumption, rendering aCRPS unfair. We demonstrate this effect using two approaches designed to minimize the expected aCRPS of a finite ensemble: (1) a linear member-by-member calibration, which couples members through a common dependency on the sample ensemble mean, and (2) a deep-learning method, which couples members via transformer self-attention across the ensemble dimension. In both cases, the results are sensitive to ensemble size and apparent gains in aCRPS can correspond to systematic unreliability characterized by over-dispersion. We introduce trajectory transformers as a proof-of-concept that ensemble-size independence can be achieved. This approach is an adaptation of the Post-processing Ensembles with Transformers (PoET) framework and applies self-attention over lead time while preserving the conditional independence required by aCRPS. When applied to weekly mean $T_{2m}$ forecasts from the ECMWF subseasonal forecasting system, this approach successfully reduces systematic model biases whilst also improving or maintaining forecast reliability regardless of the ensemble size used in training (3 vs 9 members) or real-time forecasts (9 vs 100 members).

Ensemble-size-dependence of deep-learning post-processing methods that minimize an (un)fair score: motivating examples and a proof-of-concept solution

TL;DR

This work analyzes how distribution-aware ensemble post-processing that induces inter-member dependencies can undermine the fairness of the adjusted CRPS () when ensembles are finite. It first demonstrates, via a Gaussian signal-plus-noise idealization, that linear calibrations exploiting ensemble dependencies can yield apparent improvements in at the cost of reliability, highlighting the non-fairness of under dependency. The authors then compare two transformer-based post-processing approaches: the PoET ensemble transformer, which shares information across ensemble members and can improve but introduces ensemble-size sensitivity and over-dispersion; and the trajectory transformer, which applies attention across lead times per member to preserve conditional independence and achieve ensemble-size independence. Applied to ECMWF subseasonal weekly-mean forecasts, the trajectory transformer reduces systematic biases while maintaining or improving forecast reliability across different training and real-time ensemble sizes, offering a robust proof-of-concept pathway for fair, ensemble-size–agnostic post-processing in practice.

Abstract

Fair scores reward ensemble forecast members that behave like samples from the same distribution as the verifying observations. They are therefore an attractive choice as loss functions to train data-driven ensemble forecasts or post-processing methods when large training ensembles are either unavailable or computationally prohibitive. The adjusted continuous ranked probability score (aCRPS) is fair and unbiased with respect to ensemble size, provided forecast members are exchangeable and interpretable as conditionally independent draws from an underlying predictive distribution. However, distribution-aware post-processing methods that introduce structural dependency between members can violate this assumption, rendering aCRPS unfair. We demonstrate this effect using two approaches designed to minimize the expected aCRPS of a finite ensemble: (1) a linear member-by-member calibration, which couples members through a common dependency on the sample ensemble mean, and (2) a deep-learning method, which couples members via transformer self-attention across the ensemble dimension. In both cases, the results are sensitive to ensemble size and apparent gains in aCRPS can correspond to systematic unreliability characterized by over-dispersion. We introduce trajectory transformers as a proof-of-concept that ensemble-size independence can be achieved. This approach is an adaptation of the Post-processing Ensembles with Transformers (PoET) framework and applies self-attention over lead time while preserving the conditional independence required by aCRPS. When applied to weekly mean forecasts from the ECMWF subseasonal forecasting system, this approach successfully reduces systematic model biases whilst also improving or maintaining forecast reliability regardless of the ensemble size used in training (3 vs 9 members) or real-time forecasts (9 vs 100 members).
Paper Structure (18 sections, 45 equations, 7 figures, 2 tables)

This paper contains 18 sections, 45 equations, 7 figures, 2 tables.

Figures (7)

  • Figure 1: (Top) Schematic diagram of the PoET hierarchical ensemble transformer architecture as configured during training bouallegue2024improving. (Bottom) Schematic diagram of the hierarchical trajectory transformer described in this study, as configured during training. Figure generated using PlotNeuralNet iqbal_plotneuralnet_2018.
  • Figure 2: Optimal linear calibration parameters for idealized Gaussian forecasts following equations \ref{['eq:optimal_b_maintext']} and \ref{['eq:optimal_c_maintext']} for a 10-member perfectly reliable ($\alpha = \beta$) forecast and different values of $\sigma_s$.
  • Figure 3: (a) Weighted global mean aCRPS of raw and post-processed T$_{2m}$ forecasts in normalized units as a function of training epoch for the independent evaluation period (2021-2023). Each epoch represents a single pass through the entire training dataset (1959-2017). (b-c) As above, but for unbiased estimates of the ensemble spread and root mean square error (RMSE) of the ensemble mean calculated to be unbiased with ensemble size following leutbecher2008ensemble and roberts2025unbiased. All metrics are averaged over lead times with equal weight.
  • Figure 4: Impact of PoET hierarchical Ensemble Transformer post-processing method bouallegue2024improving on the adjusted CRPS of weekly mean T$_{2m}$ forecasts as a function of forecast lead time, where negative values are indicative of lower aCRPS in the post-processed reforecasts. The ensemble transformer post-processing method was trained on 9-members (see figure \ref{['fig:training_metrics']}) for the period 1959-2017 and evaluated on either 9-member (left) or 100-member (right) forecasts for the period 2018-2023. Stippling indicates regions where 95 % confidence intervals for $\Delta$aCRPS do not include zero. Confidence intervals are estimated as the 2.5 and 97.5 percentiles of an empirical distribution derived by block bootstrap resampling (with replacement) 500 times. Start dates within the same calendar month are treated as a single block during resampling to preserve temporal correlation structure.
  • Figure 5: As figure \ref{['fig:enstrans_fcrps_maps']}, but for the hierarchical Trajectory Transformer approach described in this study.
  • ...and 2 more figures