Skip to content

Tuned DoubleDQN with prioritized experience replay - #302

Merged
muupan merged 9 commits into
chainer:masterfrom
muupan:replicate-prioritized-replay-example
Oct 1, 2018
Merged

Tuned DoubleDQN with prioritized experience replay#302
muupan merged 9 commits into
chainer:masterfrom
muupan:replicate-prioritized-replay-example

Conversation

@muupan

@muupan muupan commented Aug 31, 2018

Copy link
Copy Markdown
Member

Merge #301 first.

This PR improves train_dqn_ale.py:

  • use tuned DoubleDQN as better default settings (used in the Double DQN paper)
  • add --prioritized to enable prioritized experience replay
  • remove --activation and --use-sdl

@muupan muupan changed the title Use prioritized replay Tuned DoubleDQN with prioritized experience replay Aug 31, 2018
@muupan

muupan commented Aug 31, 2018

Copy link
Copy Markdown
Member Author

Below are preliminary results with --max-episode-len 4500. Results with --max-episode-len 27000 are not ready yet. Prioritized experience replay helps, and lr/4 stabilizes training. Results of Seaquest looks weird, though.

asterixnoframeskip-v4
beamridernoframeskip-v4
breakoutnoframeskip-v4
qbertnoframeskip-v4
seaquestnoframeskip-v4
spaceinvadersnoframeskip-v4

muupan added a commit to muupan/chainerrl that referenced this pull request Sep 15, 2018
@muupan

muupan commented Sep 18, 2018

Copy link
Copy Markdown
Member Author

Now the results with --max-episode-len 27000 are ready (except DQN). BeamRider's scores are affected by 4500 vs 27000.

Commands:

  • DQN (before this PR, only 10M steps): examples/ale/train_dqn_ale.py --env {env_id}
  • Tuned DoubleDQN + prioritized replay lr/4 (after this PR): examples/ale/train_dqn_ale.py --eval-interval 1000000 --prioritized --lr 6.25e-5 --env {env_id}
  • Tuned DoubleDQN + prioritized replay (after this PR): examples/ale/train_dqn_ale.py --eval-interval 1000000 --prioritized --env {env_id}
  • Tuned DoubleDQN (after this PR): examples/ale/train_dqn_ale.py --eval-interval 1000000 --env {env_id}

Each configuration was ran with three different random seeds. The figures below show the average evaluation scores with confidence bounds are shown.

asterixnoframeskip-v4
beamridernoframeskip-v4
breakoutnoframeskip-v4
qbertnoframeskip-v4
seaquestnoframeskip-v4
spaceinvadersnoframeskip-v4

@muupan

muupan commented Oct 1, 2018

Copy link
Copy Markdown
Member Author

I'll merge it because it's reviewed and approved by @prabhatnagarajan

@muupan
muupan merged commit 24b1b1c into chainer:master Oct 1, 2018

@prabhatnagarajan prabhatnagarajan left a comment

Copy link
Copy Markdown
Contributor

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

LGTM.

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

Projects

None yet

Development

Successfully merging this pull request may close these issues.

2 participants