Federated Offline Policy Learning

Fuente: arXiv
Saved in:
Bibliographic Details
Main Authors: Carranza, Aldo Gael, Athey, Susan
Format: Preprint
Published: 2023
Subjects:
Online Access:
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866914969656754176
author Carranza, Aldo Gael
Athey, Susan
author_facet Carranza, Aldo Gael
Athey, Susan
contents We consider the problem of learning personalized decision policies from observational bandit feedback data across multiple heterogeneous data sources. In our approach, we introduce a novel regret analysis that establishes finite-sample upper bounds on distinguishing notions of global regret for all data sources on aggregate and of local regret for any given data source. We characterize these regret bounds by expressions of source heterogeneity and distribution shift. Moreover, we examine the practical considerations of this problem in the federated setting where a central server aims to train a policy on data distributed across the heterogeneous sources without collecting any of their raw data. We present a policy learning algorithm amenable to federation based on the aggregation of local policies trained with doubly robust offline policy evaluation strategies. Our analysis and supporting experimental results provide insights into tradeoffs in the participation of heterogeneous data sources in offline policy learning.
format Preprint
id arxiv_https___arxiv_org_abs_2305_12407
institution arXiv
publishDate 2023
record_format arxiv
spellingShingle Federated Offline Policy Learning
Carranza, Aldo Gael
Athey, Susan
Machine Learning
Distributed, Parallel, and Cluster Computing
Econometrics
We consider the problem of learning personalized decision policies from observational bandit feedback data across multiple heterogeneous data sources. In our approach, we introduce a novel regret analysis that establishes finite-sample upper bounds on distinguishing notions of global regret for all data sources on aggregate and of local regret for any given data source. We characterize these regret bounds by expressions of source heterogeneity and distribution shift. Moreover, we examine the practical considerations of this problem in the federated setting where a central server aims to train a policy on data distributed across the heterogeneous sources without collecting any of their raw data. We present a policy learning algorithm amenable to federation based on the aggregation of local policies trained with doubly robust offline policy evaluation strategies. Our analysis and supporting experimental results provide insights into tradeoffs in the participation of heterogeneous data sources in offline policy learning.
title Federated Offline Policy Learning
topic Machine Learning
Distributed, Parallel, and Cluster Computing
Econometrics
url https://arxiv.org/abs/2305.12407