Track papers across labs, topics, and trends.
Search papers, labs, and topics across Lattice.
This paper introduces PolicyGuard, a novel test-time step-level defense mechanism for reinforcement learning agents that addresses vulnerabilities to backdoor attacks. By leveraging Gaussian Process posterior variance, PolicyGuard computes uncertainty at individual time steps and adapts pseudo trajectories to enhance detection capabilities. Experimental results across seven RL games show that PolicyGuard achieves state-of-the-art performance, with average AUROC scores of 0.856 for perturbation-based attacks and 0.859 for adversary-agent attacks, highlighting its effectiveness in real-world applications.
PolicyGuard achieves an impressive average AUROC of 0.859 against adversary-agent attacks, setting a new benchmark for backdoor defense in reinforcement learning.
While real-world applications of reinforcement learning (RL) are becoming increasingly popular, the security of RL systems deserve more attention and exploration. In particular, recent work has revealed that RL agents are vulnerable to backdoor attacks, where a victim agent behaves normally under standard conditions but executes malicious actions when a specific trigger is activated. Existing backdoor defenses for RL either require access to the agent's internal parameters, operate only at the model or trajectory level, or are limited to specific attack types. To ensure the security of RL agents, we propose \texttt{PolicyGuard}, a \textit{test-time step-level} backdoor defense which leverages Gaussian Process (GP) posterior variance and adapts pseudo trajectories to enable uncertainty computation for individual time step. Besides, we also provide theoretical foundations to explain the efficacy of GP posterior variance. Extensive experiments across seven RL games demonstrate that PolicyGuard achieves state-of-the-art detection performance in most cases, with average AUROC of 0.856 for perturbation-based attacks and 0.859 for adversary-agent attacks.