Automatic Reward Shaping from Confounded Offline Data

Fuente: arXiv
Saved in:
Bibliographic Details
Main Authors: Li, Mingxuan, Zhang, Junzhe, Bareinboim, Elias
Format: Preprint
Published: 2025
Subjects:
Online Access:
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866908526230634496
author Li, Mingxuan
Zhang, Junzhe
Bareinboim, Elias
author_facet Li, Mingxuan
Zhang, Junzhe
Bareinboim, Elias
contents A key task in Artificial Intelligence is learning effective policies for controlling agents in unknown environments to optimize performance measures. Off-policy learning methods, like Q-learning, allow learners to make optimal decisions based on past experiences. This paper studies off-policy learning from biased data in complex and high-dimensional domains where \emph{unobserved confounding} cannot be ruled out a priori. Building on the well-celebrated Deep Q-Network (DQN), we propose a novel deep reinforcement learning algorithm robust to confounding biases in observed data. Specifically, our algorithm attempts to find a safe policy for the worst-case environment compatible with the observations. We apply our method to twelve confounded Atari games, and find that it consistently dominates the standard DQN in all games where the observed input to the behavioral and target policies mismatch and unobserved confounders exist.
format Preprint
id arxiv_https___arxiv_org_abs_2505_11478
institution arXiv
publishDate 2025
record_format arxiv
spellingShingle Automatic Reward Shaping from Confounded Offline Data
Li, Mingxuan
Zhang, Junzhe
Bareinboim, Elias
Artificial Intelligence
Machine Learning
A key task in Artificial Intelligence is learning effective policies for controlling agents in unknown environments to optimize performance measures. Off-policy learning methods, like Q-learning, allow learners to make optimal decisions based on past experiences. This paper studies off-policy learning from biased data in complex and high-dimensional domains where \emph{unobserved confounding} cannot be ruled out a priori. Building on the well-celebrated Deep Q-Network (DQN), we propose a novel deep reinforcement learning algorithm robust to confounding biases in observed data. Specifically, our algorithm attempts to find a safe policy for the worst-case environment compatible with the observations. We apply our method to twelve confounded Atari games, and find that it consistently dominates the standard DQN in all games where the observed input to the behavioral and target policies mismatch and unobserved confounders exist.
title Automatic Reward Shaping from Confounded Offline Data
topic Artificial Intelligence
Machine Learning
url https://arxiv.org/abs/2505.11478