arXiv:2604.01613cs.LG2026-04被引 1

提出一种抗噪声TD误差的伪量化强化学习算法。

Pseudo-Quantized Actor-Critic Algorithm for Robustness to Noisy Temporal Difference Error

  • 用逻辑函数建模最优性分布,通过梯度消失抑制噪声干扰
  • 多级伪量化机制降低噪声影响,提升学习稳定性
  • 无需额外网络,适合对鲁棒性要求高的强化学习场景

在强化学习中,时序差分(TD)误差广泛用于优化价值和策略函数。但由于其基于自举法计算,常存在噪声并导致学习不稳。现有改进方法如目标网络和集成模型虽有效,但带来计算开销大、学习效率下降等副作用。本文基于控制即推理框架重新审视TD学习,提出一种新型算法:以逻辑函数表示最优性二元随机变量分布,结合前向与反向KL散度,推导出鲁棒学习规则——当逻辑函数因大误差饱和时,梯度自动消失,隐式排除噪声数据。两种散度具有不同的梯度消失特性。在此基础上,将最优性分解为多层级,实现对TD误差的伪量化,进一步抑制噪声。此外,近似推导出基于Jensen-Shannon散度的方法,继承两类散度的优势。在多个强化学习基准测试中验证,即使缺乏传统启发式方法或奖励含噪,仍能实现稳定学习。

原文摘要 · Abstract (English)

In reinforcement learning (RL), temporal difference (TD) errors are widely adopted for optimizing value and policy functions. However, since the TD error is defined by a bootstrap method, its computation tends to be noisy and destabilize learning. Heuristics to improve the accuracy of TD errors, such as target networks and ensemble models, have been introduced so far. While these are essential approaches for the current deep RL algorithms, they cause side effects like increased computational cost and reduced learning efficiency. Therefore, this paper revisits the TD learning algorithm based on control as inference, deriving a novel algorithm capable of robust learning against noisy TD errors. First, the distribution model of optimality, a binary random variable, is represented by a sigmoid function. Alongside forward and reverse Kullback-Leibler divergences, this new model derives a robust learning rule: when the sigmoid function saturates with a large TD error probably due to noise, the gradient vanishes, implicitly excluding it from learning. Furthermore, the two divergences exhibit distinct gradient-vanishing characteristics. Building on these analyses, the optimality is decomposed into multiple levels to achieve pseudo-quantization of TD errors, aiming for further noise reduction. Additionally, a Jensen-Shannon divergence-based approach is approximately derived to inherit the characteristics of both divergences. These benefits are verified through RL benchmarks, demonstrating stable learning even when heuristics are insufficient or rewards contain noise.

强化学习鲁棒性伪量化TD误差

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