arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2305.04073cs.AIcs.LG

用轨迹解释强化学习决策

Explaining RL Decisions with Trajectories

Media and Data Science Research, Adobe(Adobe媒体与数据科学研究);International Institute of Information Technology Hyderabad(海得拉巴国际信息技术学院);University of Illinois Urbana-Champaign(伊利诺伊大学厄巴纳-香槟分校);Adobe Research(Adobe研究院)

Shripad Vilasrao Deshmukh, Arpan Dasgupta, Balaji Krishnamurthy, Nan Jiang, Chirag Agarwal, Georgios Theocharous, Jayakumar Subramanian

AI总结:

本文面向离线RL提出轨迹级决策解释方法,通过单独与集体编码训练轨迹并估计决策敏感性实现归因,在多类环境和人类研究中验证其质量与可扩展性。

摘要:

解释是强化学习(RL)在许多现实决策问题中得到采用的关键组成部分。在已有文献中,解释通常通过对RL智能体状态特征进行显著性归因来提供。在本工作中,我们提出一种与这些解释互补的方法,尤其针对离线RL:将训练后RL智能体的策略决策归因于其在训练期间遇到的轨迹。为此,我们分别对离线训练数据中的轨迹进行单独编码,也进行集体编码(即编码一组轨迹)。随后,我们通过估计决策对某一轨迹集合的敏感性,在该编码空间中将策略决策归因于这组轨迹。此外,我们在涉及离散与连续状态及动作空间的多种环境中,从归因质量和实际可扩展性两方面证明了所提方法的有效性,这些环境包括网格世界、电子游戏(Atari)和连续控制(MuJoCo)。我们还在一个简单导航任务上开展了人类研究,以观察参与者对任务的理解与为训练后RL策略所归因数据之间的比较。关键词——可解释AI、AI决策的可验证性、可解释RL。

英文摘要:

Explanation is a key component for the adoption of reinforcement learning (RL) in many real-world decision-making problems. In the literature, the explanation is often provided by saliency attribution to the features of the RL agent's state. In this work, we propose a complementary approach to these explanations, particularly for offline RL, where we attribute the policy decisions of a trained RL agent to the trajectories encountered by it during training. To do so, we encode trajectories in offline training data individually as well as collectively (encoding a set of trajectories). We then attribute policy decisions to a set of trajectories in this encoded space by estimating the sensitivity of the decision with respect to that set. Further, we demonstrate the effectiveness of the proposed approach in terms of quality of attributions as well as practical scalability in diverse environments that involve both discrete and continuous state and action spaces such as grid-worlds, video games (Atari) and continuous control (MuJoCo). We also conduct a human study on a simple navigation task to observe how their understanding of the task compares with data attributed for a trained RL policy. Keywords -- Explainable AI, Verifiability of AI Decisions, Explainable RL.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑