PNNL

Abstract

Reinforcement learning (RL) is a branch of machine learning where an agent learns to maximize reward through trial and error. RL is challenging and data/compute intensive, and RL practitioners can easily become overwhelmed and make poor modeling decisions. Our contribution is a visual analytics tool designed to help data scientists maintain situation awareness during RL experimentation. Our tool allows the user to under- stand which hyper-parameter values lead to better or worse outcomes, what behaviors are associated with high and low reward, and how behaviors evolve and progress over training time. We evaluated our tool by applying state of the art deep RL models to six different RL tasks. We present three tasks as use cases that show how our tool leads to insightful findings and improves situation awareness.