A major challenge of Reinforcement Learning is that agents often learn undesired behaviors that seem to defy the reward structure they were given. Explainable Reinforcement Learning (XRL) methods can answer queries such as "explain this specific action", "explain this specific trajectory", and "explain the entire policy". However, XRL lacks a formal definition for behavior as a pattern of actions across many episodes. We provide such a definition, and use it to enable a new query: "Explain this behavior". We present Behavior-Explainable Reinforcement Learning (BXRL), a new problem formulation that treats behaviors as first-class objects. BXRL defines a behavior measure as any function m:Π→R, allowing users to precisely express the pattern of actions that they find interesting and measure how strongly the policy exhibits it. We define contrastive behaviors that reduce the question "why does the agent prefer a to a′?" to "why is m(π) high?" which can be explored with differentiation. We do not implement an explainability method; we instead analyze three existing methods and propose how they could be adapted to explain behavior. We present a port of the HighwayEnv driving environment to JAX, which provides an interface for defining, measuring, and differentiating behaviors with respect to the model parameters.
Figures & tables
Figure 1: HighJax TUI for defining behavior scenarios by selecting states and actions from existing rollouts.
Figure 2: Left: Return breakdown of a PPO agent training on HighJax. Training is deliberately slowed down to DKL=5×10−4 per epoch. Right: Value of the collision behavior measure mc across training.
Figure 3: Six scenarios defining the collision behavior measure mc . Each causes the ego car (dark gray) to crash within two timesteps. Columns correspond to the action taken: left , right , or faster .
Appendix figures & tables1 asset
Supplementary material from the paper’s appendix.
Appendix
Table 4: Observation matrices for the six scenarios defining the collision behavior measure mc (Eq. 4 ). Each matrix O∈R5×5 contains the ego vehicle (absolute) and four nearest NPCs (relative). Columns: presence, longitudinal position, lateral position, longitudinal velocity, lateral velocity.