REVIEW 5 major objections 4 minor 30 references
Efficient Zeroth-Order Federated Finetuning of Language Models on Resource-Constrained Devices
T0 review · 5 major / 4 minor · reviewed 2026-08-07 · deepseek-v4-flash
Pith's one-line read Splitting a language model into two perturbation blocks cuts zeroth-order federated finetuning computation by 1.6x to 3x while keeping memory near inference-level.
desk verdict The split-perturbation idea is real, but the published algorithm has a seed-list swap that makes the reported results unreproducible; needs major revision. 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 mechanism is split-perturbation zero-order gradient estimation with activation reuse. Writing the model as $y=f_2(\theta_2; f_1(\theta_1; B))$, FedSPZO perturbs only $f_1$ in the two directions of a central difference, stores its final-layer output $y_l$, then perturbs $f_2$ $P_2$ times against that stored output. The scalar gradient for $f_1$ averages loss differences over all second-block perturbations, while $f_2$'s gradient accumulates the usual per-perturbation scalars. This makes the forward cost $2\cdot\mathrm{fw}_1\times P_1 + 2\cdot\mathrm{fw}_2\times P_2$ instead of $2\times\mathrm{fw}_{\mathrm{total}}\times P$, which is what yields the claimed reduction.
What would settle it
Measure the total FLOPs FedSPZO needs to reach a target loss on a held-out model (e.g., OPT-1.3B or LLaMA-3-3.2B) and a held-out dataset, then compare with FedZO and DecomFL using the same perturbation budgets; if the ratio falls below 1.6x, or the overhead of caching and re-perturbing the split exceeds the reuse savings, the central claim does not transfer.
Extended reading notes
Core claim
FedSPZO shows that the computational bottleneck of zero-order federated finetuning can be reduced by treating the model as two blocks with different perturbation budgets instead of one uniform parameter vector. With $P_1=2$ perturbations on the earlier block and $P_2=8$ on the later block, the method estimates gradients for both blocks from a single set of forward evaluations: the first block is perturbed in the two central-difference directions, its output at the split is cached, and the second block is perturbed many times against that cached output. The paper reports 1.6x-3x less computation than FedZO and DecomFL to reach the same loss on SST2, RTE, and WIC with RoBERTa-large, while sending only scalar gradients (with regenerable seeds) to the server and keeping peak memory within 0.44% of the MeZO baseline.
Load-bearing premise
The claimed 1.6x-3x computation reduction rests on $P_1=2$ and $P_2=8$ being chosen by grid search on the same datasets used for evaluation, so the split may not carry the same gains to other models or tasks.
Editorial extensions
If this is right
- Zero-order federated finetuning becomes more practical on resource-constrained edge devices because the dominant per-step cost (forward passes under many perturbations) is reduced.
- The memory advantage of MeZO-style training is preserved: only the activation at the split point is stored, adding less than 0.5% to peak memory.
- Communication stays at the scale of a few scalar gradients per local step, since the server can regenerate the perturbation seeds and reconstruct each client's model without forward passes.
- Central-difference estimation, which is more accurate than forward difference, becomes affordable, giving accuracy close to first-order FedAvg while keeping communication negligible.
Reading between the lines
- The split point and perturbation counts could be chosen per model or per layer based on FLOPs and parameter counts; the paper fixes $P_1=2$, $P_2=8$ by grid search on the evaluation tasks.
- The abstract advertises experiments on OPT-1.3B and LLaMA-3-3.2B, while the detailed body experiments are for RoBERTa-large; confirming the 1.6x-3x reduction on those larger models would test whether the split transfers.
- The two-block idea extends naturally to more than two blocks or to block-specific perturbation schedules, though each additional split stores another intermediate activation that must be weighed against the reuse savings.
Editorial analysis
A structured set of objections, weighed in public.
Referee Report
Summary. The paper proposes FedSPZO, a federated zero-order finetuning method that splits the model into two consecutive blocks, allocates P1 perturbation directions to the first block and P2 perturbation directions to the second block, reuses the first block's output activations to avoid recomputing them for every second-block perturbation, and communicates only gradient scalars (and optionally seeds) to the server. The central claims are a 1.6-3x reduction in computation relative to FedZO and DecomFL, memory close to MeZO, and accuracy close to first-order FedAvg. The experimental section reports RoBERTa-large results on SST2, RTE, and WIC, with a memory comparison against FedAvg and LoRA.
Significance. If correct, FedSPZO would be a useful contribution to zero-order federated finetuning on resource-constrained devices, and the idea of reusing intermediate activations across perturbations is sensible. The paper provides a complete algorithmic skeleton, a memory formula, and comparisons to relevant ZO baselines. However, the published Algorithm 2 contains internal inconsistencies that prevent reproduction of the client update and server reconstruction, the abstract promises evaluations on OPT1.3B and LLaMa-3-3.2B that do not appear in the results, and the main efficiency numbers rely on perturbation counts chosen on the same evaluation tasks. These issues currently outweigh the strengths.
major comments (5)
- [Algorithm 2, lines 22-23] The client update passes the wrong seed lists to Update. After the P1 loop, sl1 has length P1=2 and sl2 has length P2=8 (because each Forward call appends Ps seeds and there are two Forward calls per P1 iteration). Line 22 calls Update(θ2, sl1, g2/P2, µ, P2), which indexes beyond sl1, and line 23 calls Update(θ1, sl2, g1/P1, µ, P1), updating the first block in directions that were generated for the second block. The model updated on the client is therefore not the model the server reconstructs in Algorithm 1, and the reported loss/accuracy curves are not tied to the published procedure. The obvious fix is θ1 ← Update(θ1, sl1, g1/P1, µ, P1) and θ2 ← Update(θ2, sl2, g2/P2, µ, P2), but as written the algorithm is not executable.
- [Algorithm 2, lines 12 and 41] The gradient computation for the first block is also not well defined as written. L+ and L- are allocated with length 2Ps, but Forward writes L±[pj] and L±[pj+1] inside a loop over p_j in Ps. For Ps>1, consecutive iterations overwrite odd-index entries and the arrays are never filled, so Eq. (5) and line 17 do not compute the intended g1. The loop should write to indices 2*p_j and 2*p_j+1 (or use a loop over 2Ps) to produce the 2Ps loss values assumed by the formula.
- [Abstract and Section 4.1] The abstract claims evaluations on OPT1.3B and LLaMa-3-3.2B models, but the experimental section only reports RoBERTa-large results. Section 4.1 additionally lists BOOLQ as one of four tasks, yet Table 1 and Figure 4 contain only SST2, WIC, and RTE. The paper should either add the promised results or revise the abstract and setup to state exactly which models and tasks were used.
- [Section 4.1] The central computation savings are not robust as presented because P1=2 and P2=8 were selected by testing combinations on the same tasks used for evaluation, and no sensitivity analysis is given across model sizes, tasks, or block boundaries. The same issue applies to the FedZO baseline, whose P=5 is also grid-searched on these tasks. Without ablations or a theoretical rationale for the split, the 1.6-3x reduction is a tuned empirical quantity rather than a demonstrated property of the method.
- [Section 3.2.2, Eq. (7)] Eq. (7) accounts only for forward-pass FLOPs, but the reported GFLOPs comparisons in Figure 4 must also include the perturbation loops (Perturb operations over every parameter) and the parameter update loops, which the text only hand-waves as 'divided similarly.' Please state precisely which terms are included in the reported FLOPs and confirm that Eq. (7) is the complete cost model used in the experiments; otherwise the comparison with FedZO and DecomFL is not reproducible.
minor comments (4)
- [Eq. (5) and Algorithm 2, line 17] The symbol P2s is used without definition; it appears to mean P_s^2, but the paper should define it explicitly and align it with the loop bounds 0..2Ps in Eq. (5).
- [Section 4.2, Figure 3] The comparison with FedAvg(LoRA) should state the batch sizes, local steps, and context lengths for both methods, since Figure 2 shows different batch sizes for FedAvg and FedSPZO and this affects the GFLOPs comparison.
- [Section 4.3] The sentence 'FedZO use more 1.04x computation' is missing a comparison target; it should say 'FedZO uses 1.04x more computation per round than FedSPZO.'
- [FedSPZO Communication Extension] The extension claims the server can regenerate seeds from a single initial seed, but Algorithm 2 samples s1, s2, and sh inside the client loop. The paper should specify the deterministic seed schedule or state that the reported experiments use the basic mode with seeds uploaded.
Circularity Check
No circularity: FedSPZO's 1.6-3x computation reduction is an empirical FLOPs comparison with tuned hyperparameters, not a derivation equivalent to its inputs.
full rationale
The central claim is that splitting the network into two blocks and using P1 and P2 perturbations per block lowers zero-order FL computation. This is supported by the FLOPs accounting in Eq. (7): the forward cost is 2*fw1*P1 + 2*fw2*P2, which is an arithmetic consequence of the number of perturbed forward passes, not a quantity defined in terms of the reported speedup. The values P1=2 and P2=8 are selected in Section 4.1 by testing combinations and keeping those with the highest computational savings on the same evaluation tasks; this is a hyperparameter-tuning and generalization concern, not a circular reduction, because the FLOPs comparison is an independent empirical measurement and the method does not define its output in terms of its inputs. The seed trick is adopted from external prior work (Malladi et al., 2023), not from a self-citation chain, and no uniqueness theorem or ansatz is imported from the authors' own prior papers. One internal inconsistency does appear in Algorithm 2: lines 22-23 call Update(theta2, sl1, ...) and Update(theta1, sl2, ...), swapping the seed lists built for the two blocks, which would also index beyond sl1 for the theta2 update when P2=8. This is a correctness/reproducibility defect, but it is not circularity: it does not make the reported reduction equal to an input by construction. Accordingly, the derivation chain is self-contained and no circular step is present.
Assumptions & free parameters
free parameters (3)
- P1 (first-block perturbation count) =
2
- P2 (second-block perturbation count) =
8
- FedZO baseline perturbation count =
5
assumptions (3)
- domain assumption The pseudo-random generator with a fixed seed reproduces identical perturbation vectors on client and server.
- ad hoc to paper The gradient estimator for the first block, Eq. (5), is an unbiased or sufficiently low-noise estimate of the gradient along z1.
- ad hoc to paper The computational cost model in Eq. (7), 2*fw1*P1 + 2*fw2*P2, accounts for all forward passes in the algorithm.
Cite this review
Pith. "Pith review of Efficient Zeroth-Order Federated Finetuning of Language Models on Resource-Constrained Devices." pith.science (2026). https://pith.science/paper/ELF5M7OP
@misc{pith2026250210239,
author = {Pith},
title = {Pith review of: Efficient Zeroth-Order Federated Finetuning of Language Models on Resource-Constrained Devices},
year = {2026},
howpublished = {\url{https://pith.science/paper/ELF5M7OP}},
note = {Machine review of arXiv:2502.10239}
}
abstract
Federated Learning (FL) is a promising paradigm for finetuning Large Language Models (LLMs) across distributed data sources while preserving data privacy. However, finetuning such large models is challenging on edge devices due to its high resource demand. Zeroth-order Optimization (ZO) estimates gradients through finite-difference approximations, which rely on function evaluations under random perturbations of the model parameters. Consequently, ZO with task alignment provides a potential solution, allowing finetuning using only forward passes with inference-level memory requirements and low communication overhead, but it suffers from slow convergence and higher computational demand. In this paper, we propose a new ZO-based method that applies a more efficient technique to reduce the computational demand associated with using a large number of perturbations while preserving their convergence benefits. This is achieved by splitting the model into consecutive blocks and allocating a higher number of perturbations to the second block, enabling efficient reuse of intermediate activations to update the full network with fewer forward evaluations. Our evaluation on RoBERTa-large, OPT1.3B, LLaMa-3-3.2B models shows up to $3\times$ reduction in computation compared to the other ZO-based techniques, while retaining the memory and communication benefits over first-order federated learning techniques.
Figures
Reference graph
Works this paper leans on
-
[1]
write newline
" write newline "" before.all 'output.state := FUNCTION n.dashify 't := "" t empty not t #1 #1 substring "-" = t #1 #2 substring "--" = not "--" * t #2 global.max substring 't := t #1 #1 substring "-" = "-" * t #2 global.max substring 't := while if t #1 #1 substring * t #2 global.max substring 't := if while FUNCTION format.date year duplicate empty "emp...
-
[2]
Fedrolex: Model-heterogeneous federated learning with rolling sub-model extraction
Alam, S., Liu, L., Yan, M., and Zhang, M. Fedrolex: Model-heterogeneous federated learning with rolling sub-model extraction. In Advances in Neural Information Processing Systems, volume 35, pp.\ 29677--29690, 2022
work page 2022
-
[3]
SL o RA : Federated parameter efficient fine-tuning of language models
Babakniya, S., Elkordy, A., Ezzeldin, Y., Liu, Q., Song, K.-B., EL-Khamy, M., and Avestimehr, S. SL o RA : Federated parameter efficient fine-tuning of language models. In International Workshop on Federated Learning in the Age of Foundation Models in Conjunction with NeurIPS 2023, 2023
work page 2023
-
[4]
Baydin, A. G., Pearlmutter, B. A., Syme, D., Wood, F., and Torr, P. Gradients without backpropagation. arXiv preprint arXiv:2202.08587, 2022
arXiv 2022
-
[5]
R., Angeli, G., Potts, C., and Manning, C
Bowman, S. R., Angeli, G., Potts, C., and Manning, C. D. A large annotated corpus for learning natural language inference. In M \`a rquez, L., Callison-Burch, C., and Su, J. (eds.), Proceedings of the 2015 Conference on Empirical Methods in Natural Language Processing, pp.\ 632--642, Lisbon, Portugal, September 2015. Association for Computational Linguist...
-
[6]
A zeroth-order block coordinate descent algorithm for huge-scale black-box optimization
Cai, H., Lou, Y., McKenzie, D., and Yin, W. A zeroth-order block coordinate descent algorithm for huge-scale black-box optimization. In International Conference on Machine Learning, pp.\ 1193--1203. PMLR, 2021
work page 2021
-
[7]
Caldas, S., Kone c ny, J., McMahan, H. B., and Talwalkar, A. Expanding the reach of federated learning by reducing client resource requirements. arXiv:1812.07210, 2018
arXiv 2018
-
[8]
B ool Q : Exploring the surprising difficulty of natural yes/no questions
Clark, C., Lee, K., Chang, M.-W., Kwiatkowski, T., Collins, M., and Toutanova, K. B ool Q : Exploring the surprising difficulty of natural yes/no questions. In Proceedings of the 2019 Conference of the North A merican Chapter of the Association for Computational Linguistics: Human Language Technologies, Volume 1 (Long and Short Papers) , pp.\ 2924--2936, ...
Show all 30 references
-
[9]
N., and Zhou, Y
Fang, W., Yu, Z., Jiang, Y., Shi, Y., Jones, C. N., and Zhou, Y. Communication-efficient stochastic zeroth-order optimization for federated learning. IEEE Transactions on Signal Processing, 70: 0 5058--5073, 2022
2022
-
[10]
Does federated learning really need backpropagation
Feng, H., Pang, T., Du, C., Chen, W., Yan, S., and Lin, M. Does federated learning really need backpropagation. arXiv preprint arXiv:2301.12195, 2023
2023 arXiv
-
[11]
Making pre-trained language models better few-shot learners
Gao, T., Fisch, A., and Chen, D. Making pre-trained language models better few-shot learners. In Zong, C., Xia, F., Li, W., and Navigli, R. (eds.), Proceedings of the 59th Annual Meeting of the Association for Computational Linguistics and the 11th International Joint Conferen...
2021
-
[12]
J., Shen, Y., Wallis, P., Allen-Zhu, Z., Li, Y., Wang, S., Wang, L., and Chen, W
Hu, E. J., Shen, Y., Wallis, P., Allen-Zhu, Z., Li, Y., Wang, S., Wang, L., and Chen, W. Lo RA : Low-rank adaptation of large language models. In International Conference on Learning Representations, 2022
2022
-
[13]
Imteaj, A., Thakker, U., Wang, S., Li, J., and Amini, M. H. A survey on federated learning for resource-constrained iot devices. IEEE Internet of Things Journal, 9 0 (1): 0 1--24, 2022. doi:10.1109/JIOT.2021.3095077
2022
-
[14]
Achieving dimension-free communication in federated learning via zeroth-order optimization
Li, Z., Ying, B., Liu, Z., Dong, C., and Yang, H. Achieving dimension-free communication in federated learning via zeroth-order optimization. arXiv preprint arXiv:2405.15861, 2024
2024 arXiv
-
[15]
On the convergence of zeroth-order federated tuning for large language models
Ling, Z., Chen, D., Yao, L., Li, Y., and Shen, Y. On the convergence of zeroth-order federated tuning for large language models. In Proceedings of the 30th ACM SIGKDD Conference on Knowledge Discovery and Data Mining, pp.\ 1827--1838, 2024
2024
-
[16]
Roberta: A robustly optimized bert pretraining approach
Liu, Y. Roberta: A robustly optimized bert pretraining approach. arXiv preprint arXiv:1907.11692, 364, 2019
1907 arXiv
-
[17]
D., Chen, D., and Arora, S
Malladi, S., Gao, T., Nichani, E., Damian, A., Lee, J. D., Chen, D., and Arora, S. Fine-tuning language models with just forward passes. Advances in Neural Information Processing Systems, 36: 0 53038--53075, 2023
2023
-
[18]
McMahan, B., Moore, E., Ramage, D., Hampson, S., and y Arcas, B. A. Communication-efficient learning of deep networks from decentralized data. In Artificial intelligence and statistics, pp.\ 1273--1282. PMLR, 2017
2017
-
[19]
Black-box generalization: Stability of zeroth-order learning
Nikolakakis, K., Haddadpour, F., Kalogerias, D., and Karbasi, A. Black-box generalization: Stability of zeroth-order learning. In Koyejo, S., Mohamed, S., Agarwal, A., Belgrave, D., Cho, K., and Oh, A. (eds.), Advances in Neural Information Processing Systems, volume 35, pp.\ ...
2022
-
[20]
Thinking forward: Memory-efficient federated finetuning of language models
Panchal, K., Parikh, N., Choudhary, S., Zhang, L., Brun, Y., and Guan, H. Thinking forward: Memory-efficient federated finetuning of language models. arXiv preprint arXiv:2405.15551, 2024
2024 arXiv
-
[21]
Aggregating capacity in fl through successive layer training for computationally-constrained devices
Pfeiffer, K., Khalili, R., and Henkel, J. Aggregating capacity in fl through successive layer training for computationally-constrained devices. In Oh, A., Naumann, T., Globerson, A., Saenko, K., Hardt, M., and Levine, S. (eds.), Advances in Neural Information Processing System...
2023
-
[22]
Federated learning for computationally constrained heterogeneous devices: A survey
Pfeiffer, K., Rapp, M., Khalili, R., and Henkel, J. Federated learning for computationally constrained heterogeneous devices: A survey. ACM Computing Surveys, 55 0 (14s): 0 1--27, 2023 b
2023
-
[23]
Pilehvar, M. T. and Camacho-Collados, J. W i C : the word-in-context dataset for evaluating context-sensitive meaning representations. In Proceedings of the 2019 Conference of the North A merican Chapter of the Association for Computational Linguistics: Human Language Technolo...
2019 doi
-
[24]
Federated full-parameter tuning of billion-sized language models with communication cost under 18 kilobytes
Qin, Z., Chen, D., Qian, B., Ding, B., Li, Y., and Deng, S. Federated full-parameter tuning of billion-sized language models with communication cost under 18 kilobytes. In Proceedings of the 41st International Conference on Machine Learning, ICML'24. JMLR.org, 2024
2024
-
[25]
Shi, Y., Yang, K., Jiang, T., Zhang, J., and Letaief, K. B. Communication-efficient edge ai: Algorithms and systems. IEEE Communications Surveys & Tutorials, 22 0 (4): 0 2167--2191, 2020
2020
-
[26]
D., Ng, A
Socher, R., Perelygin, A., Wu, J., Chuang, J., Manning, C. D., Ng, A. Y., and Potts, C. Recursive deep models for semantic compositionality over a sentiment treebank. In Proceedings of the 2013 conference on empirical methods in natural language processing, pp.\ 1631--1642, 2013
2013
-
[27]
Spall, J. C. Multivariate stochastic approximation using a simultaneous perturbation gradient approximation. IEEE transactions on automatic control, 37 0 (3): 0 332--341, 1992
1992
-
[28]
Compressing rnns for iot devices by 15-38x using kronecker products
Thakker, U., Beu, J., Gope, D., Zhou, C., Fedorov, I., Dasika, G., and Mattina, M. Compressing rnns for iot devices by 15-38x using kronecker products. arXiv:1906.02876, 2019
1906 arXiv
-
[29]
Progfed: effective, communication, and computation efficient federated learning by progressive training
Wang, H.-P., Stich, S., He, Y., and Fritz, M. Progfed: effective, communication, and computation efficient federated learning by progressive training. In International Conference on Machine Learning, pp.\ 23034--23054. PMLR, 2022
2022
-
[30]
FwdLLM : Efficient federated finetuning of large language models with perturbed inferences
Xu, M., Cai, D., Wu, Y., Li, X., and Wang, S. FwdLLM : Efficient federated finetuning of large language models with perturbed inferences. In 2024 USENIX Annual Technical Conference (USENIX ATC 24), pp.\ 579--596, 2024
2024
Reviewed August 7, 2026 · model on record in the stance chip above.
Discussion (0). Continue with ORCID to comment.