REVIEW 4 major objections 4 minor 13 references
Rethinking the initialization of Momentum in Federated Learning with Heterogeneous Data
T0 review · 4 major / 4 minor · reviewed 2026-08-12 · deepseek-v4-flash
Pith's one-line read The paper claims that reversing the weight order in momentum accumulation—favoring early local gradients over recent ones—consistently improves federated learning accuracy and F1-score under heterogeneous data.
desk verdict The method's defining equation contradicts its own motivation, and the empirical claims are unverifiable without code or a clear update rule. 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 central object is the reverse-exponential weighted average of Eq. (3), $\hat v_t = (1-\beta)v_0 + (1-\beta)\sum_{i=0}^{t-1}\beta^i \nabla f(x_i) + \beta^t \nabla f(x_t)$. It assigns the largest weight to the first local gradient and exponentially smaller weights to later gradients, the opposite of the standard momentum update in Eq. (1), so that the later, more biased local gradients contribute least to the momentum used to initialize the next local training round.
What would settle it
Measure the average cosine similarity (or projection) between client gradients and the mean gradient at each local step in a heterogeneous federated task. If the similarity does not decrease monotonically with the local step — for instance if it dips and then recovers — the premise of RMFL fails. A second check: run RMFL on a task where later gradients are known to be more aligned with the global objective (e.g., synthetic noise added only to early steps); if RMFL still wins, its benefit does not come from the claimed mechanism.
Extended reading notes
Core claim
The central claim is that the bias introduced by heterogeneous client data accumulates during local training, so a momentum buffer that emphasises recent gradients carries more bias than one that emphasises early gradients. The paper therefore replaces the standard exponential moving average of local gradients with the reverse-exponential weighted average defined in Eq. (3), and shows empirically that this estimated momentum outperforms the standard momentum baseline across three datasets, three heterogeneity levels, and local epochs of 2, 5, and 10. The authors interpret the gains as evidence that later local gradients are less useful for the global objective, supporting the reverse weighting.
Load-bearing premise
The method assumes that the bias of local gradients grows monotonically with the number of local training steps, so that early gradients are always less biased than recent ones; if that curve is not monotonic, the reverse weighting has no principled reason to help.
Editorial extensions
If this is right
- RMFL improves accuracy and F1-score over the MFL baseline on all three benchmark datasets and all heterogeneity levels tested, with the largest gains at the most skewed data (CIFAR10 with $\alpha=0.01$: accuracy 0.472 vs 0.247).
- The advantage of RMFL grows with the number of local epochs, so it makes longer local training less harmful than standard momentum in heterogeneous settings.
- RMFL shows smaller standard deviation than MFL in most settings, indicating more stable convergence across random seeds.
- The reweighting idea could be applied to other momentum-based federated optimization methods that use exponential moving averages, as the paper itself notes.
Reading between the lines
- A direct testable extension is to measure the actual gradient-bias-versus-local-step curve on a real heterogeneous task; the claimed mechanism predicts a monotonically increasing bias curve, and the benefit of RMFL should correlate with the slope of that curve.
- The case-study figures in Section III appear to conflate cumulative momentum with gradient cosine similarity, so the direct empirical support for the monotonic-bias premise is thinner than the main accuracy comparison; readers should verify that premise independently.
- If early local gradients were not more aligned with the global objective, RMFL could underperform standard momentum, so the method's success hinges on the direction of the bias, not merely its existence.
- The same reweighting idea could generalize to adaptive optimizers such as Adam-style second-moment estimates in federated learning, where the bias pattern is likely similar.
Signed reviews
Editorial analysis
A structured set of objections, weighed in public.
Referee Report
Summary. The manuscript proposes Reversed Momentum Federated Learning (RMFL), a modification of momentum in federated learning in which the standard exponential moving-average weighting of gradients is replaced by a weighting intended to give larger weight to early local gradients and smaller weight to later, putatively more biased gradients. The method is defined in Eq. (3). The authors report experiments on MNIST, CIFAR10, and CIFAR100 under Dirichlet heterogeneity levels alpha = 1, 0.1, 0.01 and local epochs 2, 5, and 10, claiming that RMFL consistently outperforms MFL in accuracy and macro F1-score.
Significance. The motivating observation that late local gradients are more biased in heterogeneous FL is worth studying, and the experimental matrix is broad. If the proposed weighting were correctly specified and robustly validated, the method would be a simple, practical contribution. However, the central definition in Eq. (3) does not realize the claimed reverse weighting: for beta = 0.9 the latest gradient receives nine times the weight of the immediately preceding one, so the experiments evaluate a method that is inconsistent with the paper's own description. In addition, the local update rule is under-specified and the experimental protocol has reproducibility and tuning issues. The contribution is therefore not established as written.
major comments (4)
- [§III-B, Eq. (3)] The proposed momentum estimate is not reverse-exponentially weighted. Expanding Eq. (3), the weights on ∇f(x_0), ..., ∇f(x_t) are (1−β), (1−β)β, ..., (1−β)β^{t−1}, β^{t}. For β = 0.9 and t = 10, ∇f(x_10) has weight 0.3487 while ∇f(x_9) has weight 0.0387, and the ratio of the most recent gradient to the immediately preceding one is β/(1−β) = 9. Thus the latest gradient receives the largest weight, directly contradicting the Section III claim that the weights are exponentially decayed with time going forward. Since Eq. (3) is the definition of RMFL, the experiments do not evaluate the proposed method as described.
- [§III-B and §IV] The local update rule is under-specified. Eq. (3) defines a momentum estimate, but the manuscript never states how this estimate is used to update the model parameters, for example whether x_{t+1} = x_t − η \hat v_t or whether \hat v_t is combined with an additional recursion. Without this equation, the implementation of RMFL is ambiguous and the reported results are not reproducible from the text alone, especially since no code is released.
- [§IV-A and Table I] The experimental protocol is reported inconsistently and the comparison is not fully controlled. Section IV.A states that 10 repeated runs are conducted, while the Table I caption reports results averaged over 5 runs. The learning rate is searched per method in [0.3, 0.1, 0.03, 0.01, 0.003, 0.001] for the fastest convergence, but neither the selected learning rates nor the convergence criterion are reported; a per-method search without a fixed budget can favor one method. These issues make it impossible to verify that the reported gains are due to the momentum weighting rather than to tuning or implementation details.
- [§III-A] The monotonic-bias premise is not established by the presented evidence. The text claims that gradient bias increases with local steps, but the supporting figures are mislabeled: Fig. 4 is captioned as cumulative momentum while the text describes average gradient cosine similarity, and Fig. 5 is captioned as active clients while the text describes average gradient projection. No numerical summary or statistical test is provided. If the bias does not increase monotonically, the reverse weighting has no principled basis; this premise must be quantified before the method can be assessed.
minor comments (4)
- [Throughout] Typos include 'beed proved' in the abstract, 'address more basis' in Section III, 'As the MFL parer suggested' in Section IV.A, and 'the most challenge CIFAR100' in Section IV.A; these should be corrected.
- [§IV.B and Table I] The text says Table I shows results on MNIST, CIFAR10, and CIFAR100, but the table caption says the results are on MNIST and CIFAR10 datasets; the caption should be aligned with the reported content.
- [Figures 4 and 5] The figure captions do not match the quantities described in the text; Fig. 4 is captioned as cumulative momentum while the text reports average gradient cosine similarity, and Fig. 5's caption says active clients while the text reports average gradient projection.
- [§II.B, reference [4]] The cited FedDyn paper is described as a federated distillation approach on recommender systems, but the dynamic-regularization FedDyn commonly referenced in federated learning is a different paper (Acar et al., ICLR 2021); the reference should be checked.
Circularity Check
No significant circularity: the RMFL weighting is a hand-designed heuristic and the reported accuracy comparisons are external, not derived from the proposal's definition.
full rationale
The paper proposes RMFL as a heuristic reweighting of local gradients, motivated by the observed increase in gradient divergence during local training. The central empirical claim is that RMFL outperforms MFL on MNIST, CIFAR10, and CIFAR100 across multiple heterogeneity levels. This comparison is independent of the method's definition: no parameter is fitted to the reported accuracy or F1-scores, and the learning rate is tuned for both MFL and RMFL under the same protocol. The motivating observation of increasing bias is supported by the authors' diagnostic measurements (Figs. 4 and 5), which are separate from the final benchmark comparison, so the empirical success is not used as evidence for the premise in a way that makes the claim self-validating. There is a notable internal inconsistency: Eq. (3), as written, does not actually realize the claimed reverse exponential decay, since for beta = 0.9 the most recent gradient receives beta/(1-beta) = 9 times the weight of the immediately preceding gradient. However, this is a correctness or specification flaw, not circularity: the empirical comparison still tests the formula that is stated, even if that formula conflicts with the prose. No self-citation is load-bearing, no uniqueness theorem is imported, and no known result is merely renamed. Therefore the paper does not exhibit the circularity patterns targeted by this review.
Assumptions & free parameters
free parameters (2)
- Learning rate =
not disclosed per setting
- Momentum coefficient β =
0.9
assumptions (2)
- domain assumption Gradient bias increases monotonically with the number of local training steps
- domain assumption Dirichlet-distributed class splits with α ∈ {1, 0.1, 0.01} adequately simulate realistic non-IID conditions
Cite this review
Pith. "Pith review of Rethinking the initialization of Momentum in Federated Learning with Heterogeneous Data." pith.science (2026). https://pith.science/paper/CW4ISMSN
@misc{pith2026241119798,
author = {Pith},
title = {Pith review of: Rethinking the initialization of Momentum in Federated Learning with Heterogeneous Data},
year = {2026},
howpublished = {\url{https://pith.science/paper/CW4ISMSN}},
note = {Machine review of arXiv:2411.19798}
}
read the original abstract
Data Heterogeneity is a major challenge of Federated Learning performance. Recently, momentum based optimization techniques have beed proved to be effective in mitigating the heterogeneity issue. Along with the model updates, the momentum updates are transmitted to the server side and aggregated. Therefore, the local training initialized with a global momentum is guided by the global history of the gradients. However, we spot a problem in the traditional cumulation of the momentum which is suboptimal in the Federated Learning systems. The momentum used to weight less on the historical gradients and more on the recent gradients. This however, will engage more biased local gradients in the end of the local training. In this work, we propose a new way to calculate the estimated momentum used in local initialization. The proposed method is named as Reversed Momentum Federated Learning (RMFL). The key idea is to assign exponentially decayed weights to the gradients with the time going forward, which is on the contrary to the traditional momentum cumulation. The effectiveness of RMFL is evaluated on three popular benchmark datasets with different heterogeneity levels.
Figures
Figures from the paper (16 more)
Reference graph
Works this paper leans on
-
[1]
Communication-efficient learning of deep networks from decentralized data,
H. B. McMahan, E. Moore, D. Ramage, S. Hampson, and B. A. y Arcas, “Communication-efficient learning of deep networks from decentralized data,” Proceedings of the 20th International Conference on Artificial Intelligence and Statistics, AISTATS 2017 , vol. 54, 2017
work page 2017
-
[2]
Accelerating federated learn- ing via momentum gradient descent,
W. Liu, L. Chen, Y . Chen, and W. Zhang, “Accelerating federated learn- ing via momentum gradient descent,” IEEE Transactions on Parallel and Distributed Systems, vol. 31, no. 8, pp. 1754–1766, 2020
2020
-
[3]
SCAFFOLD: stochastic controlled averaging for federated learning,
S. P. Karimireddy, S. Kale, M. Mohri, S. J. Reddi, S. U. Stich, and A. T. Suresh, “SCAFFOLD: stochastic controlled averaging for federated learning,” in Proceedings of the 37th International Conference on Machine Learning, ICML 2020, 13-18 July 2020, Virtual Event , ser. Proceedings of Machine Learning Research, vol. 119. PMLR, 2020, pp. 5132–5143. [Onlin...
2020
-
[4]
Feddyn: A dynamic and efficient federated distillation approach on recommender system,
C. Jin, X. Chen, Y . Gu, and Q. Li, “Feddyn: A dynamic and efficient federated distillation approach on recommender system,” in 28th IEEE International Conference on Parallel and Distributed Systems, ICPADS 2022, Nanjing, China, January 10-12, 2023 . IEEE, 2022, pp. 786–793. [Online]. Available: https://doi.org/10.1109/ICPADS56603.2022.00107
arXiv 2022
-
[5]
Adaptive federated optimization,
S. J. Reddi, Z. Charles, M. Zaheer, Z. Garrett, K. Rush, J. Kone ˇcný, S. Kumar, and H. B. McMahan, “Adaptive federated optimization,” in 9th International Conference on Learning Representations, ICLR 2021, Virtual Event, Austria, May 3-7, 2021 . OpenReview.net, 2021. [Online]. Available: https://openreview.net/forum?id=LkFG3lB13U5
work page 2021
-
[6]
Fedcm: Federated learning with client-level momentum,
J. Xu, S. Wang, L. Wang, and A. C.-C. Yao, “Fedcm: Federated learning with client-level momentum,” arXiv preprint arXiv:2106.10874 , 2021
arXiv 2021
-
[7]
Faster adaptive federated learning,
X. Wu, F. Huang, Z. Hu, and H. Huang, “Faster adaptive federated learning,” in Thirty-Seventh AAAI Conference on Artificial Intelligence, AAAI 2023, Thirty-Fifth Conference on Innovative Applications of Artificial Intelligence, IAAI 2023, Thirteenth Symposium on Educational Advances in Artificial Intelligence, EAAI 2023, Washington, DC, USA, February 7-14...
doi:10.1609/aaai.v37 2023
-
[8]
Communication-efficient adaptive federated learning,
Y . Wang, L. Lin, and J. Chen, “Communication-efficient adaptive federated learning,” in International Conference on Machine Learning, ICML 2022, 17-23 July 2022, Baltimore, Maryland, USA , ser. Proceedings of Machine Learning Research, K. Chaudhuri, S. Jegelka, L. Song, C. Szepesvári, G. Niu, and S. Sabato, Eds., vol
work page 2022
Show all 13 references
-
[9]
The mnist database of handwritten digit images for machine learning research,
L. Deng, “The mnist database of handwritten digit images for machine learning research,” IEEE Signal Processing Magazine , vol. 29, no. 6, pp. 141–142, 2012
2012
-
[10]
Learning multiple layers of features from tiny images,
A. Krizhevsky, “Learning multiple layers of features from tiny images,” University of Toronto, Technical Report TR-2009, 2009
2009
-
[11]
Gradient-based learning applied to document recognition,
Y . LeCun, L. Bottou, Y . Bengio, and P. Haffner, “Gradient-based learning applied to document recognition,” Proceedings of the IEEE , vol. 86, no. 11, pp. 2278–2324, 1998
1998
-
[12]
Deep residual learning for image recognition,
K. He, X. Zhang, S. Ren, and J. Sun, “Deep residual learning for image recognition,” in 2016 IEEE Conference on Computer Vision and Pattern Recognition, CVPR 2016, Las Vegas, NV , USA, June 27-30, 2016 . IEEE Computer Society, 2016, pp. 770–778. [Online]. Available: https://do...
2016 doi
-
[162]
22 802–22 838
PMLR, 2022, pp. 22 802–22 838. [Online]. Available: https: //proceedings.mlr.press/v162/wang22o.html
2022
Reviewed August 12, 2026 · model on record in the stance chip above.
Discussion (0). Continue with ORCID to comment.