arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2408.09112cs.LGcs.ROcs.SYeess.SY
收藏

使用基于集合的强化学习训练可验证的鲁棒智能体

Training Verifiably Robust Agents Using Set-Based Reinforcement Learning

Technical University of Munich(慕尼黑技术大学)

Manuel Wendl, Lukas Koller, Tobias Ladner, Matthias Althoff

AI总结:

研究利用基于集合的强化学习训练可验证鲁棒智能体的问题,通过传播整个扰动输入集,利用基于集合的策略梯度惩罚输出集大小,提升智能体对所有攻击的鲁棒性并提高验证性能。

摘要:

由深度神经网络参数化的强化学习策略在连续控制中取得了强大性能,但对输入扰动敏感。我们的工作解决了现有对抗训练方法与形式验证之间的差距,通过传播整个扰动输入集,利用基于集合的策略梯度惩罚输出集大小,使智能体对所有攻击具有鲁棒性,提升了验证性能。

英文摘要:

Reinforcement learning policies parametrized by deep neural networks have achieved strong performance for continuous control, yet even small input perturbations may lead to unpredictable behavior. This sensitivity limits their use in safety-critical domains, where robustness guarantees are required. Our work addresses this gap between state-of-the-art adversarial training methods and formal verification to train verifiably robust agents. Previous works train networks with individual adversarial perturbations, making them only robust against the specific adversarial attacks used. In contrast, our approach propagates entire perturbed input sets, enclosing all possible adversarial attacks within a single network pass. We leverage this to explicitly penalize the size of the output set (minimizing closed-loop uncertainty) and thereby make the actor robust against all possible attacks. This is realized by the use of set-based policy gradients, where each output within the set has a different gradient, thereby balancing the accuracy and robustness of the network. Doing so, we achieve formal verifiability across different verification frameworks for up to 9 times larger input perturbations compared to standard reinforcement learning and improve certified worst-case performance.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑