`

Timezone: »

 
Poster
An Equivalence between Loss Functions and Non-Uniform Sampling in Experience Replay
Scott Fujimoto · David Meger · Doina Precup

Tue Dec 08 09:00 AM -- 11:00 AM (PST) @ Poster Session 1 #569

Prioritized Experience Replay (PER) is a deep reinforcement learning technique in which agents learn from transitions sampled with non-uniform probability proportionate to their temporal-difference error. We show that any loss function evaluated with non-uniformly sampled data can be transformed into another uniformly sampled loss function with the same expected gradient. Surprisingly, we find in some environments PER can be replaced entirely by this new loss function without impact to empirical performance. Furthermore, this relationship suggests a new branch of improvements to PER by correcting its uniformly sampled loss function equivalent. We demonstrate the effectiveness of our proposed modifications to PER and the equivalent loss function in several MuJoCo and Atari environments.

Author Information

Scott Fujimoto (McGill University)
David Meger (McGill University)
Doina Precup (McGill University / Mila / DeepMind Montreal)

More from the Same Authors