Primal-Dual Spectral Representation for Off-policy Evaluation

Fuente: arXiv
Saved in:
Bibliographic Details
Main Authors: Hu, Yang, Chen, Tianyi, Li, Na, Wang, Kai, Dai, Bo
Format: Preprint
Published: 2024
Subjects:
Online Access:
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866929555009175552
author Hu, Yang
Chen, Tianyi
Li, Na
Wang, Kai
Dai, Bo
author_facet Hu, Yang
Chen, Tianyi
Li, Na
Wang, Kai
Dai, Bo
contents Off-policy evaluation (OPE) is one of the most fundamental problems in reinforcement learning (RL) to estimate the expected long-term payoff of a given target policy with only experiences from another behavior policy that is potentially unknown. The distribution correction estimation (DICE) family of estimators have advanced the state of the art in OPE by breaking the curse of horizon. However, the major bottleneck of applying DICE estimators lies in the difficulty of solving the saddle-point optimization involved, especially with neural network implementations. In this paper, we tackle this challenge by establishing a linear representation of value function and stationary distribution correction ratio, i.e., primal and dual variables in the DICE framework, using the spectral decomposition of the transition operator. Such primal-dual representation not only bypasses the non-convex non-concave optimization in vanilla DICE, therefore enabling an computational efficient algorithm, but also paves the way for more efficient utilization of historical data. We highlight that our algorithm, SpectralDICE, is the first to leverage the linear representation of primal-dual variables that is both computation and sample efficient, the performance of which is supported by a rigorous theoretical sample complexity guarantee and a thorough empirical evaluation on various benchmarks.
format Preprint
id arxiv_https___arxiv_org_abs_2410_17538
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle Primal-Dual Spectral Representation for Off-policy Evaluation
Hu, Yang
Chen, Tianyi
Li, Na
Wang, Kai
Dai, Bo
Machine Learning
Artificial Intelligence
Optimization and Control
Off-policy evaluation (OPE) is one of the most fundamental problems in reinforcement learning (RL) to estimate the expected long-term payoff of a given target policy with only experiences from another behavior policy that is potentially unknown. The distribution correction estimation (DICE) family of estimators have advanced the state of the art in OPE by breaking the curse of horizon. However, the major bottleneck of applying DICE estimators lies in the difficulty of solving the saddle-point optimization involved, especially with neural network implementations. In this paper, we tackle this challenge by establishing a linear representation of value function and stationary distribution correction ratio, i.e., primal and dual variables in the DICE framework, using the spectral decomposition of the transition operator. Such primal-dual representation not only bypasses the non-convex non-concave optimization in vanilla DICE, therefore enabling an computational efficient algorithm, but also paves the way for more efficient utilization of historical data. We highlight that our algorithm, SpectralDICE, is the first to leverage the linear representation of primal-dual variables that is both computation and sample efficient, the performance of which is supported by a rigorous theoretical sample complexity guarantee and a thorough empirical evaluation on various benchmarks.
title Primal-Dual Spectral Representation for Off-policy Evaluation
topic Machine Learning
Artificial Intelligence
Optimization and Control
url https://arxiv.org/abs/2410.17538