Epistemic Monte Carlo Tree Search

Fuente: arXiv
Saved in:
Bibliographic Details
Main Authors: Oren, Yaniv, Vadocz, Viliam, Spaan, Matthijs T. J., Böhmer, Wendelin
Format: Preprint
Published: 2022
Subjects:
Online Access:
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866916014180007936
author Oren, Yaniv
Vadocz, Viliam
Spaan, Matthijs T. J.
Böhmer, Wendelin
author_facet Oren, Yaniv
Vadocz, Viliam
Spaan, Matthijs T. J.
Böhmer, Wendelin
contents The AlphaZero/MuZero (A/MZ) family of algorithms has achieved remarkable success across various challenging domains by integrating Monte Carlo Tree Search (MCTS) with learned models. Learned models introduce epistemic uncertainty, which is caused by learning from limited data and is useful for exploration in sparse reward environments. MCTS does not account for the propagation of this uncertainty however. To address this, we introduce Epistemic MCTS (EMCTS): a theoretically motivated approach to account for the epistemic uncertainty in search and harness the search for deep exploration. In the challenging sparse-reward task of writing code in the Assembly language SUBLEQ, AZ paired with our method achieves significantly higher sample efficiency over baseline AZ. Search with EMCTS solves variations of the commonly used hard-exploration benchmark Deep Sea - which baseline A/MZ are practically unable to solve - much faster than an otherwise equivalent method that does not use search for uncertainty estimation, demonstrating significant benefits from search for epistemic uncertainty estimation.
format Preprint
id arxiv_https___arxiv_org_abs_2210_13455
institution arXiv
publishDate 2022
record_format arxiv
spellingShingle Epistemic Monte Carlo Tree Search
Oren, Yaniv
Vadocz, Viliam
Spaan, Matthijs T. J.
Böhmer, Wendelin
Machine Learning
Artificial Intelligence
The AlphaZero/MuZero (A/MZ) family of algorithms has achieved remarkable success across various challenging domains by integrating Monte Carlo Tree Search (MCTS) with learned models. Learned models introduce epistemic uncertainty, which is caused by learning from limited data and is useful for exploration in sparse reward environments. MCTS does not account for the propagation of this uncertainty however. To address this, we introduce Epistemic MCTS (EMCTS): a theoretically motivated approach to account for the epistemic uncertainty in search and harness the search for deep exploration. In the challenging sparse-reward task of writing code in the Assembly language SUBLEQ, AZ paired with our method achieves significantly higher sample efficiency over baseline AZ. Search with EMCTS solves variations of the commonly used hard-exploration benchmark Deep Sea - which baseline A/MZ are practically unable to solve - much faster than an otherwise equivalent method that does not use search for uncertainty estimation, demonstrating significant benefits from search for epistemic uncertainty estimation.
title Epistemic Monte Carlo Tree Search
topic Machine Learning
Artificial Intelligence
url https://arxiv.org/abs/2210.13455