In this paper
AbstractIntroductionMethodExperimentsConclusionAcknowledgments and Disclosure of FundingUse of LLMs.ReferencesPosterior under a Gaussian likelihood: uniform vs. Gaussian priorUniform priorGaussian priorConditioning on the balanceThe diffusion priorParameterizationEM trainingGuided posterior samplingExperimental detailsAdditional resultsPer-component accuracyPrior fidelityCalibrationRobustness to noise misspecificationWorksEngineering
Reconciling European Electricity Data with Diffusion Priors Learned from Observations Alone
Abstract
The ENTSO-E Transparency Platform publishes generation, load, and cross-border power flows for European bidding zones. The values carry undocumented errors and violate energy conservation, and data reconciliation consists in correcting them to satisfy that balance. In this work, we formulate data reconciliation as Bayesian inference. Classical methods implicitly take a uniform prior, which we replace with a diffusion model estimated from the observations alone by empirical-Bayes expectation-maximization. The resulting posterior is sampled with a guided process whose projection step makes every sample satisfy the balance exactly. On synthetic systems of six to thirty-five bidding zones, built from ENTSO-E resource-adequacy data, the diffusion prior gives the most accurate reconciliation, increasingly so as the system grows. Its credible intervals are calibrated or conservative, and it remains the most accurate under a misspecified noise covariance. Finally, when the generation of entire zones goes unreported, the classical methods fail while the learned prior recovers it from cross-zone structure.
Introduction
Monitoring, planning, and operating electricity systems requires accurate data on generation, load, and cross-border power flows, a need the European Union answered by requiring their publication through the ENTSO-E Transparency Platform, established in 2015 under Regulation (EU) No 543/2013 [1]. The platform reports for every bidding zone the generation by fuel type, the electrical load, and the physical flow on each border. The values are communicated by the transmission system operators (TSOs), who meter large units and interconnections, while the load is derived from generation and exchanges [1]. The quality of the values varies across TSOs and reported quantities [2, 3, 4], and no TSO documents the uncertainty of the submitted data. In practice, we observe that, at any hour, the reported generation and imports of a zone do not balance its reported load and exports, with residuals sometimes reaching thousands of megawatts. We call these reported values observations. The state is the same quantities free of reporting errors and satisfying energy conservation.
Recovering the state from the observations is a data reconciliation problem, developed in process engineering for material and energy balances [5]. It differs from power system state estimation, which recovers the bus voltages and angles from grid measurements, where the physics enters through the measurement equations [6]. Here, the state and the observations are vectors of the same quantities, and the physics enters as the zonal balance constraint. Earlier work has trained neural networks to estimate the state of transmission and distribution grids [7, 8] and used diffusion models to perform data reconciliation on corrupted grid measurements [9] and dynamic state estimation [10], but none of it addresses bidding-zone observations.
Let us consider discrete times , over which the state and the observations evolve. We represent the bidding zones and their interconnections as a directed graph , where vertices are the zones and the edges are the interconnections, each contributing the two edges and , one per flow direction. Each zone hosts a set of generation sources . At time , zone produces from each source , consumes the load , and exchanges the flow on each edge . Together these quantities compose the state , with , and we call its components signals. Energy conservation, i.e., generation plus imports equal load plus exports in each zone, is expressed by the linear constraints at each time, with independent of . Generation and flows are non-negative, while load can be negative, because the reported load is consumption minus local generation, and these constraints make the feasible states a polyhedral cone . An observation is a noisy view of the state and may violate the energy balance.
The goal is to infer, at each time , the state that generated the observations, together with the remaining uncertainty. We formulate reconciliation as Bayesian inference, under the latent variable model of Figure 2 in Appendix § 7, defined by a prior and a likelihood. For the likelihood, we model the observations as

Figure 2. The latent variable model for data reconciliation. White circles are latent variables, gray circles are observed ones, and the black square is a fixed parameter. The rectangular box denotes the times, modeled independently, and sits outside it, shared by all times. The mask is read from which signals the platform publishes at time , and the observation is a deterministic function of its three parents through (1).
where is a diagonal matrix whose -th diagonal entry is 1 if signal is reported at time . The observation noise is assumed to have a diagonal covariance , the same for every time , so the noise is independent across signals. Under the model, is Gaussian with mean , which defines the likelihood . Its only parameter is , which no TSO publicly documents, so it must be set from domain knowledge. The states themselves follow an unknown distribution supported on . This source distribution is the ideal prior, and the methods differ in the prior they adopt in its place. We model the states as independent and identically distributed across time, ignoring their temporal correlation, and Bayes’ theorem gives the posterior
the distribution of the state given the observation. Classical reconciliation implicitly takes the prior uniform over , and we replace it with a diffusion model estimated from the observations alone by empirical-Bayes expectation-maximization [11].
Method
A reconciliation method fixes two ingredients, the prior and the likelihood through its covariance , and gives an algorithm for evaluating or sampling the corresponding posterior.
Priors. The uniform prior over is the implicit choice of classical reconciliation. In practice, however, the state distribution is far from uniform. Generation is correlated across zones, and the system moves between operating regimes, defined by the border flow directions and the units committed, so the distribution is multimodal and concentrated. A prior capturing this concentration should narrow the posterior. A Gaussian prior is unimodal, so it captures the correlations but not the regimes. A diffusion prior, a denoising network defining a distribution over states [12, 13], can be multimodal and represent both. Its parameterization is in Appendix § 8.
Estimation. The uniform prior has no free parameters. The Gaussian and diffusion priors are estimated by empirical Bayes (EB), from historical observations and without access to states. For the Gaussian EB prior, and are estimates of the mean and covariance of the states. The states are not observed, but at times where every signal is reported, , so the model gives the mean and covariance of the states in closed form as and . Replacing both by their sample versions over those times gives and . The sample estimate of need not be positive semidefinite, so we clip its negative eigenvalues to a small positive floor. Times with unreported signals contribute nothing to this estimator. The diffusion prior should represent the state distribution, but training its denoiser requires state samples, which are unavailable, so the expectation-maximization (EM) scheme of Rozet et al. [11] estimates it from the observations. The expectation step (E-step) draws plausible states for each observation by sampling the posterior under the current prior, and the maximization step (M-step) refits the network on those samples. The iterations carry the usual EM guarantees [14, 15, 16]. Only the network is estimated, while the covariance keeps the value specified from domain knowledge and enters only the E-step posterior. We call the result the diffusion EB prior. Appendix § 8 gives the training loss, the EM hyperparameters, and the guided posterior sampler. The oracle prior, an upper bound, trains the same model on the states, which real data never provides.
Inference. Under the uniform prior, classical reconciliation returns the maximum-a-posteriori (MAP) estimate. This is the weighted least squares (WLS) estimator, which solves the quadratic program
where and stack the generation and flow signals of . The objective is the negative log-likelihood, and the constraints restrict to , the support of the prior. The WLS variants compared in § 3 differ from (3) in the covariance they use in place of the unknown and, for naive and oracle WLS, in dropping the non-negativity constraints to obtain a closed form. Under the Gaussian EB prior the posterior is Gaussian in closed form, conditioned on the balance (Appendix § 7). Under the diffusion prior it has no closed form and must be sampled. Each step of the guided denoising process [17] is pulled toward the observation through the likelihood and orthogonally projected onto the null space of as in DDNM [18], so every sample satisfies the balance exactly (Appendix § 8). The projection enforces only the equality constraint , while the non-negativity inequalities have no closed-form projection, so posterior draws can violate them. We call the posterior under the diffusion prior the diffusion posterior, and an ensemble of draws gives its credible intervals.
Experiments
Reconciliation accuracy cannot be measured on real ENTSO-E data, which lacks known states. Instead we build four synthetic systems of increasing size from the ENTSO-E European Resource Adequacy Assessment (ERAA) 2024 [19], each a set of bidding zones and their interconnections, named EU6 (6 zones, signals), EUWEST, EUWIDE, and EUFULL (35 zones, ). For each of the 35 climate years, the weather scenarios of ERAA, an hourly merit-order dispatch produces states satisfying the balance exactly, and the observations are these states plus independent Gaussian noise with per-fuel, per-zone standard deviations between 50 and 300 MW. The learned priors are trained on the observations of climate years 1-28, never on the states, and every method is evaluated on years 32-35 (35 040 hours) against the states. Appendix C details the systems, the noise model, and the de Chalendar heuristic, and Table 2 summarizes the compared methods.
| Method | Historical data | |
| Naive WLS | I | None |
| Oracle WLS | True | None |
| Constrained WLS QP | True | None |
De Chalendar (xrefEq. (14)) | From magnitudes | None |
| Gaussian EB prior | True | CY 1-28 observations |
| Diffusion EB prior [11] | True | CY 1-28 observations |
Table 2. Reconciliation methods compared.
We compare naive WLS (), oracle WLS given the true , the constrained WLS QP adding non-negativity, the heuristic of de Chalendar and Benson [20], the Gaussian EB prior, and the diffusion EB prior. RMSE against the states scores each method’s best point estimate, the posterior mean for the learned priors, and the unreconciled observations give the reference error. The posteriors are judged separately, by the coverage of their credible intervals and by the CRPS.
The diffusion EB prior is the most accurate on every system, as the left half of Table 1 shows, and the RMSE gap to the constrained WLS QP widens from 13.3% to 24.0% as the system grows. The Gaussian EB prior, which captures the correlations but not the regimes, splits this gain in two, the correlations driving the growth with system size and the regimes contributing the rest. Naive and oracle WLS improve on the raw observations in proportion to what they know of , and the de Chalendar heuristic is the only method worse than the raw observations, because its magnitude-based weights mismatch our per-fuel noise and its under-weighted loads absorb the balance violations. Table 3 shows that the advantage of the diffusion EB prior holds on every component.
| Method | RMSE, fully observed | CRPS, fully observed | RMSE on masked signals | |||||||||
| EU6 | EUWEST | EUWIDE | EUFULL | EU6 | EUWEST | EUWIDE | EUFULL | EU6 | EUWEST | EUWIDE | EUFULL | |
| Raw observations | 146.4 | 142.4 | 138.2 | 140.8 | – | – | – | – | – | – | – | – |
| Naive WLS | 141.9 | 138.0 | 134.0 | 136.5 | – | – | – | – | – | – | – | – |
| Oracle WLS | 134.9 | 130.1 | 126.9 | 129.3 | 75.4 | 79.3 | 84.8 | 84.7 | – | – | – | – |
| Constrained WLS QP | 115.1 | 110.0 | 104.0 | 103.9 | – | – | – | – | 5 197 | 5 311 | 5 580 | 5 615 |
| De Chalendar [20] | 287.3 | 232.7 | 197.9 | 194.2 | – | – | – | – | – | – | – | – |
| Gaussian EB prior (ours) | 117.6 | 106.1 | 90.6 | 87.6 | 61.6 | 60.3 | 54.6 | 51.1 | 2 050 | 1 647 | 1 401 | 1 255 |
| Diffusion EB prior (ours) | 99.8 | 92.0 | 80.5 | 79.0 | 52.0 | 52.7 | 51.7 | 50.1 | 1 920 | 1 411 | 1 149 | 1 035 |
Table 1. Overall RMSE (MW) and CRPS (MW) of the three posteriors on the test set, and RMSE (MW) on the masked signals when the generation of the six core zones is masked.
| Method | Gen | Load | Flow | Overall |
| Raw observations | 131.7 | 151.5 | 200.0 | 146.4 |
| Naive WLS | 129.5 | 147.5 | 187.9 | 141.9 |
| Oracle WLS | 120.3 | 147.4 | 185.6 | 134.9 |
| Constrained WLS QP | 103.7 | 149.7 | 147.9 | 115.1 |
| De Chalendar | 237.5 | 729.9 | 233.5 | 287.3 |
| Gaussian EB prior | 106.2 | 141.0 | 153.8 | 117.6 |
| Diffusion EB prior | 92.5 | 135.0 | 117.5 | 99.8 |
Table 3. RMSE (MW) on the six-zone system, all test hours. The generation, load, and flow columns report the per-component RMSE, and the overall column is the all-signal RMSE.
A posterior is useful only if its credible intervals can be trusted. It is calibrated when the intervals contain the state as often as they claim. The diffusion posterior is calibrated on the two smaller systems, at worst 88.6% coverage for a 90% target, and conservative on the larger ones, where the intervals contain the state more often than they claim, as shown in Figure 3. The continuous ranked probability score (CRPS), the standard score for probabilistic forecasts, is lower when the posterior is calibrated and its intervals are narrow, and the diffusion posterior has the best CRPS on every system, as Table 1 shows. The uniform prior also yields a full posterior, the balance-conditioned Gaussian whose mode is the WLS estimate (Appendix § 7), with wider intervals than the diffusion posterior.
Figure 3. Empirical coverage against credible level for the diffusion posterior, the Gaussian EB posterior, and the oracle WLS. The oracle WLS and the Gaussian EB lie on the diagonal, and the diffusion posterior turns conservative at intermediate levels on the larger systems.
Masking the generation of the six core zones, while their load and border flows stay reported, tests recovery of unreported generation. The same 84 signals are masked on every system, so an overall error would shrink with system size by averaging alone, and we score only the masked signals. For a masked zone, the balance fixes the total generation, which all methods recover within 360-610 MW, but not its division among the fuels. A uniform prior carries no information on that division, so the constrained WLS QP fails, its error exceeding 5 000 MW on every system. The learned priors infer the division from the zones that still report, more accurately as the system grows, as Figure 1 shows, and the diffusion EB prior’s advantage over the Gaussian one grows from 6% to 18%.
Figure 1. RMSE on the masked signals against the number of core zones with masked generation.
The covariance is a belief, so we re-estimate and evaluate both learned priors with it scaled by 0.5 and by 2. WLS is immune to this stress, since a global rescaling of leaves its minimizer unchanged, and the two learned priors fail in opposite directions. Overstating the noise costs the diffusion prior little and understating erases its advantage, while the Gaussian EB prior tolerates understating but degrades past the raw observations when the noise is overstated, since its covariance subtraction then removes variance the states actually have. The diffusion prior stays the most accurate under every belief. The diffusion EB prior is not a faithful recovery of the state distribution, its marginals 77% too wide, yet it reconciles best, because reconciliation is governed by the likelihood and a wide, well-calibrated prior still regularizes the posterior. The prior predictive comparison of Appendix § 10.§ 2, prior samples plus the believed noise against the observed , reproduces their spread and balance-gap magnitude, a ground-truth-free check that carries over to real data. Appendix § 10 gives the complete results.
Conclusion
We formulated ENTSO-E data reconciliation as Bayesian inference and replaced the uniform prior of classical reconciliation with a diffusion model estimated from observations alone. On synthetic systems with known ground truth, the diffusion prior gives the most accurate reconciliation with calibrated or conservative credible intervals, and it recovers unreported generation where the uniform prior fails. The observation noise of the benchmark remains Gaussian, independent, and constant per fuel, while real reporting errors scale with magnitude, are autocorrelated, and carry per-TSO bias that the empirical-Bayes loop cannot remove [21], though a magnitude-dependent fits the same likelihood. Posterior draws do not enforce non-negativity, which predict-project-renoise [22] would impose. On real data, validation rests on balance residuals and cross-source references.
Acknowledgments and Disclosure of Funding
The authors would like to thank Engie GMA’s team, especially Alexandre Huynen, for sharing expert knowledge about the problem. Adrien Bolland acknowledges the financial support received during part of this research through a research fellowship from the F.R.S.-FNRS.
Use of LLMs.
We used an LLM for grammar and spelling correction and sometimes to suggest clearer phrasing. All research content, including the method, the experiments, the results and their interpretation, is our own. We reviewed every sentence and take full responsibility for the content of the paper.
References
Posterior under a Gaussian likelihood: uniform vs. Gaussian prior
This appendix derives the Gaussian posteriors under the uniform prior of the WLS family and under a generic Gaussian prior, which includes the Gaussian EB prior of § 2. Proposition 1 gives the posterior under the uniform prior, Proposition 2 under a Gaussian prior. The last subsection conditions both on the balance . The experiments sample the Gaussian EB and oracle WLS posteriors from this conditioned form.
Let be the latent state and the observation. We drop the time subscript . The likelihood is linear-Gaussian,
where is the diagonal mask of (1) and is the noise covariance of the main text. As a function of , this density is proportional to the likelihood for any mask, which is all Bayes’ theorem uses. Since is diagonal, its exponent is a sum of one residual per signal, , and an unreported signal has and , so its residual is zero and only the reported signals enter.
Uniform prior
The main text takes the prior uniform over . To keep the posterior Gaussian, this subsection relaxes it to the prior uniform over , and the conditioning of the last subsection restores the balance exactly. Only the non-negativity constraints of are then missing, so posterior draws can violate them (§ 4). Either uniform prior is improper, being an unbounded cone, but the posterior below is proper when every signal is reported.
Proposition 1. Under the uniform prior , with of full column rank, the posterior is Gaussian,
with
Its mean is the minimizer of the unconstrained objective of (3).
Proof. With a uniform prior the log-posterior equals the log-likelihood up to an additive constant, a quadratic in with Hessian by the full-rank assumption. Completing the square gives (5) with parameters (6).
The full column rank required here means, for the diagonal mask, that every signal is reported. When a signal is unreported, it drops out of the likelihood, so the observations carry no information about it and the inverse in (6) does not exist. The posterior has infinite variance along the unreported signals, and is then called improper. The missing information must come from elsewhere, from the prior, or, under the uniform prior, from the balance and the non-negativity constraints. The maximum-a-posteriori estimate is then the constrained program of (3).
Gaussian prior
Proposition 2. Under the Gaussian prior with , the posterior is Gaussian,
with
Proof. The log-posterior is the sum of two quadratics in , with Hessian , positive definite as the sum of a positive semidefinite and a positive definite matrix, and linear coefficient . Completing the square gives a Gaussian with covariance and mean , which are (8) and (9).
Conditioning on the balance
Both posteriors above have the form , where and abbreviate the mean and covariance of either proposition, and ignore the balance. Conditioning on keeps only the states that close the balance. The pair is jointly Gaussian, so the standard Gaussian conditioning formula applies and gives the map
where is invertible because and has full row rank. The formulas have a direct reading. The residual is the balance gap of the mean, one entry per zone. The map subtracts a correction that spreads this gap over the signals according to their variances and correlations in , and is the closest state to that closes the balance, in the metric weighted by , the reconciliation of itself. The covariance removes all variance along the directions the balance fixes and keeps the rest, so every draw satisfies the balance exactly, and , and is singular with rank . Draws are therefore generated from a factorization of rather than its inverse. Neither posterior imposes the non-negativity constraints. For the map reduces to the orthogonal projection of the diffusion sampler (Appendix § 8). The Gaussian EB posterior of § 2 applies the map to Proposition 2 with . The oracle WLS posterior of the calibration experiment applies it to Proposition 1.
For the uniform prior, and , so is the balance-only WLS reconciliation map and the conditioned posterior is . Adding fresh noise to the observation and reconciling the result, with , draws exactly from this posterior, because . Each draw is one extra WLS solve. The identity holds for the balance constraint alone, not for the quadratic program with non-negativity.
The diffusion prior
This appendix gives the parameterization of the diffusion prior of § 2, its EM training, and the guided posterior sampler.
Parameterization
A diffusion model pairs a fixed forward process that gradually noises the state with a learned reverse process that undoes it. The forward process adds Gaussian noise to the state, producing , with marginals ranging from the state distribution at to pure noise as grows large [12]. This is the noise level of the diffusion process, distinct from the observation noise of (1). A denoiser estimates the state from its noised version. New states are generated by reversing the process, starting from pure noise and denoising down to , so the denoiser defines a distribution over states, the prior . Training the denoiser on states would minimize the denoising loss of Karras et al. [13]
where is the state distribution, a distribution over noise levels, the forward process, and a noise-dependent weighting. We have no samples of the state, only the observations , so this loss cannot be evaluated directly.
EM training
The EM scheme of [11] estimates the denoiser from the observations alone, alternating two steps over iterations . The E-step fills in plausible states for each observation, drawing samples from the posterior with the moment-matching posterior sampler (MMPS) of [11]. The M-step minimizes (12) on those samples for epochs from a fresh optimizer. As a warm-up, our implementation first fits the denoiser to the observations for epochs, an M-step with the observations in place of posterior samples, so the initial prior still carries the observation noise, which the iterations sharpen toward the state distribution. We run iterations with posterior samples per observation, warm-up epochs, and refit epochs per iteration. The denoiser is a four-block residual MLP of width 256 to 1024, growing with the system, and both the E-step and inference samplers use 64 reverse steps. Reconciling one time step with 64 draws thus takes 4,096 network evaluations, about 5 GFLOP on EU6 and 82 GFLOP on EUFULL, and training takes about forward-pass equivalents, dominated by the M-steps. The Gaussian EB prior uses no network and needs about FLOP per time step on EUFULL.
Guided posterior sampling
At inference we sample from the learned prior with a denoising diffusion implicit model (DDIM) reverse process [17] on normalized states, standardized per signal with the mean and variance of the training observations. At every step, the MMPS correction replaces the denoised estimate (the subscript is the noise level , not a time) by the Gaussian update , with the covariance of the state given its noised version under a Gaussian prior with a fixed covariance , estimated by moment matching from observations, so the update needs no differentiation of the denoiser. Signals with small observation noise are pulled strongly toward their reported values, and unreported signals, carrying a huge variance in , are left to the prior. To enforce the balance, each step then projects the corrected estimate onto the null space of with (denoising diffusion null-space model, DDNM [18]). In image restoration DDNM also reinserts the observation into the estimate, but our constraint carries no observation, so the step is a pure projection. The projection acts in normalized coordinates, where the mean shift of the normalization leaves a small balance residual in physical units. A final projection in physical units removes it, so every sample satisfies the balance to numerical precision. The guided reverse process draws approximately from the posterior implied by the learned prior and the likelihood.
Experimental details
ERAA combines the installed capacities, demand, net transfer capacities, and renewable capacity factors of a target year with 35 historical weather years, the climate years. Grounding the benchmark in this model gives it the structure of the real European system. The interconnections come from the ERAA net-transfer-capacity data, and each zone carries 14 fuel categories. Each system contains the previous one and adds zones and their interconnections:
EU6 (, , ): the base system BE, NL, DE, FR, CH, AT.
EUWEST (, , ): adds Iberia (ES, PT), Great Britain (UK), and the Nordic zones FI and SE1-SE4.
EUWIDE (, , ): adds Italy (seven bidding zones) and the complete Nordic set (DK, NO).
EUFULL (, , ): adds Central Europe (CZ, SK, HU, SI) and the Baltics (EE, LT, LV).
We take 2026 as the target year, so every climate year runs the same installed capacities and network through different weather, giving 35 years of hourly data from a single unchanged system. The dispatch is a linear program and is deterministic, one realization per climate year, so the only randomness in the observations is the observation noise. Each border flow is bounded by its net transfer capacity and Kirchhoff’s laws are not imposed, so the benchmark flows are exchanges without loop flows, while the platform reports physical flows. The noise is independent and zero-mean, so is diagonal with per-signal standard deviations
where is a base standard deviation for each signal type and a per-zone multiplier. Each fuel has one constant base, a plausible few percent of a continental zone’s typical magnitude. Onshore wind and solar are the noisiest at 300 MW, cross-border flows 200 MW, load 150 MW, and well-metered sources such as nuclear and run-of-river the cleanest at 50 MW. The per-zone multipliers lie in [0.8, 1.2] and add mild TSO-to-TSO variation, for which no published ranking exists. Every signal is reported, so the mask of Eq. (1) is the identity, and partial reporting is studied separately in the masking experiment. Holding the noise constant per fuel keeps the benchmark controlled, so the comparison isolates the reconciliation method rather than the noise model. We split along the climate-year axis to avoid temporal leakage, giving 245 280 training hours, 26 280 for validation, and 35 040 for testing. The states enter only the evaluation.
Table 2: Reconciliation methods compared. “” is the noise covariance the method uses (or its substitute). “Historical data” is what the method sees beyond the observations it reconciles.
On this benchmark the documented covariance equals the true one, since the noise is generated with it, so documented WLS and oracle WLS are the same method and we report only oracle WLS. The de Chalendar heuristic makes the covariance implicit, assuming variances proportional to signal magnitudes,
where is the rolling mean of signal and , are constants of the heuristic. Their published objective minimizes the squared adjustments , each divided by , over , which is the program (3) after substituting . We keep their preprocessing, in which static and rolling thresholds reject implausible values and rejected or missing cells are imputed by interpolation, and drop their aggregation and anti-symmetry constraints, since our state has no aggregate signals and represents each border by two non-negative directed flows instead of one signed flow.
The all-signal root-mean-square is about 10 GW, so an overall RMSE near 100 MW is a relative error of about one percent, though individual signals vary widely. The diffusion EB prior and the oracle WLS approximate the posterior with an ensemble of 64 draws each, and their point estimate is the ensemble mean, which for the oracle WLS coincides with the WLS solution. The Gaussian EB point estimate is its posterior mean, in closed form. The expected squared error of an individual draw exceeds that of the posterior mean by the posterior variance, so scoring draws would penalize the prior for honestly reporting its spread. RMSE then compares each method’s best point summary. A posterior mean cannot represent a multimodal or wide posterior, so the distribution itself is judged separately, by the calibration and CRPS results of Appendix § 10.
Additional results
Per-component accuracy
Table 3: RMSE (MW) on the six-zone system, all test hours. The generation, load, and flow columns report the per-component RMSE, and the overall column is the all-signal RMSE.
Table 3 reports RMSE per component on EU6. The non-negativity constraints lower the flow residual of the constrained WLS QP relative to the unconstrained naive and oracle WLS, 147.9 MW against roughly 186 MW, and the diffusion EB prior lowers it further to 117.5 MW while also attaining the lowest generation and load residuals.
The Chalendar weights suit the EIA data they were built for, where reporting error grows with the size of the balancing authority [20]. The heuristic needs no noise covariance and no historical corpus, which makes it the reference for real-time reconciliation where neither is available.
Prior fidelity
Comparing prior samples directly to the test states needs the ground truth, so that check runs only on synthetic data. The same question has a ground-truth-free form usable on real data, the prior predictive check. We draw unconditional prior samples, add the documented noise to form the prior predictive , and compare the two distributions with per-signal and sliced Wasserstein distances and the standard-deviation mismatch (Table 4). On all four systems the diffusion EB prior predictive matches the observations closely, with per-signal standard deviations within 8 to 12 percent, and on EU6 it is on par with the oracle. The balance-gap magnitude is reproduced to within about one percent on every system, for example a root-mean-square of 705 MW against 709 MW in the observations on EU6. The excess width seen against the states is mostly hidden here, because the observation noise is present in both the predictive and the data, so an over-wide prior plus noise still lands near the observed spread. Figure 4 shows the check on the directed flows of three borders and on solar and load in three zones. The state distribution is a mixture of operating regimes, a border exports in one direction or the other, a line runs at its transfer limit or idles, solar is zero at night, and load and solar rise together during the day across zones. The prior recovers the location, the weight, and the cross-signal coupling of these regimes from noisy observations alone.
Figure 4. Prior predictive check: unconditional prior samples plus the documented noise (blue) against the test set (orange). Left: the directed flows of the BE-NL, BE-DE, and FR-DE borders, each pair active one direction at a time. Right: solar and load in BE, DE, and FR.
| System | Prior | Marginal | Sliced- | Std mismatch |
| EU6 | Oracle | 0.05 | 0.06 | 0.07 |
| EU6 | Diffusion EB | 0.08 | 0.05 | 0.08 |
| EUWEST | Diffusion EB | 0.10 | 0.06 | 0.08 |
| EUWIDE | Diffusion EB | 0.13 | 0.08 | 0.12 |
| EUFULL | Diffusion EB | 0.13 | 0.08 | 0.12 |
Table 4. Prior predictive check against the test set, in observation-standard-deviation units. Lower is closer.
Calibration
Calibration is tested as follows. At each level , the 1024 draws per time give one central credible interval per signal, and the empirical coverage is the fraction of these intervals that contain the state. The check uses the test states, not draws from the learned prior, so nominal coverage is not guaranteed by construction, and it tests the prior, the likelihood, and the sampler together, one signal at a time. Figure 3 plots empirical coverage against over the active signals, those whose state standard deviation over the test set exceeds 1 MW, because the structurally constant signals are covered trivially by every method and would only raise the coverage. A trustworthy posterior lies on the diagonal. The oracle WLS lies on the diagonal on every system, a control for the diagnostic, and the Gaussian EB stays close to nominal throughout. The diffusion posterior turns conservative at intermediate levels on the two larger systems, up to 14 points above the diagonal on EUFULL, and its 90% coverage stays between 88.6% and 93.5% on all four.
Table 5 reports the mean 90% interval width. The diffusion posterior is 31% to 41% below the oracle WLS and 15.6% below the Gaussian EB at six zones, a gap that narrows to 2.0% at thirty-five. The oracle WLS is calibrated only by being wide, and the Gaussian EB is calibrated on average but its posterior covariance is the same every hour and cannot adapt.
| System | Diffusion EB | Gaussian EB | Oracle WLS |
| EU6 | 291 | 344 | 419 |
| EUWEST | 308 | 333 | 442 |
| EUWIDE | 345 | 303 | 472 |
| EUFULL | 347 | 284 | 471 |
Table 5. Mean 90% interval width in MW over the active signals, full test set (35 040 hours).
Robustness to noise misspecification
The documented covariance is a belief, and in deployment a wrong belief enters everywhere at once. The E-step removes the believed noise during training, the Gaussian EB subtracts it from the observation covariance, and the likelihood uses the same belief again at inference, while the data keep their true noise. We therefore re-estimate both learned priors with the believed covariance scaled by and evaluate on the unchanged test observations with the same wrong belief. We use EU6, where the diffusion training is cheapest. The constrained WLS QP receives the same belief, but its minimizer is unchanged by a global rescaling of the weights, so the small variation in its row is numerical and the stress falls on the learned priors. The experiment misspecifies only the overall scale of the belief, so every variance is off by the same factor. A belief that is wrong about which signals are noisy is not tested.
Table 6 shows the two learned priors failing in opposite directions while the diffusion prior stays ahead everywhere. For the diffusion prior the gap to the constrained WLS QP is 11.6% at against 13.3% under the correct belief, and understating leaves 112.9 against 114.1 MW at , because the E-step and the inference likelihood then both over-trust the noisy observations. A plausible reason for the robustness of the diffusion prior is that the EM loop drives it toward consistency of prior plus believed noise with the observations, so the prior itself absorbs part of the wrong belief. The misspecified scale is also detectable, since the balance residual has covariance , so a wrong scale appears as residuals larger or smaller than the belief predicts and can be corrected before training.
| Constrained WLS QP | 114.1 | 115.1 | 118.6 |
| Gaussian EB prior | 121.6 | 117.6 | 151.2 |
| Diffusion EB prior | 112.9 | 99.8 | 104.9 |
| Diffusion EB gap to QP | 1.1% | 13.3% | 11.6% |
| Diffusion EB gap to Gaussian EB | 7.2% | 15.1% | 30.6% |
Table 6. Noise-belief misspecification on EU6, full test set. Overall RMSE in MW under the believed covariance .
References
- [1]European Commission. Commission Regulation (EU) No 543/2013 of 14 June 2013 on submission and publication of data in electricity markets. Official Journal of the European Union, L 163, pp. 1–12, 2013. URL https://eur-lex.europa.eu/eli/reg/2013/543/oj.
- [2]Lion Hirth, Jonathan Mühlenpfordt, and Marisa Bulkeley. The ENTSO-E Transparency Platform–A review of Europe’s most ambitious electricity data platform. Applied Energy, 225:1054–1067, 2018.DOI
- [3]Ember. European Electricity Review 2024. Technical report, Ember, 2024. URL https://ember-energy.org/latest-insights/european-electricity-review-2024/.
- [4]ENTSO-E. ENTSO-E Transparency User Group, 2024. URL https://www.entsoe.eu/data/transparency-platform/etg/. Accessed: February 5th 2026.
- [5]Shankar Narasimhan and Cornelius Jordache. Data Reconciliation and Gross Error Detection: An Intelligent Use of Process Data. Elsevier, 1999.
- [6]Fred C Schweppe and J. Wildes. Power System Static-State Estimation, Part I: Exact Model. IEEE Transactions on Power Apparatus and systems, PAS-89(1):120–125, 1970.DOI
- [7]Kursat Rasim Mestav, Jaime Luengo-Rozas, and Lang Tong. Bayesian State Estimation for Unobservable Distribution Systems via Deep Learning. IEEE Transactions on Power Systems, 34(6):4910–4920, 2019.arxiv.org/abs/1811.02756
- [8]Yi He, Songjian Chai, Zhao Xu, Chun Sing Lai, and Xu Xu. Power system state estimation using conditional generative adversarial network. IET Generation, Transmission & Distribution, 14(24):5823–5833, 2020.
- [9]Jianhua Pei, Jingyu Wang, Dongyuan Shi, and Ping Wang. Detection and Imputation-Based Two-Stage Denoising Diffusion Power System Measurement Recovery Under Cyber-Physical Uncertainties. IEEE Transactions on Smart Grid, 15(6):5965–5980, 2024.
- [10]Jianhua Pei, Ping Wang, Jingyu Wang, and Dongyuan Shi. Deep Generative Model-Aided Power System Dynamic State Estimation and Reconstruction with Unknown Control Inputs or Data Distributions. arXiv preprint arXiv:2501.02928, 2025.arxiv.org/abs/2501.02928
- [11]François Rozet, Gérôme Andry, François Lanusse, and Gilles Louppe. Learning diffusion priors from observations by expectation maximization. In The Thirty-eighth Annual Conference on Neural Information Processing Systems, 2024.arxiv.org/abs/2405.13712
- [12]Yang Song, Jascha Sohl-Dickstein, Diederik P Kingma, Abhishek Kumar, Stefano Ermon, and Ben Poole. Score-Based Generative Modeling through Stochastic Differential Equations. In International Conference on Learning Representations, 2021. URL https://openreview.net/forum?id=PxTIG12RRHS.
- [13]Tero Karras, Miika Aittala, Timo Aila, and Samuli Laine. Elucidating the Design Space of Diffusion-Based Generative Models. In Alice H. Oh, Alekh Agarwal, Danielle Belgrave, and Kyunghyun Cho, editors, Advances in Neural Information Processing Systems, 2022. URL https://openreview.net/forum?id=k7FuTOWMUC7.arxiv.org/abs/2206.00364
- [14]CF Jeff Wu. On the Convergence Properties of the EM Algorithm. The Annals of statistics, 11 (1):95–103, 1983.DOI
- [15]Geoffrey J McLachlan and Thriyambakam Krishnan. The EM Algorithm and Extensions. John Wiley & Sons, 2008.
- [16]Sivaramam Balakrishnan, Martin J Wainwright, and Bin Yu. Statistical guarantees for the EM algorithm: From population to sample-based analysis. The Annals of Statistics, 45(1):77–120, 2017.arxiv.org/abs/1408.2156
- [17]Jiaming Song, Chenlin Meng, and Stefano Ermon. Denoising Diffusion Implicit Models. In International Conference on Learning Representations, 2021.arxiv.org/abs/2010.02502
- [18]Yinhua Wang, Jiwen Yu, and Jian Zhang. Zero-Shot Image Restoration Using Denoising Diffusion Null-Space Model. In The Eleventh International Conference on Learning Representations, 2023. URL https://openreview.net/forum?id=mRieQgMtNTQ.arxiv.org/abs/2212.00490
- [19]ENTSO-E. European resource adequacy assessment 2024 - modelling data, 2024. URL https://www.entsoe.eu/eraa/2024/modelling-data/. Input and modelling data: demand, PECD renewable potentials, NTCs, generation capacities.
- [20]Jacques A de Chalendar and Sally M Benson. A physics-informed data reconciliation framework for real-time electricity and emissions tracking. Applied Energy, 304:117761, 2021.
- [21]Hussain Kazmi and Zhenmin Tao. How good are TSO load and renewable generation forecasts: Learning curves, challenges, and the road ahead. Applied Energy, 323:119565, 2022.DOI
- [22]Omer Rochman-Sharabi and Gilles Louppe. Predict-Project-Renoise: Sampling Diffusion Models under Hard Constraints. arXiv preprint arXiv:2601.21033, 2026.


