← Search

Michael Janner

11 accepted papers

2024

H-GAP: Humanoid Control with a Generalist Planner

ICLR 2024spotlight

Humanoid control is an important research challenge offering avenues for integration into human-centric infrastructures and enabling physics-driven humanoid animations. The daunting challenges in this field stem from the difficulty of optimizing in high-dimensional action spaces and the instability…

Cited by 9SourcePDFScholar
2024

Training Diffusion Models with Reinforcement Learning

ICLR 2024poster

Diffusion models are a class of flexible generative models trained with an approximation to the log-likelihood objective. However, most use cases of diffusion models are not concerned with likelihoods, but instead with downstream objectives such as human-perceived image quality or drug effectiveness…

2023

Efficient Planning in a Compact Latent Action Space

ICLR 2023poster

Planning-based reinforcement learning has shown strong performance in tasks in discrete and low-dimensional continuous action spaces. However, planning usually brings significant computational overhead for decision making, so scaling such methods to high-dimensional action spaces remains challenging…

2022

Lyapunov Density Models: Constraining Distribution Shift in Learning-Based Control

ICML 2022spotlight

Learned models and policies can generalize effectively when evaluated within the distribution of the training data, but can produce unpredictable and erroneous outputs on out-of-distribution inputs. In order to avoid distribution shift when deploying learning-based control algorithms, we seek a mech…

2022

Planning with Diffusion for Flexible Behavior Synthesis

ICML 2022oral

Model-based reinforcement learning methods often use learning only for the purpose of recovering an approximate dynamics model, offloading the rest of the decision-making work to classical trajectory optimizers. While conceptually simple, this combination has a number of empirical shortcomings, sugg…

2021

Offline Reinforcement Learning as One Big Sequence Modeling Problem

NeurIPS 2021spotlight

Reinforcement learning (RL) is typically viewed as the problem of estimating single-step policies (for model-free RL) or single-step models (for model-based RL), leveraging the Markov property to factorize the problem in time. However, we can also view RL as a sequence modeling problem: predict a se…

2020

Gamma-Models: Generative Temporal Difference Learning for Infinite-Horizon Prediction

NeurIPS 2020poster

We introduce the gamma-model, a predictive model of environment dynamics with an infinite, probabilistic horizon. Replacing standard single-step models with gamma-models leads to generalizations of the procedures that form the foundation of model-based control, including the model rollout and model-…

2019

Entity Abstraction in Visual Model-Based Reinforcement Learning

CoRL 2019

We present OP3, a framework for model-based reinforcement learning that acquires object representations from raw visual observations without supervision and uses them to predict and plan. To ground these abstract representations of entities to actual objects in the world, we formulate an interactive

2019

Reasoning About Physical Interactions with Object-Oriented Prediction and Planning

ICLR 2019poster

Object-based factorizations provide a useful level of abstraction for interacting with the world. Building explicit object representations, however, often requires supervisory signals that are difficult to obtain in practice. We present a paradigm for learning object-centric representations for phys…

Cited by 145SourcePDFScholar
2019

When to Trust Your Model: Model-Based Policy Optimization

NeurIPS 2019poster

Designing effective model-based reinforcement learning algorithms is difficult because the ease of data generation must be weighed against the bias of model-generated data. In this paper, we study the role of model usage in policy optimization both theoretically and empirically. We first formulate a…

2017

Self-Supervised Intrinsic Image Decomposition

NeurIPS 2017poster

Intrinsic decomposition from a single image is a highly challenging task, due to its inherent ambiguity and the scarcity of training data. In contrast to traditional fully supervised learning approaches, in this paper we propose learning intrinsic image decomposition by explaining the input image. O…

Cited by 143SourcePDFScholar