Kernel-Based Reinforcement Learning on Representative States

Authors

  • Branislav Kveton Technicolor Labs
  • Georgios Theocharous Yahoo Labs

DOI:

https://doi.org/10.1609/aaai.v26i1.8294

Keywords:

reinforcement learning, kernels, Markov decision processes

Abstract

Markov decision processes (MDPs) are an established framework for solving sequential decision-making problems under uncertainty. In this work, we propose a new method for batch-mode reinforcement learning (RL) with continuous state variables. The method is an approximation to kernel-based RL on a set of k representative states. Similarly to kernel-based RL, our solution is a fixed point of a kernelized Bellman operator and can approximate the optimal solution to an arbitrary level of granularity. Unlike kernel-based RL, our method is fast. In particular, our policies can be computed in O(n) time, where n is the number of training examples. The time complexity of kernel-based RL is Ω(n2). We introduce our method, analyze its convergence, and compare it to existing work. The method is evaluated on two existing control problems with 2 to 4 continuous variables and a new problem with 64 variables. In all cases, we outperform state-of-the-art results and offer simpler solutions.

Downloads

Published

2021-09-20

How to Cite

Kveton, B., & Theocharous, G. (2021). Kernel-Based Reinforcement Learning on Representative States. Proceedings of the AAAI Conference on Artificial Intelligence, 26(1), 977-983. https://doi.org/10.1609/aaai.v26i1.8294

Issue

Section

AAAI Technical Track: Machine Learning