FedGRPO: Privately Optimizing Foundation Models with Group-Relative Rewards from Domain Clients
Gongxi Zhu, Hanlin Gu, Lixin Fan, Qiang Yang, Yuxing Han
Abstract
One important direction of Federated Foundation Models (FedFMs) is leveraging data from small client models to enhance the performance of a large server‑side foundation model. Existing methods based on model level or representation level knowledge transfer either require expensive local training or incur high communication costs and introduce unavoidable privacy risks. We reformulate this problem as a reinforcement learning style evaluation process and propose FedGRPO, a privacy preserving framework comprising two modules. The first module performs competence-based expert selection by building a lightweight confidence graph from auxiliary data to identify the most suitable clients for each question. The second module leverages the “Group Relative” concept from the Group Relative Policy Optimization (GRPO) framework by packaging each question together with its solution rationale into candidate policies, dispatching these policies to a selected subset of expert clients, and aggregating solely the resulting scalar reward signals via a federated group–relative loss function. By exchanging reward values instead of data or model updates, FedGRPO reduces privacy risk and communication overhead while enabling parallel evaluation across heterogeneous devices. Empirical results on diverse domain tasks demonstrate that FedGRPO achieves superior downstream accuracy and communication efficiency compared to conventional FedFMs baselines.
Ask Lune
Ask anything. The answer quotes the paper.
Lune reads all of it and the papers around it, then shows you the passages behind every answer.
Builds on9
- Solving Quantitative Reasoning Problems with Language ModelsAitor Lewkowycz, Anders Andreassen, David Dohan, Ethan Dyer et al.NeurIPS 2022 · 2,039 citations
- Improving LoRA in Privacy-preserving Federated LearningYoubang Sun, Zitao Li, Yaliang Li, Bolin DingICLR 2024 · 173 citations
- FedPETuning: When Federated Learning Meets the Parameter-Efficient Tuning Methods of Pre-trained Language ModelsZhuo Zhang, Yuanhang Yang, Yong Dai, Qifan Wang et al.ACL Findings 2023 · 98 citations
- Multimodal Federated Learning via Contrastive Representation EnsembleQiying Yu, Yang Liu, Yimu Wang, Ke Xu et al.ICLR 2023 · 35 citations
- OlympiadBench: A Challenging Benchmark for Promoting AGI with Olympiad-Level Bilingual Multimodal Scientific ProblemsChaoqun He, Renjie Luo, Yuzhuo Bai, Shengding Hu et al.ACL 2024 · 18 citations
Related papers
- Graph-GRPO: Stabilizing Multi-Agent Topology Learning via Group Relative Policy OptimizationYueyang Cang, Xiaoteng Zhang, Erlu Zhao, Zehua Ji et al.ACL Findings 2026
- From Absolute to Relative: Rethinking Reward Shaping in Group-Based Reinforcement LearningWenzhe Niu, Wei He, Zongxia Xie, Jinpeng Ou et al.ICML 2026 · 1 citation
- Reinforcement Learning for Large Language Models via Group Preference Reward ShapingHuaisheng Zhu, Siyuan Xu, Hangfan Zhang, Teng Xiao et al.EMNLP 2025
- GRPO is Secretly a Process Reward ModelMichael Sullivan, Alexander KollerICML 2026 · 8 citations
- Bridging Model Heterogeneity in Federated Learning via Uncertainty-based Asymmetrical Reciprocity LearningJiaqi Wang, Chenxu Zhao, Lingjuan Lyu, Quanzeng You et al.ICML 2024 · 16 citations
