Title | Quantum Algorithms for Reinforcement Learning with a Generative Model |

Publication Type | Journal Article |

Year of Publication | 2021 |

Authors | Wang, D, Sundaram, A, Kothari, R, Kapoor, A, Roetteler, M |

Journal | Proceedings of the 38th International Conference on Machine Learning, PMLR |

Volume | 139 |

Start Page | 10916-10926 |

Date Published | 12/15/2021 |

Keywords | FOS: Computer and information sciences, FOS: Physical sciences, Machine Learning (cs.LG), Quantum Physics (quant-ph) |

Abstract | Reinforcement learning studies how an agent should interact with an environment to maximize its cumulative reward. A standard way to study this question abstractly is to ask how many samples an agent needs from the environment to learn an optimal policy for a γ-discounted Markov decision process (MDP). For such an MDP, we design quantum algorithms that approximate an optimal policy (π∗), the optimal value function (v∗), and the optimal Q-function (q∗), assuming the algorithms can access samples from the environment in quantum superposition. This assumption is justified whenever there exists a simulator for the environment; for example, if the environment is a video game or some other program. Our quantum algorithms, inspired by value iteration, achieve quadratic speedups over the best-possible classical sample complexities in the approximation accuracy (ϵ) and two main parameters of the MDP: the effective time horizon (11−γ) and the size of the action space (A). Moreover, we show that our quantum algorithm for computing q∗ is optimal by proving a matching quantum lower bound. |

URL | https://arxiv.org/abs/2112.08451 |

DOI | 10.48550/ARXIV.2112.08451 |