arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2212.07486cs.LGcs.AI
收藏

基于状态抽象的高维状态空间边缘化重要性采样

Scaling Marginalized Importance Sampling to High-Dimensional State-Spaces via State Abstraction

University of Wisconsin – Madison(威斯康星大学麦迪逊分校)

Brahma S. Pavse, Josiah P. Hanna

摘要:

我们考虑强化学习(RL)中的非策略评估(OPE)问题,其中目标是使用固定数据集$\mathcal{D}$来估计评估策略$π_e$的性能,所述固定数据集由可能不同于$π_e$的一个或多个策略收集。当前的OPE算法可能在策略分布偏移下产生差的OPE估计,当特定状态-动作对在$π_e$下发生的概率与同一对在$\mathcal{D}$中发生的概率非常不同时(Voloshin等,2021; Fu等,2021)。在这项工作中,我们建议通过使用状态抽象文献中的概念将高维状态空间投影到低维状态空间来提高OPE估计的准确性。具体来说,我们考虑边缘化的重要性采样(MIS)OPE算法计算状态-动作分布校正比,以产生他们的OPE估计。在原始基态空间中,这些比率可能具有高方差,这可能导致高方差OPE。然而,我们证明了在低维抽象状态空间的比率可以有较低的方差导致较低的方差OPE。然后,我们强调的挑战时,估计的抽象比率从数据,确定充分的条件来克服这些问题,并提出了一个极大极小优化问题,其解决方案产生这些抽象的比率。最后,我们对困难的高维状态空间OPE任务的经验评估表明,抽象比率可以使MIS OPE估计器实现更低的均方误差,并且比地面比率对超参数调整更鲁棒。

英文摘要:

We consider the problem of off-policy evaluation (OPE) in reinforcement learning (RL), where the goal is to estimate the performance of an evaluation policy, $π_e$, using a fixed dataset, $\mathcal{D}$, collected by one or more policies that may be different from $π_e$. Current OPE algorithms may produce poor OPE estimates under policy distribution shift i.e., when the probability of a particular state-action pair occurring under $π_e$ is very different from the probability of that same pair occurring in $\mathcal{D}$ (Voloshin et al. 2021, Fu et al. 2021). In this work, we propose to improve the accuracy of OPE estimators by projecting the high-dimensional state-space into a low-dimensional state-space using concepts from the state abstraction literature. Specifically, we consider marginalized importance sampling (MIS) OPE algorithms which compute state-action distribution correction ratios to produce their OPE estimate. In the original ground state-space, these ratios may have high variance which may lead to high variance OPE. However, we prove that in the lower-dimensional abstract state-space the ratios can have lower variance resulting in lower variance OPE. We then highlight the challenges that arise when estimating the abstract ratios from data, identify sufficient conditions to overcome these issues, and present a minimax optimization problem whose solution yields these abstract ratios. Finally, our empirical evaluation on difficult, high-dimensional state-space OPE tasks shows that the abstract ratios can make MIS OPE estimators achieve lower mean-squared error and more robust to hyperparameter tuning than the ground ratios.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑