Knowledge Transfer for Deep Reinforcement Learning with Hierarchical Experience Replay

By Haiyan Yin, Sinno Jialin Pan. 2017

124 citations recorded by the source corpus.

The process for transferring knowledge of multiple reinforcement learning policies into a single multi-task policy via distillation technique is known as policy distillation. When policy distillation is under a deep reinforcement learning setting, due to the giant parameter size and the huge state space for each task domain, it requires extensive computational efforts to train the multi-task policy network. In this paper, we propose a new policy distillation architecture for deep reinforcement learning, where we assume that each task uses its task-specific high-level convolutional features as the inputs to the multi-task policy network. Furthermore, we propose a new sampling framework termed hierarchical prioritized experience replay to selectively choose experiences from the replay memories of each task domain to perform learning on the network. With the above two attempts, we aim to accelerate the learning of the multi-task policy network while guaranteeing a good performance. We use Atari 2600 games as testing environment to demonstrate the efficiency and effectiveness of our proposed solution for policy distillation

Subfields: Artificial Intelligence, Cognitive Neuroscience, Computational Theory and Mathematics

How researchers see this paper

The researchers whose trusted networks rank it highest, strongest advocate first. Every researcher gets a different order.

In Artificial Intelligence

  1. Joelle Pineau
  2. Amy Zhang
  3. Shagun Sodhani
  4. F. Meier
  5. Kaixiang Lin
  6. Thanh Thi Nguyen
  7. Jiayu Zhou
  8. Anil K. Jain

In Cognitive Neuroscience

  1. Kaixiang Lin
  2. Joelle Pineau
  3. Anil K. Jain
  4. Jiayu Zhou
  5. Saeid Nahavandi
  6. R. Meir
  7. Amy Zhang
  8. Shagun Sodhani

In Computational Theory and Mathematics

  1. Thanh Thi Nguyen
  2. Joelle Pineau
  3. Anil K. Jain
  4. Jiayu Zhou
  5. Roger Wattenhofer
  6. Saeid Nahavandi
  7. R. Meir
  8. Amy Zhang

Citations in the corpus