REVIEW 5 major objections 5 minor 4 cited by
Electrical Load Forecasting in Smart Grid: A Personalized Federated Learning Approach
T0 review · 5 major / 5 minor · reviewed 2026-08-12 · deepseek-v4-flash
Pith's one-line read A personalized federated LSTM lets each smart meter choose its own learning rate and forecasts next-hour electricity load more accurately than fixed-rate federated or centralized LSTM baselines on non-IID household data.
desk verdict This is a weakly validated hyperparameter-tuning trick dressed up as meta-learning, and the printed algorithm cannot even select a learning rate; not ready for peer review. read the letter →
The pith
A machine-rendered reading of the paper's core claim, the machinery that carries it, and where it could break.
The reading
What carries the argument
The load-bearing object is the temporary-validation learning-rate selector inside each smart meter: a small set of candidate learning rates $\alpha_1,\dots,\alpha_j$, a 20 percent holdout slice $D^{(k)}_{\text{temp},m}$ of the client's own training data, and a rule that after ten local rounds picks the rate $\beta^{(k)}_m$ giving the lowest loss, then uses $\beta^{(k)}_m$ for the remaining fifty local rounds. The server side is ordinary federated averaging, $w_{k+1} = \frac{1}{M}\sum_{m\in M} w_{m,k}$, so the personalization is entirely client-local and adds no aggregation cost. The underlying sequence model is an LSTM with one 50-unit layer, dropout 0.2, and a dense output neuron, fed 24 hourly load values to forecast the next hour. The paper also connects its objective to a Moreau-envelope formulation of personalization via pFedMe, but the algorithm itself does not compute proximal updates; the mechanism that carries the reported gains is the per-client rate choice.
What would settle it
Take the exact non-IID five-client setup and rerun Algorithm 1 with the selection rule inverted—choose the learning rate with the highest loss on the temporary validation slice—while keeping every other detail fixed; if the inverted rule matches or beats the proposed rule on test MAE and RMSE, the validation-based meta-learning step is not the cause of the reported improvement. A second check is to compare the rate chosen at round 10 with the rate that would have produced the lowest loss at round 50; any substantial disagreement undermines the transfer assumption.
Extended reading notes
Core claim
The paper's central claim is that per-client learning-rate selection, framed as meta-learning, is enough to make federated load forecasting competitive under non-IID smart-meter data. In Algorithm 1, after receiving the global model, each sampled meter runs ten local rounds with each candidate learning rate on a temporary 20 percent slice of its training data, records the rate with the lowest loss, and then trains for fifty local rounds on its full training data using that chosen rate; the server aggregates the resulting local models by plain federated averaging. The simulations use five clients with deliberately different batch sizes and data fractions and a 24-hour input LSTM predicting next-hour load, and compare against fixed-rate PFL variants and against LSTM and federated-learning baselines. The paper interprets the results as showing that no single learning rate is best for all clients, that meta-learning effectively combines the strengths of the candidate rates, and that the resulting forecasts track rapid consumption changes better than the baselines.
Load-bearing premise
The whole method depends on the assumption that the learning rate with the lowest loss after ten quick validation rounds is also the learning rate that will train best over the following fifty local rounds on the full training data.
Editorial extensions
If this is right
- If the reported gains are real, utilities can improve next-hour load forecasts without pulling household consumption histories to a central server, preserving the privacy that regulations have protected.
- Meters with different hardware, batch sizes, and data volumes can participate in the same global round, because personalization lives in the local rate choice rather than in server-side tuning.
- The method adds no server-side computation and no extra communication beyond the normal model upload and download, so the personalization step scales with the number of meters.
- On non-IID splits the meta-learning curve stays below fixed-rate curves in the reported MAE and RMSE plots, meaning the chosen rate avoids both the fast-starting-but-unstable high rate and the too-slow low rate.
Reading between the lines
- The reported comparison is against fixed learning rates and standard baselines; an untested implication is that the same temporary-validation rule could be applied to batch size or number of local epochs, not just learning rate.
- One way to stress the claim is to benchmark the selector against a cheap random or round-robin learning-rate schedule; if random selection matches the validation-selected rates, the paper's meta-learning label would overstate what the mechanism contributes.
- Because the selector uses only a client's own validation slice, the idea should transfer to other non-IID time-series tasks such as decentralized forecasting of traffic or renewable generation, though the paper does not test those.
- The method's per-client choice could in principle be combined with other personalization layers such as local fine-tuning or regularization, yielding further gains beyond those the paper reports.
Editorial analysis
A structured set of objections, weighed in public.
Referee Report
Summary. The paper proposes a personalized federated learning method for short-term electrical load forecasting in smart grids, called personalized meta-LSTM. The method lets each smart meter select one of three learning rates per global round by evaluating loss on a temporary validation subset, then trains a local LSTM with the chosen rate and aggregates local models through FedAvg. Experiments on the UCI Individual Household Electric Power Consumption dataset compare the method with standalone LSTM, centralized LSTM, and a federated LSTM baseline under IID and non-IID client partitions. The authors claim that the proposed approach outperforms state-of-the-art ML and FL methods in load forecasting accuracy.
Significance. If substantiated, per-client adaptive learning-rate selection would be a simple, server-light personalization mechanism for non-IID federated load forecasting and would preserve data privacy, which is a timely and important problem. The paper clearly motivates the privacy and heterogeneity challenges, uses a real-world dataset, and provides an intuitive architecture. However, the central contribution is not established: the algorithmic description is internally inconsistent, the experiments lack repeated runs and error bars, and the comparison omits the personalized federated learning baselines that the paper itself cites as closest work. Thus the claimed advantage over state-of-the-art methods is unsupported in the present version.
major comments (5)
- [Algorithm 1, lines 8–13; Eqs. (2)–(3)] The loss functions in Eqs. (2) and (3) do not depend on the learning rate α_i, so the quantity f′_{i,k}(α_i) in line 9 is identical for every candidate rate and cannot identify a best rate. The prose in Sec. IV-B describes a different procedure—10 trial rounds with each rate on D^{temp}, then 50 training rounds with the winner—but Algorithm 1 places the selection inside the per-epoch local loop and has no trial rounds. The central claimed mechanism is therefore undefined as written; please reconcile the pseudocode with the prose and provide a well-defined selection rule.
- [Sec. IV-C, Figs. 3–5] The empirical comparison is based on a single run with no error bars, no multiple seeds, and no statistical tests. The differences between PFL and FL in Fig. 5 are small in absolute terms, so without repeated runs the conclusions that PFL is 'far more stable' and achieves 'better' accuracy are not supported. Please report mean and standard deviation over at least several independent runs and state the number of runs.
- [Sec. IV-C, Fig. 5; Sec. III] The comparison omits the personalized federated learning baselines cited in the paper, including pFedMe [18], Wang et al. [1], and Qu et al. [12]. The paper claims to outperform 'state-of-the-art ML and FL methods,' but without these PFL baselines the claim is unsupported. Please add the relevant PFL baselines or narrow the claim to the methods actually compared.
- [Sec. IV-A and IV-B] The experiments use one household from the UCI dataset and partition its data into synthetic clients; this does not reflect heterogeneous smart meters from different households, which is the stated motivation. The claim that the method handles 'diverse SMs' with different data sizes is only tested by splitting a single time series. A multi-household or multi-meter dataset, or a clear justification of why synthetic clients are sufficient, is needed to support the heterogeneity claims.
- [Sec. III, Eqs. (10)–(16)] The Moreau-envelope/pFedMe formulation is not connected to Algorithm 1. The notation prox_{f_i/x_i}(w_k) in Eq. (13) is undefined, Eq. (14) is not a standard Moreau envelope, and the per-client learning-rate selection in Algorithm 1 does not correspond to any update in Eqs. (10)–(16). This section neither provides theoretical support nor clarifies the algorithm; either derive the algorithm from this formulation or remove the formalization.
minor comments (5)
- [Sec. II-B] The notation D^{temp}_{m,k}, Trainloader, and Testloader is not defined precisely; in particular, the relationship between D^{temp}_{m,k} and the 80%/20% split described in Sec. II-A should be stated in one place.
- [Algorithm 1] Line 9 has mismatched parentheses and uses f′_{i,k} before it is defined; also, the outer loop iteration over global rounds k and the inner loop over local epochs t make the variable k overloaded.
- [Eq. (1)] In Eq. (1), k is used both as the global-round index and as an index for the data sample, which is confusing; please use different symbols for these two roles.
- [Sec. IV-C, Figs. 3–5] The x-axis is labelled 'Epochs' but the text describes global rounds; the y-axis is labelled 'Loss' while the captions say MAE and RMSE. Please make the axes consistent and explicitly state which quantity is plotted.
- [References] Reference [11], a self-citation on modulation recognition, appears in the sentence about FL struggling with non-IID data but is not related to that claim; please replace it with an appropriate citation or remove it.
Circularity Check
No significant circularity: the paper is an empirical comparison; per-client learning-rate selection is validation-based model selection, not a derivation from the target.
full rationale
The paper's central claim is an empirical accuracy comparison (Figs. 3-6), not a derived prediction from first principles. The meta-learning step selects a learning rate by evaluating three candidates on a 20% temporary validation slice (Sec. II-A, IV-B, Algorithm 1 lines 8-13), then trains on a separate Trainloader and evaluates on a held-out Testloader. This is standard hyperparameter selection on a training subset, so the test result is not forced by construction. The one self-citation ([11], Rahman and Nguyen) supports the general statement that FL struggles with non-IID data and is not load-bearing for the proposed algorithm; the baselines LSTM [3] and FL [6] are external. Section III's Moreau-envelope/pFedMe equations are not used to derive the actual Algorithm 1 update, and any mismatch between the prose and pseudocode is a correctness or clarity concern, not circularity. No load-bearing step reduces to its own inputs by definition.
Assumptions & free parameters
free parameters (5)
- Candidate learning rate set =
{0.05, 0.001, 0.0001}
- Local evaluation rounds and training rounds =
10 evaluation rounds per LR, 50 local training rounds
- Client partition configuration =
5 clients with data shares 20/20/20/10/25 percent and batch sizes 128/128/128/64/256
- LSTM architecture hyperparameters =
50 units, dropout 0.2, sequence length 24
- Global rounds K =
100
assumptions (3)
- domain assumption A 24-hour window of past load is sufficient input for next-hour load forecasting with an LSTM.
- ad hoc to paper The learning rate with the lowest loss after 10 local rounds on a temporary validation slice is the best learning rate for 50 local training rounds.
- domain assumption Federated averaging of model weights protects customer privacy.
Cite this review
Pith. "Pith review of Electrical Load Forecasting in Smart Grid: A Personalized Federated Learning Approach." pith.science (2026). https://pith.science/paper/L7JCJPIW
@misc{pith2026241110619,
author = {Pith},
title = {Pith review of: Electrical Load Forecasting in Smart Grid: A Personalized Federated Learning Approach},
year = {2026},
howpublished = {\url{https://pith.science/paper/L7JCJPIW}},
note = {Machine review of arXiv:2411.10619}
}
read the original abstract
Electric load forecasting is essential for power management and stability in smart grids. This is mainly achieved via advanced metering infrastructure, where smart meters (SMs) are used to record household energy consumption. Traditional machine learning (ML) methods are often employed for load forecasting but require data sharing which raises data privacy concerns. Federated learning (FL) can address this issue by running distributed ML models at local SMs without data exchange. However, current FL-based approaches struggle to achieve efficient load forecasting due to imbalanced data distribution across heterogeneous SMs. This paper presents a novel personalized federated learning (PFL) method to load prediction under non-independent and identically distributed (non-IID) metering data settings. Specifically, we introduce meta-learning, where the learning rates are manipulated using the meta-learning idea to maximize the gradient for each client in each global round. Clients with varying processing capacities, data sizes, and batch sizes can participate in global model aggregation and improve their local load forecasting via personalized learning. Simulation results show that our approach outperforms state-of-the-art ML and FL methods in terms of better load forecasting accuracy.
Figures
Figures from the paper (3 more)
Forward citations
Cited by 4 Pith papers
-
Sporadic Federated Learning Approach in Quantum Environment to Tackle Quantum Noise
SpoQFL applies sporadic learning to quantum federated learning by suppressing noisy client updates, claiming higher accuracy and more stable convergence in noisy-device simulations.
-
Meta-Federated Learning: A Novel Approach for Real-Time Traffic Flow Management
Claims that combining federated and meta-learning improves simulated traffic prediction, but the method description is inconsistent and no code or data are provided.
-
Advancing Personalized Federated Learning: Integrative Approaches with AI for Enhanced Privacy and Customization
The paper claims a new personalized federated learning method with adaptive optimization, transfer learning, and differential privacy, but its methods and experiments concern traffic-flow prediction and contain no pri...
-
Integrating Personalized Federated Learning with Control Systems for Enhanced Performance
The paper proposes FedAvg plus an exponential learning-rate decay driven by loss reduction and calls it a control system, but provides insufficient evidence for the claimed gains.
Reference graph
Works this paper leans on
-
[1]
Personalized federated learning for individual consumer load forecasting,
Y . Wang, N. Gao, and G. Hug, “Personalized federated learning for individual consumer load forecasting,” CSEE Journal of Power and Energy Systems, vol. 9, no. 1, pp. 326–330, 2023
work page 2023
-
[12]
Person- alized federated learning for heterogeneous residential load forecasting,
X. Qu, C. Guan, G. Xie, Z. Tian, K. Sood, C. Sun, and L. Cui, “Person- alized federated learning for heterogeneous residential load forecasting,” Big Data Mining and Analytics , vol. 6, no. 4, pp. 421–432, 2023
work page 2023
-
[11]
Improved modulation recognition using personalized federated learning,
R. Rahman and D. C. Nguyen, “Improved modulation recognition using personalized federated learning,” IEEE Transactions on V ehicular Technology, 2024
work page 2024
-
[18]
Personalized federated learning with moreau envelopes,
C. T Dinh, N. Tran, and J. Nguyen, “Personalized federated learning with moreau envelopes,” Advances in Neural Information Processing Systems, vol. 33, pp. 21394–21405, 2020
work page 2020
-
[2]
A deep learning method for short-term residential load forecasting in smart grid,
Y . Hong, Y . Zhou, Q. Li, W. Xu, and X. Zheng, “A deep learning method for short-term residential load forecasting in smart grid,” IEEE Access , vol. 8, pp. 55785–55797, 2020
work page 2020
-
[3]
S. Bouktif, A. Fiaz, A. Ouni, and M. A. Serhani, “Optimal deep learning lstm model for electric load forecasting using feature selection and genetic algorithm: Comparison with machine learning approaches,” Energies, vol. 11, no. 7, p. 1636, 2018
work page 2018
-
[4]
Smart metering and privacy in Europe: Lessons from the Dutch case,
C. Cuijpers and B.-J. Koops, “Smart metering and privacy in Europe: Lessons from the Dutch case,” European data protection: Coming of age, pp. 269–293, 2013
work page 2013
-
[5]
Federated learning for internet of things: A comprehensive survey,
D. C. Nguyen, M. Ding, P. N. Pathirana, A. Seneviratne, J. Li, and H. V . Poor, “Federated learning for internet of things: A comprehensive survey,” IEEE Communications Surveys & Tutorials , vol. 23, no. 3, pp. 1622–1658, 2021
2021
Show all 19 references
-
[6]
Distributed load forecasting using smart meter data: Federated learning with recurrent neural net- works,
M. N. Fekri, K. Grolinger, and S. Mir, “Distributed load forecasting using smart meter data: Federated learning with recurrent neural net- works,” International Journal of Electrical Power & Energy Systems , vol. 137, p. 107669, 2022
2022
-
[7]
Electrical load forecasting using edge com- puting and federated learning,
A. Ta ¨ık and S. Cherkaoui, “Electrical load forecasting using edge com- puting and federated learning,” in ICC 2020-2020 IEEE international conference on communications (ICC) , pp. 1–6, IEEE, 2020
2020
-
[8]
Meta-learning in neural networks: A survey,
T. Hospedales, A. Antoniou, P. Micaelli, and A. Storkey, “Meta-learning in neural networks: A survey,” IEEE Transactions on Pattern Analysis and Machine Intelligence , vol. 44, no. 9, pp. 5149–5169, 2022
2022
-
[9]
A short-term load forecasting method using integrated cnn and lstm network,
S. H. Rafi, S. R. Deeba, E. Hossain, et al., “A short-term load forecasting method using integrated cnn and lstm network,” IEEE access , vol. 9, pp. 32436–32448, 2021
2021
-
[10]
Federated learning for short-term residential load forecasting,
C. Briggs, Z. Fan, and P. Andras, “Federated learning for short-term residential load forecasting,” IEEE Open Access Journal of Power and Energy, vol. 9, pp. 573–583, 2022
2022
-
[13]
Error feedback fixes signsgd and other gradient compression schemes,
S. P. Karimireddy, Q. Rebjock, S. Stich, and M. Jaggi, “Error feedback fixes signsgd and other gradient compression schemes,” in International Conference on Machine Learning , pp. 3252–3261, PMLR, 2019
2019
-
[14]
Powersgd: Practical low- rank gradient compression for distributed optimization,
T. V ogels, S. P. Karimireddy, and M. Jaggi, “Powersgd: Practical low- rank gradient compression for distributed optimization,” Advances in Neural Information Processing Systems , vol. 32, 2019
2019
-
[15]
Personalized federated learning: A meta-learning approach,
A. Fallah, A. Mokhtari, and A. Ozdaglar, “Personalized federated learning: A meta-learning approach,” arXiv preprint arXiv:2002.07948 , 2020
2002 arXiv
-
[16]
Short-term residential load forecasting based on lstm recurrent neural network,
W. Kong, Z. Y . Dong, Y . Jia, D. J. Hill, Y . Xu, and Y . Zhang, “Short-term residential load forecasting based on lstm recurrent neural network,” IEEE transactions on smart grid , vol. 10, no. 1, pp. 841–851, 2017
2017
-
[17]
Catalyst acceleration for first- order convex optimization: from theory to practice,
H. Lin, J. Mairal, and Z. Harchaoui, “Catalyst acceleration for first- order convex optimization: from theory to practice,” Journal of Machine Learning Research, vol. 18, no. 1, pp. 7854–7907, 2018
2018
-
[19]
Individual household electric power con- sumption data set,
G. Hebrail and A. Berard, “Individual household electric power con- sumption data set,” UCI Machine Learning Repository , 2012
2012
Reviewed August 12, 2026 · model on record in the stance chip above.
Discussion (0). Continue with ORCID to comment.