Skip to content
Repository for our ICML 2019 paper: Curiosity-Bottleneck
Python
Branch: master
Clone or download
Fetching latest commit…
Cannot retrieve the latest commit at this time.
Permalink
Type Name Latest commit message Commit time
Failed to load latest commit information.
assets
baselines
policies
.gitignore
README.md
atari_wrappers.py
cmd_util.py
console_util.py
monitor.py
mpi_util.py
ppo_agent.py
requirements.txt
run_atari.py
stochastic_policy.py
tf_util.py
utils.py
vec_env.py

README.md

Curiosity-Bottleneck:
Exploration by Distilling Task-Specific Novelty

This project hosts the code for our ICML 2019 paper.

Model

Authors

Youngjin Kim, Wontae Nam*, Hyunwoo Kim*, Jihoon Kim, Gunhee Kim (*equal contribution)

Vision and Learning Lab. @ Computer Science and Engineering, Seoul National University
Clova, NAVER

Installation

pip install -r requirements

Run

The following command should train a PPO agent with Curiosity-Bottleneck on Gravitar.

python run_atari.py

Acknowledgements

This code is based on the RND implementation by Yuri Burda.

Citation

@inproceedings{
	kim2019curiositybottleneck,
	title={Curiosity-Bottleneck: Exploration by Distilling Task-Specific Novelty},
	author={Youngjin Kim and Wontae Nam and Hyunwoo Kim and Jihoon Kim and Gunhee Kim},
	booktitle={International Conference on Machine Learning},
	year={2019}
}
You can’t perform that action at this time.