arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2202.03609cs.LGcs.AI
收藏

PolicyCleanse:强化学习中的后门检测与缓解

PolicyCleanse: Backdoor Detection and Mitigation in Reinforcement Learning

University of Maryland, College Park(马里兰大学帕克分校);Simular Research(Simular研究院);Northwestern University(西北大学);UC Riverside(加州大学河滨分校)

Junfeng Guo, Ang Li, Cong Liu

AI总结:

针对多智能体强化学习中的特洛伊木马后门威胁,本文提出PolicyCleanse方法,利用特洛伊智能体累积奖励下降的特性检测后门及触发动作,并结合机器遗忘技术缓解后门,实验证明其检测准确且胜率比基线至少高3%。

摘要:

尽管强化学习的现实应用正日益普及,但RL系统的安全性和鲁棒性仍值得更多关注与探索。特别是最近的研究表明,在多智能体RL环境中,后门触发动作可以被注入到受害者智能体(又称特洛伊木马智能体)中,一旦其观察到该后门触发动作,就会导致灾难性的失败。为了确保RL智能体免受恶意后门威胁,本文提出了多智能体竞争强化学习系统中的后门检测问题,旨在检测特洛伊智能体及其对应的潜在触发动作,并进一步尝试缓解其特洛伊行为。为解决此问题,我们提出了PolicyCleanse,它基于被激活的特洛伊智能体在若干时间步后累积奖励会显著下降这一特性。伴随PolicyCleanse,我们还设计了一种基于机器遗忘的方法,能够有效缓解检测到的后门。大量实验表明,所提方法能准确检测特洛伊智能体,并在多种类型的智能体和环境中,其胜率比现有的后门缓解基线方法至少高出3%。

英文摘要:

While real-world applications of reinforcement learning are becoming popular, the security and robustness of RL systems are worthy of more attention and exploration. In particular, recent works have revealed that, in a multi-agent RL environment, backdoor trigger actions can be injected into a victim agent (a.k.a. Trojan agent), which can result in a catastrophic failure as soon as it sees the backdoor trigger action. To ensure the security of RL agents against malicious backdoors, in this work, we propose the problem of Backdoor Detection in a multi-agent competitive reinforcement learning system, with the objective of detecting Trojan agents as well as the corresponding potential trigger actions, and further trying to mitigate their Trojan behavior. In order to solve this problem, we propose PolicyCleanse that is based on the property that the activated Trojan agents accumulated rewards degrade noticeably after several timesteps. Along with PolicyCleanse, we also design a machine unlearning-based approach that can effectively mitigate the detected backdoor. Extensive experiments demonstrate that the proposed methods can accurately detect Trojan agents, and outperform existing backdoor mitigation baseline approaches by at least 3% in winning rate across various types of agents and environments.

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

↑