Designing Interpretable Approximations to Deep Reinforcement Learning with Soft Decision Trees

By Nathan Dahlin, Krishna Chaitanya Kalagarla, Nikhil Vijay Naik, Rahul Jain, Pierluigi Nuzzo. 2020

10 citations recorded by the source corpus.

In an ever expanding set of research and application areas, deep neural networks (DNNs) set the bar for algorithm performance. However, depending upon additional constraints such as processing power and execution time limits, or requirements such as verifiable safety guarantees, it may not be feasible to actually use such high-performing DNNs in practice. Many techniques have been developed in recent years to compress or distill complex DNNs into smaller, faster or more understandable models and controllers. This work seeks to provide a quantitative framework with metrics to systematically evaluate the outcome of such conversion processes, and identify reduced models that not only preserve a desired performance level, but also, for example, succinctly explain the latent knowledge represented by a DNN. We illustrate the effectiveness of the proposed approach on the evaluation of decision tree variants in the context of benchmark reinforcement learning tasks.

Subfields: Artificial Intelligence, Reinforcement Learning, Safety, Robustness & Interpretability

How researchers see this paper

The researchers whose trusted networks rank it highest, strongest advocate first. Every researcher gets a different order.

In Artificial Intelligence

  1. Lesia Semenova
  2. Chudi Zhong
  3. Zhi Chen
  4. Haiyang Huang
  5. Cynthia Rudin
  6. Tong Wang
  7. Octavio Loyola-González
  8. Berk Ustun

In Reinforcement Learning

  1. Zhi Chen
  2. Cynthia Rudin

In Safety, Robustness & Interpretability

  1. Lesia Semenova
  2. Zhi Chen
  3. Chudi Zhong
  4. Haiyang Huang
  5. Cynthia Rudin
  6. Octavio Loyola-González
  7. Tong Wang
  8. S. Lo Piano

Citations in the corpus