REVIEW 3 major objections 5 minor 42 references
Learn How to Query from Unlabeled Data Streams in Federated Learning
T0 review · 3 major / 5 minor · reviewed 2026-08-11 · deepseek-v4-flash
Pith's one-line read LeaDQ treats federated data querying as a multi-agent decision problem and learns local selection policies that improve global model accuracy beyond current federated active learning baselines.
desk verdict Streaming federated active learning cast as a Dec-POMDP with QMIX; a solid contribution whose evaluation is undermined by an un-costed labeled held-out set at the server. 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 mechanism is the Dec-POMDP/QMIX combination. A Dec-POMDP formalizes the fact that each client chooses which samples to query under partial observation—only its own logits—while the objective is joint. QMIX provides the training-time bridge between local policies and the global goal: the joint Q-value is Qtot = ψ(Q1, ..., QK, sr), a monotonic function of the local Q-values conditioned on the global state sr, and the network is updated by a temporal-difference loss against the shared held-out accuracy reward. This is what lets the algorithm train locally executable policies against a signal no single client can see, without ever moving raw data off the clients.
What would settle it
Run the paper's SVHN setup with a mismatched held-out set (for example, CIFAR-100 in place of SVHN); if the global state and reward lose all information about the target distribution, the proposed method should collapse to baseline-level accuracy if the central claim about learning from global signals is right.
Extended reading notes
Core claim
The paper's central discovery is that stream-based federated active learning can be cast as a Dec-POMDP in which every client is an agent. Each agent's local observation is the current global model's predictive logits on the newly arrived unlabeled samples, its action is the choice of which Nq samples to label, and all agents share the reward Rr = Acc(θr; Dheld) − Acc(θr−1; Dheld), the change in held-out accuracy after training on the newly queried labels. Training uses the QMIX factorization: a monotonic mixing network combines local Q-values with a global state (the model's maximum prediction confidence on the held-out set) into a joint Q-value, so each client can learn a decentralized policy that is nevertheless optimized for the global objective. The accompanying motivating experiment shows where the benefit comes from: a Global Coreset that coordinates selections across clients clearly beats Local Coreset, and LeaDQ approaches the global oracle's accuracy while keeping data decentralized.
Load-bearing premise
The method assumes the server has a labeled held-out dataset that follows the target distribution, because that dataset is the only source of both the global state signal and the reward used to train the querying policies; without it, there is no learning signal.
Editorial extensions
If this is right
- In stream-based federated active learning, a learned query policy can beat both generic active-learning heuristics (uncertainty, coreset) and federated-specific baselines (LoGo, KAFAL) under non-IID client data.
- Because execution only requires the current model's logits and the local policy network, the querying decisions stay decentralized and compatible with FedAvg-style privacy constraints.
- The method remains effective when the number of clients grows to 50 and when data heterogeneity is varied in both distribution-based and quantity-based ways.
- The learned policies select samples whose label and feature distributions resemble the target global distribution rather than the local client distribution, which is the mechanism behind the accuracy gains.
- The framework can be adapted to different arrival volumes and query budgets by changing the action constraint |a| = Nq, and its performance is stable across episode-length hyperparameters.
Reading between the lines
- The method's practical reach depends on the labeled held-out set; a deployment without a representative D_held would need a proxy signal (for example, confidence calibrated on a small validation slice or a self-supervised estimate), and that substitution is the most direct testable extension.
- Since the reward is measured after full FedAvg rounds, policy training is expensive; computing the reward on a small validation batch or every few rounds would make the approach more plausible in bandwidth-limited settings.
- The QMIX monotonicity constraint limits the interaction structure among clients; richer factorizations could capture cases where one client's querying decision should negatively influence another's, which the current formulation rules out.
Signed reviews
Editorial analysis
A structured set of objections, weighed in public.
Referee Report
Summary. The paper studies active data querying in federated learning when unlabeled samples arrive at clients as a data stream. It formulates the querying process as a Dec-POMDP and proposes LeaDQ, a QMIX-based multi-agent reinforcement learning approach in which each client's local policy selects N_q samples per round from its arrived unlabeled stream, while a server-side mixing network coordinates the local Q-functions using a global state derived from a held-out dataset. The reward is the one-round change in held-out accuracy. Experiments on SVHN, CIFAR-100, 20Newsgroup, and Tiny-ImageNet compare LeaDQ with Uncertainty, Coreset, LoGo, and KAFAL, reporting consistent accuracy improvements.
Significance. If the claims hold, LeaDQ is a useful step toward principled, learned querying policies for streaming federated active learning. The paper provides a clean problem formulation, a reasonable CTDE instantiation via QMIX, a public code release, a motivating experiment, and evaluations across image and text tasks with non-IID data. The central weakness is that the policy-training signal in Eq. (10) requires ground-truth labels for a server-held dataset of size 1000 (Table 6), and this annotation cost is never charged to LeaDQ's budget nor given to the baselines. In addition, the empirical evidence rests on three-seed averages without error bars or significance tests, and one state-of-the-art baseline (KAFAL) collapses in two of the reported settings without explanation. These issues currently prevent the paper from establishing its headline claim of outperforming the benchmarking algorithms under an equal annotation budget.
major comments (3)
- [Section 5.1, Eq. (10), Table 6] The reward R^r = Acc(θ^r; D_held) − Acc(θ^{r−1}; D_held) explicitly requires ground-truth labels for the held-out dataset D_held, whose size is listed as 1000 in Table 6. The server uses these labels to train the query policy, while the uncertainty, coreset, LoGo, and KAFAL baselines receive no analogous labeled validation set. Because the paper's motivating problem treats labels as expensive and scarce, the reported accuracy gains may come from the extra labeled data used only by LeaDQ rather than from the learned querying policy itself. Please either remove the label requirement (e.g., use an unlabeled proxy for the reward), add an equal-annotation-budget comparison in which the baselines also receive the same labeled held-out data for their own tuning, or explicitly justify why the D_held labels should not be counted. In addition, no experiment varies the size or composition of D_held, so overfitting of the policy to that particular held-out set is unexplored.
- [Appendix B.2, Fig. 4, Tables 2–4] The main results are averaged over three random seeds with no error bars, standard deviations, or significance tests. Several of the reported margins are extremely small or even negative: in Table 3 (α=1.0) LeaDQ is 66.50% versus KAFAL's 66.48%, and in Table 2 (Nq/Nu=2/10) LeaDQ is 76.28% versus KAFAL's 76.32%. Without per-seed results or a statistical comparison, the claim that LeaDQ consistently outperforms the baselines is not supported, especially for the later-round curves in Fig. 4 where the plotted lines appear close.
- [Table 4 and Table 7] KAFAL, a state-of-the-art federated active learning baseline, collapses to roughly 20% accuracy in the quantity-skew setting (Table 4) and on Tiny-ImageNet (Table 7), while it performs competitively elsewhere (e.g., 64.67% in Table 3 with α=0.5). The paper does not explain this collapse or provide any diagnostic, such as sensitivity to hyperparameters or a check of whether the implementation is appropriate for these setups. If KAFAL is not properly configured for these scenarios, the comparison is unfair; if it genuinely fails, that is an interesting finding that should be analyzed. Please add an explanation or ablation, or remove the affected rows with justification.
minor comments (5)
- [Fig. 4, Section 6.2] There are typos: '20Newsgrpup' in the Fig. 4 subplot label, 'dsitribution' in Section 6.2, and 'Global Corset' in the Fig. 3 caption should be 'Global Coreset'.
- [Table 5] The third column header of Table 5 reads 'SVHN' but the column reports the DistilBERT model and should be labeled '20Newsgroup'.
- [Appendix B.1] The sentence 'LeCun et al. 1998.' is missing its closing parenthesis; it should be 'LeCun et al. 1998)'.
- [Table 5] The row 'Total arrived samples in Table 1' is confusing: Table 1 is the dataset summary, not the implementation-detail table. Please renumber or reword this row so it refers to the correct table.
- [Section 5.1] The Dec-POMDP tuple is written as ⟨S, A, P, R, O, K, γ⟩, which omits the observation function and the number of agents in the standard notation. This is a minor formal clarity issue, but the tuple should be defined consistently with the text that follows.
Circularity Check
No circular derivation: LeaDQ's RL reward is an optimization signal, and reported test accuracy is measured on external benchmarks.
full rationale
The paper's derivation chain does not reduce to its own inputs. The query policy is trained by multi-agent RL with reward R^r = Acc(theta^r; D_held) - Acc(theta^{r-1}; D_held) (Eq. 10) and global state given by prediction confidence on D_held (Eq. 8), but the claimed outcome is model accuracy on held-out test sets of SVHN, CIFAR-100, 20Newsgroup, and Tiny-ImageNet, which are external to the training reward. This is a standard RL objective, not a fitted parameter renamed as a prediction. The paper compares LeaDQ against Uncertainty, Coreset, LoGo, and KAFAL under the same federated training protocol, so the empirical comparison is self-contained against external benchmarks. The only self-citation (Li et al. 2023, co-authored by T. Lin) appears in a related-work list and is not load-bearing. The server's labeled held-out dataset D_held is used to compute the reward and state without being counted against the annotation budget; this is a fairness and external-validity limitation, not a circularity, because Eq. 10 is not an identity that by construction guarantees the reported test accuracy. Accordingly, no circular step is identified.
Assumptions & free parameters
free parameters (4)
- discount factor gamma =
0.99
- episode length J =
10
- warm-up timesteps =
32
- held-out dataset size =
1000
assumptions (4)
- domain assumption The server holds a labeled dataset D_held following the target distribution.
- domain assumption Aggregated client data distribution equals the target distribution P.
- domain assumption Predictive logits are a sufficient observation for selecting informative samples.
- standard math The QMIX monotonic mixing constraint holds for this problem.
Cite this review
Pith. "Pith review of Learn How to Query from Unlabeled Data Streams in Federated Learning." pith.science (2026). https://pith.science/paper/MJ555F6X
@misc{pith2026241208138,
author = {Pith},
title = {Pith review of: Learn How to Query from Unlabeled Data Streams in Federated Learning},
year = {2026},
howpublished = {\url{https://pith.science/paper/MJ555F6X}},
note = {Machine review of arXiv:2412.08138}
}
read the original abstract
Federated learning (FL) enables collaborative learning among decentralized clients while safeguarding the privacy of their local data. Existing studies on FL typically assume offline labeled data available at each client when the training starts. Nevertheless, the training data in practice often arrive at clients in a streaming fashion without ground-truth labels. Given the expensive annotation cost, it is critical to identify a subset of informative samples for labeling on clients. However, selecting samples locally while accommodating the global training objective presents a challenge unique to FL. In this work, we tackle this conundrum by framing the data querying process in FL as a collaborative decentralized decision-making problem and proposing an effective solution named LeaDQ, which leverages multi-agent reinforcement learning algorithms. In particular, under the implicit guidance from global information, LeaDQ effectively learns the local policies for distributed clients and steers them towards selecting samples that can enhance the global model's accuracy. Extensive simulations on image and text tasks show that LeaDQ advances the model performance in various FL scenarios, outperforming the benchmarking algorithms.
Figures
Figures from the paper (3 more)
Reference graph
Works this paper leans on
-
[1]
, " * write output.state after.block = add.period write newline
ENTRY address archivePrefix author booktitle chapter edition editor eid eprint howpublished institution isbn journal key month note number organization pages publisher school series title type volume year label extra.label sort.label short.list INTEGERS output.state before.all mid.sentence after.sentence after.block FUNCTION init.state.consts #0 'before.a...
-
[2]
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 word.in bbl.in capitalize " " * FUNCT...
-
[3]
Ahn, J.; Ma, Y.; Park, S.; and You, C. 2024. Federated Active Learning (F-AL): An Efficient Annotation Strategy for Federated Learning. IEEE Access , 12: 39261--39269
work page 2024
-
[4]
Ash, J. T.; and Adams, R. P. 2020. On Warm-Starting Neural Network Training. In Advances in Neural Information Processing Systems 33: Annual Conference on Neural Information Processing Systems (NeurIPS)
work page 2020
-
[5]
Cacciarelli, D.; and Kulahci, M. 2024. Active learning for data streams: A survey. Mach. Learn., 113(1): 185--239
work page 2024
-
[6]
Cao, Y.; Shi, Y.; Yu, B.; Wang, J.; and Tao, D. 2023. Knowledge-Aware Federated Active Learning with Non-IID Data. In IEEE/CVF International Conference on Computer Vision (ICCV) , 22222--22232
work page 2023
-
[7]
Chen, Y.; Ning, Y.; Slawski, M.; and Rangwala, H. 2020. Asynchronous Online Federated Learning for Edge Devices with Non-IID Data. In 2020 IEEE International Conference on Big Data (BigData) , 15--24
work page 2020
-
[8]
Fang, M.; Li, Y.; and Cohn, T. 2017. Learning how to Active Learn: A Deep Reinforcement Learning Approach. In Proceedings of the 2017 Conference on Empirical Methods in Natural Language Processing (EMNLP)
work page 2017
Show all 42 references
-
[9]
Fang, X.; and Ye, M. 2022. Robust Federated Learning with Noisy and Heterogeneous Clients. In IEEE/CVF Conference on Computer Vision and Pattern Recognition (CVPR) , 10062--10071
2022
-
[10]
Gong, C.; Zheng, Z.; Wu, F.; Shao, Y.; Li, B.; and Chen, G. 2023. To Store or Not? Online Data Selection for Federated Learning with Limited Storage. In Ding, Y.; Tang, J.; Sequeda, J. F.; Aroyo, L.; Castillo, C.; and Houben, G., eds., Proceedings of the ACM Web Conference 2023 (WWW)
2023
-
[11]
He, K.; Zhang, X.; Ren, S.; and Sun, J. 2016. Deep Residual Learning for Image Recognition. In 2016 IEEE Conference on Computer Vision and Pattern Recognition (CVPR) , 770--778
2016
-
[12]
Huang, W.; Ye, M.; and Du, B. 2022. Learn from Others and Be Yourself in Heterogeneous Federated Learning. In IEEE/CVF Conference on Computer Vision and Pattern Recognition (CVPR) , 10133--10143
2022
-
[13]
Jin, Y.; Wei, X.; Liu, Y.; and Yang, Q. 2020. Towards utilizing unlabeled data in federated learning: A survey and prospective. arXiv:2002.11545
2020 arXiv
-
[14]
Kairouz, P.; et al. 2021. Advances and open problems in federated learning. Found. Trends Mach. Learn., 14(1--2): 1--210
2021
-
[15]
Kim, S.; Bae, S.; Song, H.; and Yun, S. 2023. Re-Thinking Federated Active Learning Based on Inter-Class Diversity. In IEEE/CVF Conference on Computer Vision and Pattern Recognition (CVPR)
2023
-
[16]
Kong, X.; Zhang, W.; Qu, Y.; Yao, X.; and Shen, G. 2024. FedAWR: An Interactive Federated Active Learning Framework for Air Writing Recognition. IEEE Trans. Mob. Comput. , 23(5): 6423--6436
2024
-
[17]
Krizhevsky, A.; and Hinton, G. 2009. Learning multiple layers of features from tiny images . [Online]. Available: https://www.cs.toronto.edu/ kriz/cifar.html
2009
-
[18]
Lang, K. 1995. NewsWeeder: Learning to Filter Netnews. In Proceedings of the 12th International Conference on Machine Learning (ICML), 331--339
1995
-
[19]
LeCun, Y.; Bottou, L.; Bengio, Y.; and Haffner, P. 1998. Gradient-based learning applied to document recognition. Proc. IEEE , 86(11): 2278--2324
1998
-
[20]
Li, Q.; Diao, Y.; Chen, Q.; and He, B. 2022. Federated Learning on Non-IID Data Silos: An Experimental Study. In 38th IEEE International Conference on Data Engineering (ICDE) , 965--978
2022
-
[21]
Li, Z.; Lin, T.; Shang, X.; and Wu, C. 2023. Revisiting weighted aggregation in federated learning with neural networks. In International Conference on Machine Learning, 19767--19788. PMLR
2023
-
[22]
Liu, P.; Wang, L.; Ranjan, R.; He, G.; and Zhao, L. 2022. A Survey on Active Deep Learning: From Model Driven to Data Driven. ACM Comput. Surv. , 54(10s): 221:1--221:34
2022
-
[23]
Marfoq, O.; Neglia, G.; Kameni, L.; and Vidal, R. 2023. Federated Learning for Data Streams. In International Conference on Artificial Intelligence and Statistics (AISTATS), 8889--8924
2023
-
[24]
McMahan, B.; Moore, E.; Ramage, D.; Hampson, S.; and y Arcas, B. A. 2017. Communication-Efficient Learning of Deep Networks from Decentralized Data. In Proceedings of the 20th International Conference on Artificial Intelligence and Statistics (AISTATS)
2017
-
[25]
Y.; et al
Netzer, Y.; Wang, T.; Coates, A.; Bissacco, A.; Wu, B.; Ng, A. Y.; et al. 2011. Reading digits in natural images with unsupervised feature learning. In NIPS workshop on deep learning and unsupervised feature learning, volume 2011, 7. Granada, Spain
2011
-
[26]
A.; and Amato, C
Oliehoek, F. A.; and Amato, C. 2016. A Concise Introduction to Decentralized POMDPs. Springer Briefs in Intelligent Systems
2016
-
[27]
S.; Farquhar, G.; Foerster, J
Rashid, T.; Samvelyan, M.; de Witt, C. S.; Farquhar, G.; Foerster, J. N.; and Whiteson, S. 2018. QMIX : Monotonic Value Function Factorisation for Deep Multi-Agent Reinforcement Learning. In Dy, J. G.; and Krause, A., eds., Proceedings of the 35th International Conference on M...
2018
-
[28]
R.; Albarqouni, S.; Bakas, S.; Galtier, M
Rieke, N.; Hancox, J.; Li, W.; Milletar \` , F.; Roth, H. R.; Albarqouni, S.; Bakas, S.; Galtier, M. N.; Landman, B. A.; Maier - Hein, K. H.; Ourselin, S.; Sheller, M. J.; Summers, R. M.; Trask, A.; Xu, D.; Baust, M.; and Cardoso, M. J. 2020. The future of digital health with ...
2020
-
[29]
Sanh, V.; Debut, L.; Chaumond, J.; and Wolf, T. 2020. DistilBERT, a distilled version of BERT: smaller, faster, cheaper and lighter. arXiv:1910.01108
2020 arXiv
-
[30]
Sener, O.; and Savarese, S. 2018. Active Learning for Convolutional Neural Networks: A Core-Set Approach. In 6th International Conference on Learning Representations (ICLR)
2018
-
[31]
Settles, B. 2009. Active learning literature survey
2009
-
[32]
Shi, J.; Wu, Y.; Zeng, D.; Tao, J.; Hu, J.; and Shi, Y. 2023 a . Self-Supervised On-Device Federated Learning From Unlabeled Streams. IEEE Trans. Comput. Aided Des. Integr. Circuits Syst. , 42(12): 4871--4882
2023
-
[33]
Shi, Y.; Liang, J.; Zhang, W.; Tan, V. Y. F.; and Bai, S. 2023 b . Towards Understanding and Mitigating Dimensional Collapse in Heterogeneous Federated Learning. In The Eleventh International Conference on Learning Representations (ICLR)
2023
-
[34]
Van der Maaten, L.; and Hinton, G. 2008. Visualizing data using t-SNE. Journal of machine learning research, 9(11)
2008
-
[35]
S.; Yeo, M.; Makhzani, A.; Küttler, H.; Agapiou, J.; Schrittwieser, J.; et al
Vinyals, O.; Ewalds, T.; Bartunov, S.; Georgiev, P.; Vezhnevets, A. S.; Yeo, M.; Makhzani, A.; Küttler, H.; Agapiou, J.; Schrittwieser, J.; et al. 2017. Starcraft II : A new challenge for reinforcement learning. arXiv:1708.04782
2017 arXiv
-
[36]
Wang, D.; and Shang, Y. 2014. A new active labeling method for deep learning. In 2014 International Joint Conference on Neural Networks (IJCNN), 112--119
2014
-
[37]
Wang, J.; Xu, W.; Gu, Y.; Song, W.; and Green, T. C. 2021. Multi-agent reinforcement learning for active voltage control on power distribution networks. Advances in Neural Information Processing Systems, 34: 3271--3284
2021
-
[38]
Wu, X.; Pei, J.; Chen, C.; Zhu, Y.; Wang, J.; Qian, Q.; Zhang, J.; Sun, Q.; and Guo, Y. 2023. Federated Active Learning for Multicenter Collaborative Disease Diagnosis. IEEE Trans. Medical Imaging , 42(7): 2068--2080
2023
-
[39]
Yang, Y.; Xiao, P.; and Ji, K. 2023. SimFBO: Towards Simple, Flexible and Communication-efficient Federated Bilevel Learning. In Advances in Neural Information Processing Systems (NeurIPS)
2023
-
[40]
Ye, R.; Xu, M.; Wang, J.; Xu, C.; Chen, S.; and Wang, Y. 2023. FedDisco: Federated Learning with Discrepancy-Aware Collaboration. In International Conference on Machine Learning (ICML)
2023
-
[41]
Zhang, T.; Gao, L.; He, C.; Zhang, M.; Krishnamachari, B.; and Avestimehr, A. S. 2022 a . Federated Learning for the Internet of Things: Applications, Challenges, and Opportunities. IEEE Internet Things Mag. , 5(1): 24--29
2022
-
[42]
Zhang, Y.; Tong, H.; Xia, Y.; Zhu, Y.; Chi, Y.; and Ying, L. 2022 b . Batch Active Learning with Graph Neural Networks via Multi-Agent Deep Reinforcement Learning. In Thirty-Sixth AAAI Conference on Artificial Intelligence (AAAI)
2022
Reviewed August 11, 2026 · model on record in the stance chip above.
Discussion (0). Continue with ORCID to comment.