arXiv:2409.03237cs.LGcs.SY2024-09中稿 · the Decision and C…被引 2

提出抗奖励污染的鲁棒Q-learning算法,保障在恶意干扰下仍能稳定学习。

Robust Q-Learning under Corrupted Rewards

  • 利用历史奖励数据构建鲁棒的贝尔曼算子,抵抗部分观测奖励被篡改
  • 在对抗污染比例ε下,收敛误差仅增加可忽略的O(ε)项
  • 适用于奖励分布无界但二阶矩有界的复杂环境,适合高可靠性场景

近期对无模型强化学习算法的非渐近行为分析日益增多,但其在非理想环境(如存在污染奖励)下的表现仍不清楚。本文研究了经典Q-learning算法在强污染攻击模型下的鲁棒性,其中对手可任意扰动少量观测奖励。我们证明,原始Q-learning可能产生任意大的误差。为此,提出一种新型鲁棒同步Q-learning算法,通过历史奖励数据构造每步的鲁棒经验贝尔曼算子。理论证明该算法具有有限时间收敛速率,在无攻击情况下达到现有最优边界,仅增加与对抗污染比例ε相关的微小不可避免误差O(ε)。值得注意的是,即使真实奖励分布支持集无限,只要二阶矩有界,结论依然成立。

原文摘要 · Abstract (English)

Recently, there has been a surge of interest in analyzing the non-asymptotic behavior of model-free reinforcement learning algorithms. However, the performance of such algorithms in non-ideal environments, such as in the presence of corrupted rewards, is poorly understood. Motivated by this gap, we investigate the robustness of the celebrated Q-learning algorithm to a strong-contamination attack model, where an adversary can arbitrarily perturb a small fraction of the observed rewards. We start by proving that such an attack can cause the vanilla Q-learning algorithm to incur arbitrarily large errors. We then develop a novel robust synchronous Q-learning algorithm that uses historical reward data to construct robust empirical Bellman operators at each time step. Finally, we prove a finite-time convergence rate for our algorithm that matches known state-of-the-art bounds (in the absence of attacks) up to a small inevitable $O(\varepsilon)$ error term that scales with the adversarial corruption fraction $\varepsilon$. Notably, our results continue to hold even when the true reward distributions have infinite support, provided they admit bounded second moments.

强化学习鲁棒性Q-learning对抗攻击

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