CAMP in the Odyssey: Provably Robust Reinforcement Learning with Certified Radius Maximization

Fuente: arXiv
Enregistré dans:
Détails bibliographiques
Auteurs principaux: Wang, Derui, Moore, Kristen, Goel, Diksha, Kim, Minjune, Li, Gang, Li, Yang, Doss, Robin, Xue, Minhui, Li, Bo, Camtepe, Seyit, Zhu, Liming
Format: Preprint
Publié: 2025
Sujets:
Accès en ligne:
Tags: Ajouter un tag
Pas de tags, Soyez le premier à ajouter un tag!
_version_ 1866910898756517888
author Wang, Derui
Moore, Kristen
Goel, Diksha
Kim, Minjune
Li, Gang
Li, Yang
Doss, Robin
Xue, Minhui
Li, Bo
Camtepe, Seyit
Zhu, Liming
author_facet Wang, Derui
Moore, Kristen
Goel, Diksha
Kim, Minjune
Li, Gang
Li, Yang
Doss, Robin
Xue, Minhui
Li, Bo
Camtepe, Seyit
Zhu, Liming
contents Deep reinforcement learning (DRL) has gained widespread adoption in control and decision-making tasks due to its strong performance in dynamic environments. However, DRL agents are vulnerable to noisy observations and adversarial attacks, and concerns about the adversarial robustness of DRL systems have emerged. Recent efforts have focused on addressing these robustness issues by establishing rigorous theoretical guarantees for the returns achieved by DRL agents in adversarial settings. Among these approaches, policy smoothing has proven to be an effective and scalable method for certifying the robustness of DRL agents. Nevertheless, existing certifiably robust DRL relies on policies trained with simple Gaussian augmentations, resulting in a suboptimal trade-off between certified robustness and certified return. To address this issue, we introduce a novel paradigm dubbed \texttt{C}ertified-r\texttt{A}dius-\texttt{M}aximizing \texttt{P}olicy (\texttt{CAMP}) training. \texttt{CAMP} is designed to enhance DRL policies, achieving better utility without compromising provable robustness. By leveraging the insight that the global certified radius can be derived from local certified radii based on training-time statistics, \texttt{CAMP} formulates a surrogate loss related to the local certified radius and optimizes the policy guided by this surrogate loss. We also introduce \textit{policy imitation} as a novel technique to stabilize \texttt{CAMP} training. Experimental results demonstrate that \texttt{CAMP} significantly improves the robustness-return trade-off across various tasks. Based on the results, \texttt{CAMP} can achieve up to twice the certified expected return compared to that of baselines. Our code is available at https://github.com/NeuralSec/camp-robust-rl.
format Preprint
id arxiv_https___arxiv_org_abs_2501_17667
institution arXiv
publishDate 2025
record_format arxiv
spellingShingle CAMP in the Odyssey: Provably Robust Reinforcement Learning with Certified Radius Maximization
Wang, Derui
Moore, Kristen
Goel, Diksha
Kim, Minjune
Li, Gang
Li, Yang
Doss, Robin
Xue, Minhui
Li, Bo
Camtepe, Seyit
Zhu, Liming
Machine Learning
Cryptography and Security
Deep reinforcement learning (DRL) has gained widespread adoption in control and decision-making tasks due to its strong performance in dynamic environments. However, DRL agents are vulnerable to noisy observations and adversarial attacks, and concerns about the adversarial robustness of DRL systems have emerged. Recent efforts have focused on addressing these robustness issues by establishing rigorous theoretical guarantees for the returns achieved by DRL agents in adversarial settings. Among these approaches, policy smoothing has proven to be an effective and scalable method for certifying the robustness of DRL agents. Nevertheless, existing certifiably robust DRL relies on policies trained with simple Gaussian augmentations, resulting in a suboptimal trade-off between certified robustness and certified return. To address this issue, we introduce a novel paradigm dubbed \texttt{C}ertified-r\texttt{A}dius-\texttt{M}aximizing \texttt{P}olicy (\texttt{CAMP}) training. \texttt{CAMP} is designed to enhance DRL policies, achieving better utility without compromising provable robustness. By leveraging the insight that the global certified radius can be derived from local certified radii based on training-time statistics, \texttt{CAMP} formulates a surrogate loss related to the local certified radius and optimizes the policy guided by this surrogate loss. We also introduce \textit{policy imitation} as a novel technique to stabilize \texttt{CAMP} training. Experimental results demonstrate that \texttt{CAMP} significantly improves the robustness-return trade-off across various tasks. Based on the results, \texttt{CAMP} can achieve up to twice the certified expected return compared to that of baselines. Our code is available at https://github.com/NeuralSec/camp-robust-rl.
title CAMP in the Odyssey: Provably Robust Reinforcement Learning with Certified Radius Maximization
topic Machine Learning
Cryptography and Security
url https://arxiv.org/abs/2501.17667