/rl-detect-and-denoise-defense

[ECML 22'] Defending Observation Attacks in Deep Reinforcement Learning via Detection and Denoising

Primary LanguagePython

Defense Adversarial Attacks in Deep Reinforcement Learning via Detecting and Denoising

Code repository of ECML-PKDD 22' paper Defending Observation Attacks in Deep Reinforcement Learning via Detection and Denoising

ArXiv

Overview

image

Structure

├── neural_shield
│   ├── attack          # attack algorithms
│   │   ├── common
│   │   ├── offline     # offline attacks
│   │   └── online      # online attacks
│   ├── benchmark       # robots in simulation
│   ├── config.py       # config file, including data path, simulation, attack, and defense parameters
│   ├── controller      # pre-trained controller loader
│   ├── defense         # detect-and-denoise defense
│   └── evaluation      # evaluation scripts
└── README.MD

Quick Start

All the attack and defense related functions are summarized in
neural_shield/evaluation/run.py

Bibtex

@inproceedings{xiong2022defending,
  title={Defending Observation Attacks in Deep Reinforcement Learning via Detection and Denoising},
  author={Xiong, Zikang and Eappen, Joe and Zhu, He and Jagannathan, Suresh},
  booktitle={2022 European Conference on Machine Learning and Principles and Practice of Knowledge Discovery in Databases},
  year={2022}
}