State-Aware Perturbation Optimization for Robust Deep Reinforcement Learning
Zongyuan Zhang, Tianyang Duan, Zheng Lin, Dong Huang, Zihan Fang, Zekai Sun, Ling Xiong, Hongbin Liang, Heming Cui, Yong Cui
TL;DR
This work addresses the vulnerability of deep reinforcement learning (DRL) policies in robotic control to environmental perturbations by introducing the Adversarial Victim-Dynamics Markov Decision Process (AVD-MDP) for modeling attacker–victim interactions over time. Building on this theory, the authors present STAR, a state-aware white-box attack that combines a soft-mask state-targeting mechanism with an information-theoretic objective to maximize mutual information between perturbations, states, and actions, thereby achieving stealthy perturbations and dispersed state visitation under a fixed budget. The approach is evaluated on quadruped locomotion tasks in RaiSim with PPO-based victims, where STAR consistently outperforms existing white-box attacks in degrading reward, forward velocity, and stability, and also enables effective adversarial defense through training. The results demonstrate both the practical potency of STAR and its value as a framework for rigorously testing and improving DRL robustness in real-world robotic systems.
Abstract
Recently, deep reinforcement learning (DRL) has emerged as a promising approach for robotic control. However, the deployment of DRL in real-world robots is hindered by its sensitivity to environmental perturbations. While existing whitebox adversarial attacks rely on local gradient information and apply uniform perturbations across all states to evaluate DRL robustness, they fail to account for temporal dynamics and state-specific vulnerabilities. To combat the above challenge, we first conduct a theoretical analysis of white-box attacks in DRL by establishing the adversarial victim-dynamics Markov decision process (AVD-MDP), to derive the necessary and sufficient conditions for a successful attack. Based on this, we propose a selective state-aware reinforcement adversarial attack method, named STAR, to optimize perturbation stealthiness and state visitation dispersion. STAR first employs a soft mask-based state-targeting mechanism to minimize redundant perturbations, enhancing stealthiness and attack effectiveness. Then, it incorporates an information-theoretic optimization objective to maximize mutual information between perturbations, environmental states, and victim actions, ensuring a dispersed state-visitation distribution that steers the victim agent into vulnerable states for maximum return reduction. Extensive experiments demonstrate that STAR outperforms state-of-the-art benchmarks.
