arXiv Machine Learning

Fast-Convergent Meta-RL via Gradient-Clustered BS Sampling for Edge Caching

The paper introduces a meta‑reinforcement learning framework for wireless edge caching that uses a shared initialization to adapt quickly to each base station (BS). Each BS runs a local PPO agent modeled as a Semi‑Markov Decision Process, while a shared meta‑policy is trained via a MAML‑style loop. To reduce the high variance of the meta‑gradient estimator caused by uniform random sampling of BSs, the authors propose gradient‑based clustering, grouping BSs by local gradient similarity and sampling proportionally from each cluster, proving that this yields a lower‑variance estimator under BS heterogeneity.

arXiv Machine Learning
Sep 22

Personalized Federated Reinforcement Learning via Model-Agnostic Meta-Learning: Convergence of Exact and Hessian-Free Meta-Policy Gradients

arXiv:2609. 22833v1 Announce Type: new Abstract: We study personalized federated reinforcement learning, in which $n$ agents, each acting in its own Markov decision process, collaborate through a server to learn a shared MAML-style policy initialization that becomes effective for an individual agent once that agent adapts it with a single local policy-gradient step.

By Ali Beikmohammadi, Sarit Khirirat, Sindri Magn\'usson
arXiv AI
Oct 9

Fed-GRPO: Reward-Signal-Driven Federated Group Relative Policy Optimization

Fed-GRPO is a federated training framework for Large Language Models that extends Group Relative Policy Optimization (GRPO) to operate without sharing raw data. It introduces three reward‑signal‑driven mechanisms—signal‑weighted aggregation, global reward calibration, and adaptive sparse communication—to guide client updates based on reward statistics. Experiments on mathematical reasoning tasks show that Fed‑GRPO outperforms other federated methods, approaches centralized performance, and reduces communication by up to 32× while allowing extreme compression with minimal accuracy loss.

By Pengxin Guo, Shuang Zeng, Zonggen Li, Weiying Zheng, Mengting Liu, Liangqiong Qu