← Search

Chongyi Zheng

12 accepted papers

2026

Occupancy Reward Shaping: Improving Credit Assignment for Offline Goal-Conditioned Reinforcement Learning

ICLR 2026poster

While offline goal-conditioned reinforcement learning (GCRL) provides a simple recipe to train generalist policies from large unlabeled datasets, Offline GCRL agents trained with sparse rewards typically struggle on long-horizon tasks. Manually designing task-specific reward functions undermines the…

Cited by 0SourcecodeScholar
2025

Can a MISL Fly? Analysis and Ingredients for Mutual Information Skill Learning

ICLR 2025oral

Self-supervised learning has the potential of lifting several of the key challenges in reinforcement learning today, such as exploration, representation learning, and reward design. Recent work (METRA) has effectively argued that moving away from mutual information and instead optimizing a certain W…

2024

Generalized Animal Imitator: Agile Locomotion with Versatile Motion Prior

CoRL 2024poster

The agility of animals, particularly in complex activities such as running, turning, jumping, and backflipping, stands as an exemplar for robotic system design. Transferring this suite of behaviors to legged robotic systems introduces essential inquiries: How can a robot be trained to learn multiple…

Cited by 19SourceScholar
2024

Learning Temporal Distances: Contrastive Successor Features Can Provide a Metric Structure for Decision-Making

ICML 2024poster

Temporal distances lie at the heart of many algorithms for planning, control, and reinforcement learning that involve reaching goals, allowing one to estimate the transit time between two states. However, prior attempts to define such temporal distances in stochastic settings have been stymied by an…

2024

Stabilizing Contrastive RL: Techniques for Robotic Goal Reaching from Offline Data

ICLR 2024spotlight

Robotic systems that rely primarily on self-supervised learning have the potential to decrease the amount of human annotation and engineering effort required to learn control strategies. In the same way that prior robotic systems have leveraged self-supervised techniques from computer vision (CV) an…

2023

BridgeData V2: A Dataset for Robot Learning at Scale

CoRL 2023poster

We introduce BridgeData V2, a large and diverse dataset of robotic manipulation behaviors designed to facilitate research in scalable robot learning. BridgeData V2 contains 53,896 trajectories collected across 24 environments on a publicly available low-cost robot. Unlike many existing robotic manip…

Cited by 168SourcecodeScholar
2021

Learning Domain Invariant Representations in Goal-conditioned Block MDPs

NeurIPS 2021poster

Deep Reinforcement Learning (RL) is successful in solving many complex Markov Decision Processes (MDPs) problems. However, agents often face unanticipated environmental changes after deployment in the real world. These changes are often spurious and unrelated to the underlying problem, such as backg…

2020

Learning Nearly Decomposable Value Functions Via Communication Minimization

ICLR 2020poster

Reinforcement learning encounters major challenges in multi-agent settings, such as scalability and non-stationarity. Recently, value function factorization learning emerges as a promising way to address these challenges in collaborative multi-agent systems. However, existing methods have been focus…

Cited by 178SourcecodeScholar