AI RESEARCH

Federated Distributional Reinforcement Learning with Distributional Critic Regularization

arXiv CS.LG

ArXi:2603.17820v1 Announce Type: new Federated reinforcement learning typically aggregates value functions or policies by parameter averaging, which emphasizes expected return and can obscure statistical multimodality and tail behavior that matter in safety-critical settings. We formalize federated distributional reinforcement learning (FedDistRL), where clients parametrize quantile value function critics and federate these networks only. We also propose TR-FedDistRL, which builds a per client, risk-aware Wasserstein barycenter over a temporal buffer.