← Search

Sherjil Ozair

9 accepted papers

2024

Genie: Generative Interactive Environments

ICML 2024oral

We introduce Genie, the first *generative interactive environment* trained in an unsupervised manner from unlabelled Internet videos. The model can be prompted to generate an endless variety of action-controllable virtual worlds described through text, synthetic images, photographs, and even sketche…

Cited by 172SourcePDFScholar
2022

Planning in Stochastic Environments with a Learned Model

ICLR 2022spotlight

Model-based reinforcement learning has proven highly successful. However, learning a model in isolation from its use during planning is problematic in complex environments. To date, the most effective techniques have instead combined value-equivalent model learning with powerful tree-search methods.…

Cited by 85SourcePDFScholar
2022

Procedural generalization by planning with self-supervised world models

ICLR 2022poster

One of the key promises of model-based reinforcement learning is the ability to generalize using an internal model of the world to make predictions in novel environments and tasks. However, the generalization ability of model-based agents is not well understood because existing work has focused on m…

Cited by 39SourcePDFScholar
2021

Vector Quantized Models for Planning

ICML 2021spotlight

Recent developments in the field of model-based RL have proven successful in a range of environments, especially ones where planning is essential. However, such successes have been limited to deterministic fully-observed environments. We present a new approach that handles stochastic and partially-o…

Cited by 62SourcePDFScholar
2019

On Variational Bounds of Mutual Information

ICML 2019oral

Estimating and optimizing Mutual Information (MI) is core to many problems in machine learning, but bounding MI in high dimensions is challenging. To establish tractable and scalable objectives, recent work has turned to variational bounds parameterized by neural networks. However, the relationships…

2019

Unsupervised State Representation Learning in Atari

NeurIPS 2019poster

State representation learning, or the ability to capture latent generative factors of an environment is crucial for building intelligent agents that can perform a wide variety of tasks. Learning such representations in an unsupervised manner without supervision from rewards is an open problem. We in…

2019

Wasserstein Dependency Measure for Representation Learning

NeurIPS 2019poster

Mutual information maximization has emerged as a powerful learning objective for unsupervised representation learning obtaining state-of-the-art performance in applications such as object recognition, speech recognition, and reinforcement learning. However, such approaches are fundamentally limited…

Cited by 141SourcePDFScholar
2018

Mutual Information Neural Estimation

ICML 2018oral

We argue that the estimation of mutual information between high dimensional continuous random variables can be achieved by gradient descent over neural networks. We present a Mutual Information Neural Estimator (MINE) that is linearly scalable in dimensionality as well as in sample size, trainable t…

Cited by 1758SourcePDFScholar
2016

Deep Speech 2 : End-to-End Speech Recognition in English and Mandarin

ICML 2016poster

We show that an end-to-end deep learning approach can be used to recognize either English or Mandarin Chinese speech–two vastly different languages. Because it replaces entire pipelines of hand-engineered components with neural networks, end-to-end learning allows us to handle a diverse variety of s…