提出衡量强化学习智能体理性的新方法,揭示其泛化能力与环境变化的关系。
Rationality Measurement and Theory for Reinforcement Learning Agents
- 定义部署时最优动作方向为理性基准,以轨迹累积偏差衡量理性风险
- 理性风险差由环境分布偏移和算法泛化能力共同决定,分别受Wasserstein距离与Rademacher复杂度约束
- 实验验证正则化有益、环境漂移有害,适合关注模型可靠性与泛化性的研究者
本文提出一套用于强化学习智能体的理性度量及其理论框架,这一属性日益重要却鲜有研究。我们定义部署阶段使隐藏真实价值函数沿最陡方向最大化的动作为完全理性动作。策略动作与其理性对应动作之间的期望价值差异,按部署轨迹累积定义为期望理性风险;训练中对应的采样平均版本亦被定义。二者之差称为理性风险差,可分解为:(1) 由训练与部署间环境变化引起的外在成分,(2) 算法在动态环境中泛化能力导致的内在成分。这两部分分别被1-Wasserstein距离(过渡核与初始状态分布之间)和价值函数类的经验Rademacher复杂度所上界控制。理论推导揭示了正则化(如层归一化、ℓ₂正则化、权重归一化)及领域随机化的潜在益处,以及环境漂移带来的危害。实验结果与这些假设高度一致。代码已公开于 https://github.com/EVIEHub/Rationality。
原文摘要 · Abstract (English)
This paper proposes a suite of rationality measures and associated theory for reinforcement learning agents, a property increasingly critical yet rarely explored. We define an action in deployment to be perfectly rational if it maximises the hidden true value function in the steepest direction. The expected value discrepancy of a policy's actions against their rational counterparts, culminating over the trajectory in deployment, is defined to be expected rational risk; an empirical average version in training is also defined. Their difference, termed as rational risk gap, is decomposed into (1) an extrinsic component caused by environment shifts between training and deployment, and (2) an intrinsic one due to the algorithm's generalisability in a dynamic environment. They are upper bounded by, respectively, (1) the $1$-Wasserstein distance between transition kernels and initial state distributions in training and deployment, and (2) the empirical Rademacher complexity of the value function class. Our theory suggests hypotheses on the benefits from regularisers (including layer normalisation, $\ell_2$ regularisation, and weight normalisation) and domain randomisation, as well as the harm from environment shifts. Experiments are in full agreement with these hypotheses. The code is available at https://github.com/EVIEHub/Rationality.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。