让程序化强化学习直接学成离散代码,省去调优步骤。
DiPRL: Learning Discrete Programmatic Policies via Architecture Entropy Regularization

- 用架构熵正则化让程序在训练中自发趋于离散。
- 避免后期硬转换损失性能,直接得到可读可改的策略程序。
- 适合需要透明决策过程的复杂任务,如机器人控制。
程序化强化学习(PRL)通过将策略表示为人类可读、可编辑的程序,提供了可解释的深度强化学习替代方案。尽管已有基于梯度的方法优化程序的连续松弛版本,但在将其转换回离散程序时仍面临显著性能下降。后处理离散化会丢弃已优化的分支和参数,导致策略表达能力萎缩、任务表现降低,进而需额外微调。为此,我们提出可微离散程序化强化学习(DiPRL),一种在训练过程中即趋近于几乎离散的程序策略学习方法,无需单独后处理微调阶段。我们首先分析了基于梯度方法因后处理离散化带来的性能下降风险,随后引入程序架构熵正则化,实现平滑、可微训练,促进收敛至离散程序。DiPRL 在保持梯度优化效率的同时,缓解了后处理离散化的风险。在多个离散与连续强化学习任务上的实验表明,DiPRL 能通过可解释的程序化策略实现强性能。
原文摘要 · Abstract (English)
Programmatic reinforcement learning (PRL) offers an interpretable alternative to deep reinforcement learning by representing policies as human-readable and -editable programs. While gradient-based methods have been developed to optimize continuous relaxations of programs, they face a significant performance drop when converting the continuous relaxations back into discrete programs. Post-hoc discretization can discard optimized branches and parameters in a program, which results in a collapse of policy expressivity and lowered task performance, leading in turn to a need for additional fine-tuning. To overcome these limitations, we propose Differentiable Discrete Programmatic Reinforcement Learning (DiPRL), a method that learns programmatic policies that become nearly discrete during training, avoiding a separate post-hoc fine-tuning stage. We first analyze the inherent risks of performance drop introduced by post-hoc discretization of gradient-based methods. Then, we introduce programmatic architecture entropy regularization, which enables smooth, differentiable training that encourages convergence toward a discrete program. DiPRL maintains the efficiency of gradient-based optimization while mitigating the risks of post-hoc discretization. Our experiments across multiple discrete and continuous RL tasks demonstrate that DiPRL can achieve strong performance via interpretable programmatic policies.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。