← Search

Chuning Zhu

7 accepted papers

2025

Unified World Models: Coupling Video and Action Diffusion for Pretraining on Large Robotic Datasets

RSS 2025poster

Imitation learning has emerged as a promising approach towards building generalist robots. However, the reliance on high-quality expert demonstrations poses a challenge in scaling imitation learning for large-scale robot foundation models. On the other hand, large amounts of video data depicting a w…

Cited by 2PDFScholar
2024

ASID: Active Exploration for System Identification in Robotic Manipulation

ICLR 2024oral

Model-free control strategies such as reinforcement learning have shown the ability to learn control strategies without requiring an accurate model or simulator of the world. While this is appealing due to the lack of modeling requirements, such methods can be sample inefficient, making them impract…

Cited by 14SourcePDFScholar
2024

Distributional Successor Features Enable Zero-Shot Policy Optimization

NeurIPS 2024poster

Intelligent agents must be generalists, capable of quickly adapting to various tasks. In reinforcement learning (RL), model-based RL learns a dynamics model of the world, in principle enabling transfer to arbitrary reward functions through planning. However, autoregressive model rollouts suffer from…

2024

Free from Bellman Completeness: Trajectory Stitching via Model-based Return-conditioned Supervised Learning

ICLR 2024poster

Off-policy dynamic programming (DP) techniques such as $Q$-learning have proven to be important in sequential decision-making problems. In the presence of function approximation, however, these techniques often diverge due to the absence of Bellman completeness in the function classes considered, a…

2023

RePo: Resilient Model-Based Reinforcement Learning by Regularizing Posterior Predictability

NeurIPS 2023spotlight

Visual model-based RL methods typically encode image observations into low-dimensional representations in a manner that does not eliminate redundant information. This leaves them susceptible to spurious variations -- changes in task-irrelevant components such as background distractors or lighting co…

Cited by 13SourcePDFScholar
2023

Self-Supervised Reinforcement Learning that Transfers using Random Features

NeurIPS 2023poster

Model-free reinforcement learning algorithms have exhibited great potential in solving single-task sequential decision-making problems with high-dimensional observations and long horizons, but are known to be hard to generalize across tasks. Model-based RL, on the other hand, learns task-agnostic mo…

Cited by 11SourcePDFScholar
2021

Model-Based Reinforcement Learning via Latent-Space Collocation

ICML 2021spotlight

The ability to plan into the future while utilizing only raw high-dimensional observations, such as images, can provide autonomous agents with broad and general capabilities. However, realistic tasks require performing temporally extended reasoning, and cannot be solved with only myopic, short-sight…