arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2201.08520cs.LGcs.AIcs.CL

Learning Two-Step Hybrid Policy for Graph-Based Interpretable Reinforcement Learning

University of California San Diego(加州大学圣地亚哥分校);Amazon(亚马逊)

Tongzhou Mu, Kaixiang Lin, Feiyang Niu, Govind Thattai

英文摘要:

We present a two-step hybrid reinforcement learning (RL) policy that is designed to generate interpretable and robust hierarchical policies on the RL problem with graph-based input. Unlike prior deep reinforcement learning policies parameterized by an end-to-end black-box graph neural network, our approach disentangles the decision-making process into two steps. The first step is a simplified classification problem that maps the graph input to an action group where all actions share a similar semantic meaning. The second step implements a sophisticated rule-miner that conducts explicit one-hop reasoning over the graph and identifies decisive edges in the graph input without the necessity of heavy domain knowledge. This two-step hybrid policy presents human-friendly interpretations and achieves better performance in terms of generalization and robustness. Extensive experimental studies on four levels of complex text-based games have demonstrated the superiority of the proposed method compared to the state-of-the-art.

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

↑