Collaborative Value Function Estimation Under Model Mismatch: A Federated Temporal Difference Analysis

Fuente: arXiv
Saved in:
Bibliographic Details
Main Authors: Beikmohammadi, Ali, Khirirat, Sarit, Richtárik, Peter, Magnússon, Sindri
Format: Preprint
Published: 2025
Subjects:
Online Access:
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866913893334384640
author Beikmohammadi, Ali
Khirirat, Sarit
Richtárik, Peter
Magnússon, Sindri
author_facet Beikmohammadi, Ali
Khirirat, Sarit
Richtárik, Peter
Magnússon, Sindri
contents Federated reinforcement learning (FedRL) enables collaborative learning while preserving data privacy by preventing direct data exchange between agents. However, many existing FedRL algorithms assume that all agents operate in identical environments, which is often unrealistic. In real-world applications, such as multi-robot teams, crowdsourced systems, and large-scale sensor networks, each agent may experience slightly different transition dynamics, leading to inherent model mismatches. In this paper, we first establish linear convergence guarantees for single-agent temporal difference learning (TD(0)) in policy evaluation and demonstrate that under a perturbed environment, the agent suffers a systematic bias that prevents accurate estimation of the true value function. This result holds under both i.i.d. and Markovian sampling regimes. We then extend our analysis to the federated TD(0) (FedTD(0)) setting, where multiple agents, each interacting with its own perturbed environment, periodically share value estimates to collaboratively approximate the true value function of a common underlying model. Our theoretical results indicate the impact of model mismatch, network connectivity, and mixing behavior on the convergence of FedTD(0). Empirical experiments corroborate our theoretical gains, highlighting that even moderate levels of information sharing significantly mitigate environment-specific errors.
format Preprint
id arxiv_https___arxiv_org_abs_2503_17454
institution arXiv
publishDate 2025
record_format arxiv
spellingShingle Collaborative Value Function Estimation Under Model Mismatch: A Federated Temporal Difference Analysis
Beikmohammadi, Ali
Khirirat, Sarit
Richtárik, Peter
Magnússon, Sindri
Machine Learning
Federated reinforcement learning (FedRL) enables collaborative learning while preserving data privacy by preventing direct data exchange between agents. However, many existing FedRL algorithms assume that all agents operate in identical environments, which is often unrealistic. In real-world applications, such as multi-robot teams, crowdsourced systems, and large-scale sensor networks, each agent may experience slightly different transition dynamics, leading to inherent model mismatches. In this paper, we first establish linear convergence guarantees for single-agent temporal difference learning (TD(0)) in policy evaluation and demonstrate that under a perturbed environment, the agent suffers a systematic bias that prevents accurate estimation of the true value function. This result holds under both i.i.d. and Markovian sampling regimes. We then extend our analysis to the federated TD(0) (FedTD(0)) setting, where multiple agents, each interacting with its own perturbed environment, periodically share value estimates to collaboratively approximate the true value function of a common underlying model. Our theoretical results indicate the impact of model mismatch, network connectivity, and mixing behavior on the convergence of FedTD(0). Empirical experiments corroborate our theoretical gains, highlighting that even moderate levels of information sharing significantly mitigate environment-specific errors.
title Collaborative Value Function Estimation Under Model Mismatch: A Federated Temporal Difference Analysis
topic Machine Learning
url https://arxiv.org/abs/2503.17454