BehaviorGuard: Online Backdoor Defense for Deep Reinforcement Learning
About
Backdoor attacks pose a serious threat to deep reinforcement learning (DRL). Current defenses typically rely on reward anomalies to reverse-engineer triggers and model finetuning to remove backdoors. However, complex trigger patterns undermine their robustness, and fine-tuning entails high costs, limiting practical utility. Therefore, we shift defense concerns to trigger-agnostic backdoor output behaviors and propose BehaviorGuard, an online behavior-based backdoor detection and mitigation framework for DRL. Specifically, we find that regardless of attacks, backdoored policies induce consistent shifts in action distributions to ensure reliable activation, leaving detectable traces in high-quantile regions and distribution tails, even in the absence of triggers. Based on this, we design a novel metric that captures behavioral drift in action distributions to identify and suppress backdoor actions at runtime. To our knowledge, this is the first online backdoor defense that counters attacks both in single- and multi-agent DRL. Evaluated across diverse benchmarks with different backdoor attacks, BehaviorGuard consistently surpasses prior methods in both efficacy and efficiency.
Related benchmarks
| Task | Dataset | Result | Rank | |
|---|---|---|---|---|
| Single-agent Reinforcement Learning (SARL) | SpaceInvaders Clean | Average Episode Return775.1 | 9 | |
| Single-agent Reinforcement Learning (SARL) | Breakout Poisoned | Average Episode Return517.8 | 9 | |
| Single-agent Reinforcement Learning (SARL) | SpaceInvaders Poisoned | Average Episode Return783.2 | 9 | |
| Single-agent Reinforcement Learning (SARL) | Breakout Clean | Average Episode Return465.2 | 9 | |
| Cooperative Multi-agent Reinforcement Learning (coo-MARL) | COMA Poisoned | Average Winning Rate94.2 | 8 | |
| Cooperative Multi-agent Reinforcement Learning (coo-MARL) | QMIX Poisoned | Average Winning Rate96.9 | 8 | |
| Cooperative Multi-agent Reinforcement Learning (coo-MARL) | COMA Clean | Average Winning Rate95.3 | 8 | |
| Cooperative Multi-agent Reinforcement Learning (coo-MARL) | QMIX Clean | Average Winning Rate98.5 | 8 | |
| Single-agent Reinforcement Learning (SARL) | Pong Clean | Average Episode Return18.5 | 7 | |
| Single-agent Reinforcement Learning (SARL) | Pong Poisoned | Average Episode Return18.2 | 7 |