We gratefully acknowledge support from
the Simons Foundation and member institutions.
Full-text links:

Download:

Current browse context:

cs.LG

Change to browse by:

cs

References & Citations

DBLP - CS Bibliography

Bookmark

(what is this?)
CiteULike logo BibSonomy logo Mendeley logo del.icio.us logo Digg logo Reddit logo

Computer Science > Machine Learning

Title: Evolutionary learning of interpretable decision trees

Abstract: Reinforcement learning techniques achieved human-level performance in several tasks in the last decade. However, in recent years, the need for interpretability emerged: we want to be able to understand how a system works and the reasons behind its decisions. Not only we need interpretability to assess the safety of the produced systems, we also need it to extract knowledge about unknown problems. While some techniques that optimize decision trees for reinforcement learning do exist, they usually employ greedy algorithms or they do not exploit the rewards given by the environment. This means that these techniques may easily get stuck in local optima. In this work, we propose a novel approach to interpretable reinforcement learning that uses decision trees. We present a two-level optimization scheme that combines the advantages of evolutionary algorithms with the advantages of Q-learning. This way we decompose the problem into two sub-problems: the problem of finding a meaningful and useful decomposition of the state space, and the problem of associating an action to each state. We test the proposed method on three well-known reinforcement learning benchmarks, on which it results competitive with respect to the state-of-the-art in both performance and interpretability. Finally, we perform an ablation study that confirms that using the two-level optimization scheme gives a boost in performance in non-trivial environments with respect to a one-layer optimization technique.
Comments: 69 pages, 31 figures, code available at: this https URL
Subjects: Machine Learning (cs.LG)
ACM classes: I.2.2; I.2.6
Journal reference: IEEE Access (2023)
DOI: 10.1109/ACCESS.2023.3236260
Cite as: arXiv:2012.07723 [cs.LG]
  (or arXiv:2012.07723v3 [cs.LG] for this version)

Submission history

From: Leonardo Lucio Custode [view email]
[v1] Mon, 14 Dec 2020 17:16:46 GMT (763kb,D)
[v2] Tue, 15 Dec 2020 17:26:54 GMT (794kb,D)
[v3] Wed, 21 Apr 2021 10:03:13 GMT (1559kb,D)

Link back to: arXiv, form interface, contact.