Causal Inference Q-Network: Toward Resilient Reinforcement Learning
Authors: Chao-Han Huck Yang, Danny I-Te Hung, Yi Ouyang, Pin-Yu Chen
Keywords: Deep Reinforcement Learning, Causal Inference, Robust Reinforcement Learning, Adversarial Robustness
Abstract: Deep reinforcement learning (DRL) has demonstrated impressive performance in various gaming simulators and real-world applications. In practice, however, a DRL agent may receive faulty observation by abrupt interferences such as black-out, frozen-screen, and adversarial perturbation. How to design a resilient DRL algorithm against these rare but mission-critical and safety-crucial scenarios is an important yet challenging task. In this paper, we consider a resilient DRL framework with observational interferences. Under this framework, we discuss the importance of the causal relation and propose a causal inference based DRL algorithm called causal inference Q-network (CIQ). We evaluate the performance of CIQ in several benchmark DRL environments with different types of interferences. Our experimental results show that the proposed CIQ method could achieve higher performance and more resilience against observational interferences.