xSRL: Safety-Aware Explainable Reinforcement Learning -- Safety as a Product of Explainability

TL;DR

xSRL is a framework that enhances the safety and explainability of reinforcement learning agents by providing comprehensive local and global explanations, enabling debugging, vulnerability detection, and increased trust for real-world applications.

Contribution

The paper introduces xSRL, a novel explainability framework for RL that combines local and global explanations and supports vulnerability analysis without retraining.

Findings

xSRL improves safety and trustworthiness of RL agents.

It enables identification of policy vulnerabilities through adversarial attacks.

User studies confirm increased understanding and confidence in RL decisions.

Abstract

Reinforcement learning (RL) has shown great promise in simulated environments, such as games, where failures have minimal consequences. However, the deployment of RL agents in real-world systems such as autonomous vehicles, robotics, UAVs, and medical devices demands a higher level of safety and transparency, particularly when facing adversarial threats. Safe RL algorithms have been developed to address these concerns by optimizing both task performance and safety constraints. However, errors are inevitable, and when they occur, it is essential that the RL agents can also explain their actions to human operators. This makes trust in the safety mechanisms of RL systems crucial for effective deployment. Explainability plays a key role in building this trust by providing clear, actionable insights into the agent's decision-making process, ensuring that safety-critical decisions are well understood. While machine learning (ML) has seen significant advances in interpretability and visualization, explainability methods for RL remain limited. Current tools fail to address the dynamic, sequential nature of RL and its needs to balance task performance with safety constraints over time. The re-purposing of traditional ML methods, such as saliency maps, is inadequate for safety-critical RL applications where mistakes can result in severe consequences. To bridge this gap, we propose xSRL, a framework that integrates both local and global explanations to provide a comprehensive understanding of RL agents' behavior. xSRL also enables developers to identify policy vulnerabilities through adversarial attacks, offering tools to debug and patch agents without retraining. Our experiments and user studies demonstrate xSRL's effectiveness in increasing safety in RL systems, making them more reliable and trustworthy for real-world deployment. Code is available at https://github.com/risal-shefin/xSRL.