arXiv:2410.16517cs.LGcs.AI2024-10NeurIPS被引 2

提出RGMDT算法,实现多智能体环境下可解释决策树的高精度提取。

RGMDT: Return-Gap-Minimizing Decision Tree Extraction in Non-Euclidean Metric Space

  • 将决策树提取转化为非欧空间聚类问题,以返回差距为损失函数
  • 在D4RL任务上逼近最优回报,节点数受限时性能显著优于基线
  • 适用于需可解释性的多智能体强化学习场景,如机器人协作

深度强化学习虽在复杂任务中表现优异,但其黑箱特性阻碍了可解释性与实际应用。现有工作多聚焦单智能体可解释强化学习,而多智能体场景中的决策树提取主要依赖启发式设计,缺乏对期望回报的定量保障。本文建立了专家策略与最优决策树策略间回报差距的上界,将决策树提取问题重构为基于局部观测与动作值空间的非欧聚类问题,以动作值为聚类标签,以上界作为聚类损失。通过基于当前其他智能体决策树条件的动作值函数引导的迭代生长过程,该方法扩展至多智能体去中心化决策树提取。进一步提出返回差距最小化决策树(RGMDT)算法,其设计简洁,通过新型正则化信息最大化损失与强化学习集成。在D4RL等任务上的评估表明,RGMDT显著优于启发式基线,在给定决策树复杂度约束(如最大节点数)下可达到近最优回报。

原文摘要 · Abstract (English)

Deep Reinforcement Learning (DRL) algorithms have achieved great success in solving many challenging tasks while their black-box nature hinders interpretability and real-world applicability, making it difficult for human experts to interpret and understand DRL policies. Existing works on interpretable reinforcement learning have shown promise in extracting decision tree (DT) based policies from DRL policies with most focus on the single-agent settings while prior attempts to introduce DT policies in multi-agent scenarios mainly focus on heuristic designs which do not provide any quantitative guarantees on the expected return. In this paper, we establish an upper bound on the return gap between the oracle expert policy and an optimal decision tree policy. This enables us to recast the DT extraction problem into a novel non-euclidean clustering problem over the local observation and action values space of each agent, with action values as cluster labels and the upper bound on the return gap as clustering loss. Both the algorithm and the upper bound are extended to multi-agent decentralized DT extractions by an iteratively-grow-DT procedure guided by an action-value function conditioned on the current DTs of other agents. Further, we propose the Return-Gap-Minimization Decision Tree (RGMDT) algorithm, which is a surprisingly simple design and is integrated with reinforcement learning through the utilization of a novel Regularized Information Maximization loss. Evaluations on tasks like D4RL show that RGMDT significantly outperforms heuristic DT-based baselines and can achieve nearly optimal returns under given DT complexity constraints (e.g., maximum number of DT nodes).

强化学习可解释性决策树多智能体

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。