PubSwap: Public-Data Off-Policy Coordination for Federated RLVR
Not on the map. Why?
Abstract (arXiv)
Reasoning post-training with reinforcement learning from verifiable rewards (RLVR) is typically studied in centralized settings, yet many realistic applications involve decentralized private data distributed across organizations. Federated training is a natural solution, but scaling RLVR in this regime is challenging: full-model synchronization is expensive, and performing many local steps can cause severe client drift under heterogeneous data. We propose a federated RLVR framework that combines LoRA-based local adaptation with public-data-based off-policy steps to improve both communication efficiency and cross-client coordination. In particular, a small shared public dataset is used to periodically exchange and reuse response-level training signals across organizations, providing a lightweight anchor toward a more globally aligned objective without exposing private data. Our method selectively replaces locally incorrect responses with globally correct ones during public-data steps, thereby keeping training closer to the local policy while still benefiting from cross-client coordination. Across mathematical and medical reasoning benchmarks and models, our method consistently improves over standard baselines. Our results highlight a simple and effective recipe for federated reasoning post-training: combining low-rank communication with limited public-data coordination.
Key facts. No citing papers in the map yet (data through 14 Oct 2026). Builds on 61 of its 68 references that are in the map. First posted on arXiv 14 Apr 2026.
Builds on
61 of 68 references.
References · 61
- 2017Proximal Policy Optimization Algorithmscited by 15,215
- 2024The Llama 3 Herd of Modelscited by 14,715
- 2024DeepSeekMath: Pushing the Limits of Mathematical Reasoning in Open Language Modelscited by 9,379
- 2025Qwen3 Technical Reportcited by 8,820
- 2016Communication-Efficient Learning of Deep Networks from Decentralized Datacited by 7,669
- 2021Measuring Mathematical Problem Solving With the MATH Datasetcited by 5,371
- 2018Federated Optimization in Heterogeneous Networkscited by 2,987
- 2025DAPO: An Open-Source LLM Reinforcement Learning System at Scalecited by 2,810
- 2024HybridFlow: A Flexible and Efficient RLHF Frameworkcited by 2,057
- 2016Federated Learning: Strategies for Improving Communication Efficiencycited by 1,475
- 2020What Disease does this Patient Have? A Large-scale Open Domain Question Answering Dataset from Medical Examscited by 1,356
- 2024Qwen2.5-Math Technical Report: Toward Mathematical Expert Model via Self-Improvementcited by 1,090
38 more
- 2020Adaptive Federated Optimizationcited by 1,046
- 2020Tackling the Objective Inconsistency Problem in Heterogeneous Federated Optimizationcited by 906
- 2019Measuring the Effects of Non-Identical Data Distribution for Federated Visual Classificationcited by 851
- 2020Ensemble Distillation for Robust Model Fusion in Federated Learningcited by 682
- 2021Federated Learning Based on Dynamic Regularizationcited by 562
- 2022MedMCQA : A Large-scale Multi-Subject Multi-Choice Dataset for Medical domain Question Answeringcited by 543
- 2019Robust and Communication-Efficient Federated Learning from Non-IID Datacited by 419
- 2025Rubrics as Rewards: Reinforcement Learning Beyond Verifiable Domainscited by 366
- 2019FedPAQ: A Communication-Efficient Federated Learning Method with Periodic Averaging and Quantizationcited by 356
- 2024DeepSeek-Coder-V2: Breaking the Barrier of Closed-Source Models in Code Intelligencecited by 331
- 2025Learning to Reason under Off-Policy Guidancecited by 281
- 2025DeepMath-103K: A Large-Scale, Challenging, Decontaminated, and Verifiable Mathematical Dataset for Advancing Reasoningcited by 280
- 2018Communication-Efficient On-Device Machine Learning: Federated Distillation and Augmentation under Non-IID Private Datacited by 243
- 2023Towards Building the Federated GPT: Federated Instruction Tuningcited by 185
- 2020FetchSGD: Communication-Efficient Federated Learning with Sketchingcited by 179
- 2025MedVLM-R1: Incentivizing Medical Reasoning Capability of Vision-Language Models (VLMs) via Reinforcement Learningcited by 173
- 2020FedSplit: An algorithmic framework for fast federated optimizationcited by 108
- 2020UVeQFed: Universal Vector Quantization for Federated Learningcited by 107
- 2025Not All Rollouts are Useful: Down-Sampling Rollouts in LLM Reinforcement Learningcited by 84
- 2025Fin-R1: A Large Language Model for Financial Reasoning through Reinforcement Learningcited by 83
- 2022Heterogeneous Ensemble Knowledge Transfer for Training Large Models in Federated Learningcited by 81
- 2025Quantization Hurts Reasoning? An Empirical Study on Quantized Reasoning Modelscited by 73
- 2023FedExP: Speeding Up Federated Averaging via Extrapolationcited by 69
- 2025Training a Scientific Reasoning Model for Chemistrycited by 56
- 2025Prosperity before Collapse: How Far Can Off-Policy RL Reach with Stale Data on LLMs?cited by 53
- 2025Reinforcement Pre-Trainingcited by 40
- 2019Limits of Private Learning with Access to Public Datacited by 39
- 2024Privacy-Preserving Instructions for Aligning Large Language Modelscited by 35
- 2025Revisiting Group Relative Policy Optimization: Insights into On-Policy and Off-Policy Trainingcited by 34
- 2023Can Public Large Language Models Help Private Cross-device Federated Learning?cited by 30
- 2025Asymmetric REINFORCE for off-Policy Reinforcement Learning: Balancing positive and negative rewardscited by 29
- 2025Beyond Correctness: Harmonizing Process and Outcome Rewards through RL Trainingcited by 29
- 2023Selective Knowledge Sharing for Privacy-Preserving Federated Distillation without A Good Teachercited by 25
- 2025Tina: Tiny Reasoning Models via LoRAcited by 25
- 2025INTELLECT-2: A Reasoning Model Trained Through Globally Decentralized Reinforcement Learningcited by 25
- 2025DianJin-R1: Evaluating and Enhancing Financial Reasoning in Large Language Modelscited by 22
- 2025PrefPalette: Personalized Preference Modeling with Latent Attributescited by 16
- 2025FlexOlmo: Open Language Models for Flexible Data Usecited by 15
50 most cited of 61 All 61 (JSON)
Built on by
Not cited yet.
Similar, not cited
Close in wording, no citation either way
- 2025Optimized Local Updates in Federated Learning via Reinforcement Learning
- 2023Federated Ensemble-Directed Offline Reinforcement Learning
- 2024Federated Offline Reinforcement Learning: Collaborative Single-Policy Coverage Suffices
- 2025FOVA: Offline Federated Reinforcement Learning with Mixed-Quality Data
- 2026RolloutPipe: Overlapping Pipelined Rollout and Training in Disaggregated On-Policy LLM Reinforcement Learning
- 2026Fed-GRPO: Reward-Signal-Driven Federated Group Relative Policy Optimization
- 2026FGRPO: Federated GRPO with Adaptive Aggregation on Non-IID Data
- 2025FedRand: Enhancing Privacy in Federated Learning with Randomized LoRA Subparameter Updates