Fast counterfactual inference for history-based reinforcement learning

H Gao, T Zhang, Z Yang, Y Guo, J Ren, S Guo… - Proceedings of the …, 2023 - ojs.aaai.org
H Gao, T Zhang, Z Yang, Y Guo, J Ren, S Guo, F Chen
Proceedings of the AAAI Conference on Artificial Intelligence, 2023ojs.aaai.org
Incorporating sequence-to-sequence models into history-based Reinforcement Learning
(RL) provides a general way to extend RL to partially-observable tasks. This method
compresses history spaces according to the correlations between historical observations
and the rewards. However, they do not adjust for the confounding correlations caused by
data sampling and assign high beliefs to uninformative historical observations, leading to
limited compression of history spaces. Counterfactual Inference (CI), which estimates causal …
Abstract
Incorporating sequence-to-sequence models into history-based Reinforcement Learning (RL) provides a general way to extend RL to partially-observable tasks. This method compresses history spaces according to the correlations between historical observations and the rewards. However, they do not adjust for the confounding correlations caused by data sampling and assign high beliefs to uninformative historical observations, leading to limited compression of history spaces. Counterfactual Inference (CI), which estimates causal effects by single-variable intervention, is a promising way to adjust for confounding. However, it is computationally infeasible to directly apply the single-variable intervention to a huge number of historical observations. This paper proposes to perform CI on observation sub-spaces instead of single observations and develop a coarse-to-fine CI algorithm, called Tree-based History Counterfactual Inference (T-HCI), to reduce the number of interventions exponentially. We show that T-HCI is computationally feasible in practice and brings significant sample efficiency gains in various challenging partially-observable tasks, including Maze, BabyAI, and robot manipulation tasks.
ojs.aaai.org
以上显示的是最相近的搜索结果。 查看全部搜索结果