用知识和梯度引导强化学习,提升参数化动作决策的效率与安全性。
Knowledge- and Gradient-Guided Reinforcement Learning for Parametrized Action Markov Decision Processes

- 结合领域知识库动态筛选可行动作与参数范围
- 梯度优化细化参数,训练样本效率显著提升
- 适合需安全约束的复杂决策场景,如工业控制
本文研究参数化动作马尔可夫决策过程(PAMDP)中的强化学习问题,其中每个决策包含符号动作和数值参数。传统方法使用单次估计器确定参数,导致样本效率低下。尽管多数PAMDP环境存在不完整但明确的领域知识(如规则、安全约束或专家启发式),却很少被直接用于提升训练效率。为此,本文提出神经符号知识与梯度引导强化学习(KGRL)算法。KGRL利用Datalog知识库推导给定状态下的可适用动作与可行参数集,从而剪枝无效动作并约束参数空间。通过梯度驱动的参数精炼循环,在训练与部署中估计最优参数。同时,记录轨迹中激活的规则,提供局部过程解释。整体上,KGRL引导探索与部署向可行且约束感知的方向进行,并提升训练样本效率。在多个PAMDP任务中,KGRL在样本效率与累计回报上均优于现有先进基线。
原文摘要 · Abstract (English)
In this paper, we study Reinforcement Learning in Parametrized Action Markov Decision Processes (PAMDP), where each decision consists of a symbolic action and numerical parameters. In such settings Reinforcement Learning algorithms typically determine parameters with one-shot estimators, which makes their training sample inefficient. Though in most PAMDP environments explicit but incomplete knowledge (e.g., rules, safety constraints, or expert heuristics) is available, it is rarely directly used to increase the sample-efficiency of training Reinforcement Learning agents. We step into this gap and propose our novel Neuro-Symbolic Knowledge- and Gradient-Guided Reinforcement Learning (KGRL) algorithm. KGRL uses domain knowledge in a Datalog knowledge base to derive the set of applicable actions and feasible parameters for a given state. This allows it to prune non-applicable actions from the decision-space and constrain the parameter spaces of the remaining actions. We then use a gradient-based parameter refinement loop to estimate the optimal parameters during training and deployment of the agent. By recording activated rules along the trajectory, KGRL additionally provides local procedural explanations on the pruning of actions and constraining of parameters. Overall, KGRL guides the agent's exploration and deployment toward feasible and constraint-aware decisions, while increasing sample efficiency during training. KGRL outperforms state-of-the-art RL baselines for PAMDPs in both, sample efficiency and episodic return.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。