Spring 2022 CS 444

Assignment 5: Deep Reinforcement Learning

Due date: Friday, May 5, 11:59:59PM


In this assignment, you will implement the famous Deep Q-Network (DQN) and its successor Double DQN on the game of Atari Breakout using the OpenAI Gym. The goals of this assignment are to (1) understand how deep reinforcement learning works when interacting with the pixel-level information of an environment and (2) implement a recurrent state to encode and maintain history.

Download the starting code here.

The top-level notebook (MP5.ipynb) will guide you through all the steps of the DQN. You will mainly implement the training of the Agent in the agent.py file for DQN, and agent_double.py for double DQN. We provide you with the neural network. Do NOT change the architecture of the neural network (for consistency of grading). Due to the computational constraints, we only expect you to reach a mean score of 10 (should take around 2000 episodes). config.py contains most of the hyperparemeters. You may play around with these parameters if you want, but the provided parameters will be enough to reach the desired score.

Note, as you look in the iPython notebook, in our terminology, a single episode is a game played by the agent till it loses all its lives (in this case, your agent has 5 lives). In the paper, however, an episode refers to almost 30 minutes of training on the GPU and such training is not feasible for us.

Below is a more thorough example of expected rewards vs. number of episodes to help with your debugging.

  • 200 episodes: 1.5
  • 400 episodes: 1.5
  • 600 episodes: 1.5
  • 1000 episodes: 1.75
  • 1200 episodes: 2.5
  • 1400 episodes: 3.5
  • 1600 episodes: 5.0

Your goal is to have either agent.py or agent_double.py reach an evaluation score of 10. To have a fair comparison in the report, we ask you to run both files the same amount of episodes, but only one model is required to reach the evaluation score of 10.

We recommend that you look at the following links:

We highly recommend that you understand the Official DQN PyTorch tutorial before starting this assignment. This will give you a great starting point to implement DQN and Double DQN as the tutorial implements a version of double DQN for cartpole! However, we expect you to follow our code instructions and implement code in our format. Uploading code that does not follow our format will receive a zero.

This is a computationally expensive assignment. It is expected that your code should run for at least 4 hours to complete 2000 episodes. You can stop training early if you reach a mean score of 10 in the game. As mentioned, we will be providing some initial expectations of score values with respect to episodes on Campuswire.

This assignment requires a GPU, so use your Google Cloud credits (colab could work for this assignment as well).

Extra Credit

  • Train a DQN agent for one or more additional Atari games from OpenAI gym and report on any implementation/hyperparameter changes you had to make, and your agent's performance.

  • Implement policy gradient training or another advanced RL method for Breakout or another Atari game and compare performance (including convergence speed) to your DQN method. You need to write your own code from scratch, not train an off-the-shelf method.

Environment Setup

The assignment is given to you in the MP5.ipynb file. If you are using a local machine, ensure that iPython is installed (https://ipython.org/install.html). You may then navigate the assignment directory in terminal and start a local iPython server using the jupyter notebook command. Instructions to install dependencies are provided at the top of the notebook. Please use environment with Python 3.7. These instructions should work for local machines, Google Cloud, and Google Colab. We have tested this assignment on PyTorch version 1.10.2, so please install this version if there are other dependency issues.

If you will be working on the assignment on a local machine then you will need a Python environment set up with the appropriate packages. We suggest that you use Conda to manage Python package dependencies (https://conda.io/docs/user-guide/getting-started.html).

Unless you have a machine with a GPU, running this assignment on your local machine will be very slow and is not recommended.

Submission Instructions

This is your last assignment, so feel free to use up your remaining late days if you so choose!

  1. All of your code (Python files and ipynb file) in a single ZIP file. The filename should be netid1_netid2_mp5_code.zip.
  2. Upload your policy net model (in .pth format) as a separate file.
  3. Your iPython notebooks with output cells converted to PDF format. The filename should be netid1_netid2_mp5_output.pdf.
  4. A brief report in PDF format using this template. The filename you submit should be netid1_netid2_mp5_report.pdf.

Please refer to course policies on collaborations, late submission, etc.