← Search

Huazhe Xu

94 accepted papers

2026

AffordGen: Generating Diverse Demonstrations for Generalizable Object Manipulation with Affordance Correspondence

CVPR 2026

Despite the recent success of modern imitation learning methods in robot manipulation, their performance is often constrained by geometric variations due to limited data diversity. Leveraging powerful 3D generative models and vision foundation models (VFMs), the proposed AffordGen framework overcome

Cited by 0SourceScholar
2026

Automated Synthesis of Facial Mechanisms for Conversational Animatronic Robots

RSS 2026poster

Animatronic faces are a central component of socially interactive robots, enabling rich nonverbal communication through facial articulation. However, state-of-the-art animatronic faces are typically tailored systems: each new facial geometry requires extensive manual mechanical redesign, making larg…

Cited by 0SourceScholar
2026

DexImit: Learning Bimanual Dexterous Manipulation from Monocular Human Videos

RSS 2026poster

Data scarcity fundamentally limits the generalization of bimanual dexterous manipulation, as real-world data collection for dexterous hands is expensive and labor-intensive. Human manipulation videos, as a direct carrier of manipulation knowledge, offer significant potential for scaling up robot lea…

Cited by 0SourceScholar
2026

Entropy Regularizing Activation: Boosting Continuous Control, Large Language Models, and Image Classification with Activation as Entropy Constraints

ICLR 2026poster

We propose ERA, a new paradigm for entropy-constrained policy via output activation. It guarantees minimum sampling entropy by transforming the outputs of the last layer. Our approach demonstrates broad effectiveness across different domains: 1) for large language models~(LLMs), boosting the AIME 20…

Cited by 0SourcecodeScholar
2026

Failure-Aware RL: Reliable Offline-To-Online Reinforcement Learning with Self-Recovery for Real-World Manipulation

ICRA 2026poster

Post-training algorithms based on deep reinforcement learning can push the limits of robotic models for specific objectives, such as generalizability, accuracy, and robustness. However, Intervention-requiring Failures (IR Failures) (e.g., a robot spilling water or breaking fragile glass) during real…

2026

H$^3$DP: Triply‑Hierarchical Diffusion Policy for Visuomotor Learning

ICLR 2026poster

Visuomotor policy learning has witnessed substantial progress in robotic manipulation, with recent approaches predominantly relying on generative models to model the action distribution. However, these methods often overlook the critical coupling between visual perception and action prediction. In t…

Cited by 0SourcecodeScholar
2026

MoE-DP: An MoE-Enhanced Diffusion Policy for Robust Long-Horizon Robotic Manipulation with Skill Decomposition and Failure Recovery

ICRA 2026poster

Diffusion policies have emerged as a powerful framework for robotic visuomotor control, yet they often lack the robustness to recover from subtask failures in long-horizon, multi-stage tasks and their learned representations of observations are often difficult to interpret. In this work, we propose …

2026

One-Policy-Fits-All: Geometry-Aware Action Latents for Cross-Embodiment Manipulation

ICRA 2026poster

Cross-embodiment manipulation is crucial for enhancing the scalability of robot manipulation and reducing the high cost of data collection. However, the significant differences between embodiments, such as variations in action spaces and structural disparities, pose challenges for joint training acr…

2026

UniDex: A Robot Foundation Suite for Universal Dexterous Hand Control from Egocentric Human Videos

CVPR 2026

Dexterous manipulation remains challenging due to the cost of collecting real-robot teleoperation data, the heterogeneity of hand embodiments, and the high dimensionality of control. We present UniDex, a robot foundation suite that couples a large-scale robot-centric dataset with a unified vision-la

Cited by 0SourcecodeScholar
2025

4D Visual Pre-training for Robot Learning

ICCV 2025poster

General visual representations learned from web-scale datasets for robotics have achieved great success in recent years, enabling data-efficient robot learning on manipulation tasks; yet these pre-trained representations are mostly on 2D images, neglecting the inherent 3D nature of the world. Howeve…

2025

A Forget-and-Grow Strategy for Deep Reinforcement Learning Scaling in Continuous Control

ICML 2025poster

Deep reinforcement learning for continuous control has recently achieved impressive progress. However, existing methods often suffer from primacy bias—a tendency to overfit early experiences stored in the replay buffer—which limits an RL agent’s sample efficiency and generalizability. A common exist…

Cited by 0SourcePDFScholar
2025

Catch It! Learning to Catch in Flight with Mobile Dexterous Hands

ICRA 2025

Catching objects in flight (i.e., thrown objects) is a common daily skill for humans, yet it presents a significant challenge for robots. This task requires a robot with agile and accurate motion, a large spatial workspace, and the ability to interact with diverse objects. In this paper, we build a

Cited by 27SourcecodeScholar
2025

DOGlove: Dexterous Manipulation with a Low-Cost Open-Source Haptic Force Feedback Glove

RSS 2025poster

Dexterous hand teleoperation plays a pivotal role in enabling robots to achieve human-level manipulation dexterity. However, current teleoperation systems often rely on expensive equipment and lack multi-modal sensory feedback, restricting human operators’ ability to perceive object properties and p…

Cited by 4PDFScholar
2025

DemoGen: Synthetic Demonstration Generation for Data-Efficient Visuomotor Policy Learning

RSS 2025poster

Visuomotor policies have shown great promise in robotic manipulation but often require substantial amounts of human-collected data for effective performance. A key reason underlying the data demands is their limited spatial generalization capability, which necessitates extensive data collection acro…

Cited by 8PDFScholar
2025

DemoSpeedup: Accelerating Visuomotor Policies via Entropy-Guided Demonstration Acceleration

CoRL 2025oral

Imitation learning has shown great promise in robotic manipulation, but the policy’s execution is often unsatisfactorily slow due to commonly tardy demonstrations collected by human operators. In this work, we present DemoSpeedup, a self-supervised method to accelerate visuomotor policy execution vi…

Cited by 0SourceScholar
2025

DenseMatcher: Learning 3D Semantic Correspondence for Category-Level Manipulation from a Single Demo

ICLR 2025spotlight

Dense 3D correspondence can enhance robotic manipulation by enabling the generalization of spatial, functional, and dynamic information from one object to an unseen counterpart. Compared to shape correspondence, semantic correspondence is more effective in generalizing across different object catego…

2025

FACET: Force-Adaptive Control via Impedance Reference Tracking for Legged Robots

CoRL 2025oral

Reinforcement learning (RL) has made significant strides in legged robot control, enabling locomotion across diverse terrains and complex loco-manipulation capabilities. However, the commonly used position or velocity tracking-based objectives are agnostic to forces experienced by the robot, leading…

Cited by 0SourceScholar
2025

Fine-Tuning Hard-to-Simulate Objectives for Quadruped Locomotion: A Case Study on Total Power Saving

ICRA 2025

Legged locomotion is not just about mobility; it also encompasses crucial objectives such as energy efficiency, safety, and user experience, which are vital for real-world applications. However, key factors such as battery power consumption and stepping noise are often inaccurately modeled or missin

Cited by 2SourcecodeScholar
2025

Looking Backward: Retrospective Backward Synthesis for Goal-Conditioned GFlowNets

ICLR 2025poster

Generative Flow Networks (GFlowNets), a new family of probabilistic samplers, have demonstrated remarkable capabilities to generate diverse sets of high-reward candidates, in contrast to standard return maximization approaches (e.g., reinforcement learning) which often converge to a single optimal s…

2025

MENTOR: Mixture-of-Experts Network with Task-Oriented Perturbation for Visual Reinforcement Learning

ICML 2025poster

Visual deep reinforcement learning (RL) enables robots to acquire skills from visual input for unstructured tasks. However, current algorithms suffer from low sample efficiency, limiting their practical applicability. In this work, we present MENTOR, a method that improves both the *architecture* an…

2025

Morpheus: A Neural-driven Animatronic Face with Hybrid Actuation and Diverse Emotion Control

RSS 2025poster

Previous animatronic faces struggle to effectively express emotions due to both hardware and software limitations. On the hardware side, earlier approaches either used rigid-driven mechanisms, which provide precise control but are difficult to design within constrained spaces, or tendon-driven mecha…

Cited by 0PDFScholar
2025

Reactive Diffusion Policy: Slow-Fast Visual-Tactile Policy Learning for Contact-Rich Manipulation

RSS 2025poster

Humans can accomplish complex contact-rich tasks using vision and touch, with highly reactive capabilities such as quick adjustments to environmental changes and adaptive control of contact forces; however, this remains challenging for robots. Existing visual imitation learning (IL) approaches rely…

Cited by 8PDFcodeScholar
2025

RoboDuet: Learning a Cooperative Policy for Whole-Body Legged Loco-Manipulation

RA-L 2025

Fully leveraging the loco-manipulation capabilities of a quadruped robot equipped with a robotic arm is non-trivial, as it requires controlling all degrees of freedom (DoFs) of the quadruped robot to achieve effective whole-body coordination. In this letter, we propose a novel framework RoboDuet, wh

Cited by 13SourceScholar
2025

Robots Pre-train Robots: Manipulation-Centric Robotic Representation from Large-Scale Robot Datasets

ICLR 2025poster

The pre-training of visual representations has enhanced the efficiency of robot learning. Due to the lack of large-scale in-domain robotic datasets, prior works utilize in-the-wild human videos to pre-train robotic visual representation. Despite their promising results, representations from human vi…

2025

Stem-OB: Generalizable Visual Imitation Learning with Stem-Like Convergent Observation through Diffusion Inversion

ICLR 2025spotlight

Visual imitation learning methods demonstrate strong performance, yet they lack generalization when faced with visual input perturbations like variations in lighting and textures. This limitation hampers their practical application in real-world settings. To address this, we propose ***Stem-OB*** th…

2025

Two by Two: Learning Multi-Task Pairwise Objects Assembly for Generalizable Robot Manipulation

CVPR 2025poster

3D assembly tasks, such as furniture assembly and component fitting, play a crucial role in daily life and represent essential capabilities for future home robots. Existing benchmarks and datasets predominantly focus on assembling geometric fragments or factory parts, which fall short in addressing…

2025

World Models with Hints of Large Language Models for Goal Achieving

NAACL 2025long

Reinforcement learning struggles in the face of long-horizon tasks and sparse goals due to the difficulty in manual reward specification. While existing methods address this by adding intrinsic rewards, they may fail to provide meaningful guidance in long-horizon decision-making tasks with large sta…

Cited by 2SourcePDFScholar
2024

3D Diffusion Policy: Generalizable Visuomotor Policy Learning via Simple 3D Representations

RSS 2024poster

Imitation learning provides an efficient way to teach robots dexterous skills; however, learning complex skills robustly and generalizablely usually consumes large amounts of human demonstrations. To tackle this challenging problem, we present 3D Diffusion Policy (DP3), a novel visual imitation lear…

2024

9DTact: A Compact Vision-Based Tactile Sensor for Accurate 3D Shape Reconstruction and Generalizable 6D Force Estimation

RA-L 2024

The advancements in vision-based tactile sensors have boosted the aptitude of robots to perform contact-rich manipulation, particularly when precise positioning and contact state of the manipulated objects are crucial for successful execution. In this work, we present <underline xmlns:mml="http://ww

Cited by 80SourceScholar
2024

ACE: Off-Policy Actor-Critic with Causality-Aware Entropy Regularization

ICML 2024oral

The varying significance of distinct primitive behaviors during the policy learning process has been overlooked by prior model-free RL algorithms. Leveraging this insight, we explore the causal relationship between different action dimensions and rewards to evaluate the significance of various primi…

2024

ArrayBot: Reinforcement Learning for Generalizable Distributed Manipulation through Touch

ICRA 2024poster

We present ArrayBot, a distributed manipulation system consisting of a 16 × 16 array of vertically sliding pillars integrated with tactile sensors. Functionally, ArrayBot is designed to simultaneously support, perceive, and manipulate the tabletop objects. Towards generalizable distributed manipulat…

Cited by 14SourceScholar
2024

COPlanner: Plan to Roll Out Conservatively but to Explore Optimistically for Model-Based RL

ICLR 2024poster

Dyna-style model-based reinforcement learning contains two phases: model rollouts to generate sample for policy learning and real environment exploration using current policy for dynamics model learning. However, due to the complex real-world environment, it is inevitable to learn an imperfect dynam…

Cited by 10SourcePDFScholar
2024

DeformNet: Latent Space Modeling and Dynamics Prediction for Deformable Object Manipulation

ICRA 2024poster

Manipulating deformable objects is a ubiquitous task in household environments, demanding adequate representation and accurate dynamics prediction due to the objects’ infinite degrees of freedom. This work proposes DeformNet, which utilizes latent space modeling with a learned 3D representation mode…

Cited by 6SourceScholar
2024

DittoGym: Learning to Control Soft Shape-Shifting Robots

ICLR 2024poster

Robot co-design, where the morphology of a robot is optimized jointly with a learned policy to solve a specific task, is an emerging area of research. It holds particular promise for soft robots, which are amenable to novel manufacturing techniques that can realize learned morphologies and actuators…

2024

DrM: Mastering Visual Reinforcement Learning through Dormant Ratio Minimization

ICLR 2024spotlight

Visual reinforcement learning (RL) has shown promise in continuous control tasks. Despite its progress, current algorithms are still unsatisfactory in virtually every aspect of the performance such as sample efficiency, asymptotic performance, and their robustness to the choice of random seeds. In t…

2024

GenSim2: Scaling Robot Data Generation with Multi-modal and Reasoning LLMs

CoRL 2024poster

Robotic simulation today remains challenging to scale up due to the human efforts required to create diverse simulation tasks and scenes. Simulation-trained policies also face scalability issues as many sim-to-real methods focus on a single task. To address these challenges, this work proposes GenSi…

Cited by 11SourceScholar
2024

GenSim: Generating Robotic Simulation Tasks via Large Language Models

ICLR 2024spotlight

Collecting large amounts of real-world interaction data to train general robotic policies is often prohibitively expensive, thus motivating the use of simulation data. However, existing methods for data generation have generally focused on scene-level diversity (e.g., object instances and poses) rat…

2024

Key-Grid: Unsupervised 3D Keypoints Detection using Grid Heatmap Features

NeurIPS 2024poster

Detecting 3D keypoints with semantic consistency is widely used in many scenarios such as pose estimation, shape registration and robotics. Currently, most unsupervised 3D keypoint detection methods focus on the rigid-body objects. However, when faced with deformable objects, the keypoints they iden…

Cited by 1SourcePDFScholar
2024

Learning Visual Quadrupedal Loco-Manipulation from Demonstrations

IROS 2024poster

Quadruped robots are progressively being integrated into human environments. Despite the growing locomotion capabilities of quadrupedal robots, their interaction with objects in realistic scenes is still limited. While additional robotic arms on quadrupedal robots enable manipulating objects, they a…

Cited by 17SourceScholar
2024

Learning to Manipulate Anywhere: A Visual Generalizable Framework For Reinforcement Learning

CoRL 2024poster

Can we endow visuomotor robots with generalization capabilities to operate in diverse open-world scenarios? In this paper, we propose Maniwhere, a generalizable framework tailored for visual reinforcement learning, enabling the trained robot policies to generalize across a combination of multiple vi…

Cited by 22SourcecodeScholar
2024

Make-An-Agent: A Generalizable Policy Network Generator with Behavior-Prompted Diffusion

NeurIPS 2024poster

Can we generate a control policy for an agent using just one demonstration of desired behaviors as a prompt, as effortlessly as creating an image from a textual description? In this paper, we present **Make-An-Agent**, a novel policy parameter generator that leverages the power of conditional diffus…

2024

OMPO: A Unified Framework for RL under Policy and Dynamics Shifts

ICML 2024oral

Training reinforcement learning policies using environment interaction data collected from varying policies or dynamics presents a fundamental challenge. Existing works often overlook the distribution discrepancies induced by policy or dynamics shifts, or rely on specialized algorithms with task pri…

2024

Offline-Boosted Actor-Critic: Adaptively Blending Optimal Historical Behaviors in Deep Off-Policy RL

ICML 2024poster

Off-policy reinforcement learning (RL) has achieved notable success in tackling many complex real-world tasks, by leveraging previously collected data for policy learning. However, most existing off-policy RL algorithms fail to maximally exploit the information in the replay buffer, limiting sample…

2024

Premier-TACO is a Few-Shot Policy Learner: Pretraining Multitask Representation via Temporal Action-Driven Contrastive Loss

ICML 2024poster

We present Premier-TACO, a multitask feature representation learning approach designed to improve few-shot policy learning efficiency in sequential decision-making tasks. Premier-TACO leverages a subset of multitask offline datasets for pretraining a general feature representation, which captures cr…

2024

Rethinking Transformers in Solving POMDPs

ICML 2024poster

Sequential decision-making algorithms such as reinforcement learning (RL) in real-world scenarios inevitably face environments with partial observability. This paper scrutinizes the effectiveness of a popular architecture, namely Transformers, in Partially Observable Markov Decision Processes (POMDP…

2024

RiEMann: Near Real-Time SE(3)-Equivariant Robot Manipulation without Point Cloud Segmentation

CoRL 2024poster

We present RiEMann, an end-to-end near Real-time SE(3)-Equivariant Robot Manipulation imitation learning framework from scene point cloud input. Compared to previous methods that rely on descriptor field matching, RiEMann directly predicts the target actions for manipulation without any object segme…

Cited by 16SourceScholar
2024

Seizing Serendipity: Exploiting the Value of Past Success in Off-Policy Actor-Critic

ICML 2024poster

Learning high-quality $Q$-value functions plays a key role in the success of many modern off-policy deep reinforcement learning (RL) algorithms. Previous works primarily focus on addressing the value overestimation issue, an outcome of adopting function approximators and off-policy learning. Deviati…

2024

Uni-O4: Unifying Online and Offline Deep Reinforcement Learning with Multi-Step On-Policy Optimization

ICLR 2024poster

Combining offline and online reinforcement learning (RL) is crucial for efficient and safe learning. However, previous approaches treat offline and online learning as separate procedures, resulting in redundant designs and limited performance. We ask: *Can we achieve straightforward yet effective of…

2024

Unleashing the Power of Pre-trained Language Models for Offline Reinforcement Learning

ICLR 2024poster

Offline reinforcement learning (RL) aims to find a near-optimal policy using pre-collected datasets. Given recent advances in Large Language Models (LLMs) and their few-shot learning prowess, this paper introduces $\textbf{La}$nguage Models for $\textbf{Mo}$tion Control ($\textbf{LaMo}$), a general…

2023

$\texttt{TACO}$: Temporal Latent Action-Driven Contrastive Loss for Visual Reinforcement Learning

NeurIPS 2023poster

Despite recent progress in reinforcement learning (RL) from raw pixel data, sample inefficiency continues to present a substantial obstacle. Prior works have attempted to address this challenge by creating self-supervised auxiliary tasks, aiming to enrich the agent's learned representations with co…

2023

CEIL: Generalized Contextual Imitation Learning

NeurIPS 2023poster

In this paper, we present ContExtual Imitation Learning (CEIL), a general and broadly applicable algorithm for imitation learning (IL). Inspired by the formulation of hindsight information matching, we derive CEIL by explicitly learning a hindsight embedding function together with a contextual polic…

Cited by 23SourcePDFScholar
2023

Can Pre-Trained Text-to-Image Models Generate Visual Goals for Reinforcement Learning?

NeurIPS 2023poster

Pre-trained text-to-image generative models can produce diverse, semantically rich, and realistic images from natural language descriptions. Compared with language, images usually convey information with more details and less ambiguity. In this study, we propose Learning from the Void (LfVoid), a me…

2023

DTact: A Vision-Based Tactile Sensor that Measures High-Resolution 3D Geometry Directly from Darkness

ICRA 2023poster

Vision-based tactile sensors that can measure 3D geometry of the contacting objects are crucial for robots to perform dexterous manipulation tasks. However, the existing sensors are usually complicated to fabricate and delicate to extend. In this work, we novelly take advantage of the reflection pro…

Cited by 33SourceScholar
2023

Efficient Bimanual Handover and Rearrangement via Symmetry-Aware Actor-Critic Learning

ICRA 2023poster

Bimanual manipulation is important for building intelligent robots that unlock richer skills than single arms. We consider a multi-object bimanual rearrangement task, where a reinforcement learning (RL) agent aims to jointly control two arms to rearrange these objects as fast as possible. Solving th…

Cited by 15SourceScholar
2023

H-InDex: Visual Reinforcement Learning with Hand-Informed Representations for Dexterous Manipulation

NeurIPS 2023poster

Human hands possess remarkable dexterity and have long served as a source of inspiration for robotic manipulation. In this work, we propose a human $\textbf{H}$and-$\textbf{In}$formed visual representation learning framework to solve difficult $\textbf{Dex}$terous manipulation tasks ($\textbf{H-InDe…

Cited by 21SourcePDFScholar
2023

Is Model Ensemble Necessary? Model-based RL via a Single Model with Lipschitz Regularized Value Function

ICLR 2023poster

Probabilistic dynamics model ensemble is widely used in existing model-based reinforcement learning methods as it outperforms a single dynamics model in both asymptotic performance and sample efficiency. In this paper, we provide both practical and theoretical insights on the empirical success of th…

Cited by 16SourcePDFScholar
2023

On Pre-Training for Visuo-Motor Control: Revisiting a Learning-from-Scratch Baseline

ICML 2023poster

In this paper, we examine the effectiveness of pre-training for visuo-motor control tasks. We revisit a simple Learning-from-Scratch (LfS) baseline that incorporates data augmentation and a shallow ConvNet, and find that this baseline is surprisingly competitive with recent approaches (PVR, MVP, R3M…

2023

RL-ViGen: A Reinforcement Learning Benchmark for Visual Generalization

NeurIPS 2023poster

Visual Reinforcement Learning (Visual RL), coupled with high-dimensional observations, has consistently confronted the long-standing challenge of out-of-distribution generalization. Despite the focus on algorithms aimed at resolving visual generalization problems, we argue that the devil is in the e…

2023

RoboCook: Long-Horizon Elasto-Plastic Object Manipulation with Diverse Tools

CoRL 2023oral

Humans excel in complex long-horizon soft body manipulation tasks via flexible tool use: bread baking requires a knife to slice the dough and a rolling pin to flatten it. Often regarded as a hallmark of human cognition, tool use in autonomous robots remains limited due to challenges in understanding…

Cited by 62SourceScholar
2023

Scaling up and Stabilizing Differentiable Planning with Implicit Differentiation

ICLR 2023poster

Differentiable planning promises end-to-end differentiability and adaptivity. However, an issue prevents it from scaling up to larger-scale problems: they need to differentiate through forward iteration layers to compute gradients, which couples forward computation and backpropagation and needs to b…

Cited by 7SourcePDFScholar
2023

USEEK: Unsupervised SE(3)-Equivariant 3D Keypoints for Generalizable Manipulation

ICRA 2023poster

Can a robot manipulate intra-category unseen objects in arbitrary poses with the help of a mere demonstration of grasping pose on a single object instance? In this paper, we try to address this intriguing challenge by using USEEK, an unsupervised SE(3)-equivariant keypoints method that enjoys alignm…

Cited by 29SourceScholar
2022

Don’t Touch What Matters: Task-Aware Lipschitz Data Augmentation for Visual Reinforcement Learning

IJCAI 2022poster

One of the key challenges in visual Reinforcement Learning (RL) is to learn policies that can generalize to unseen environments. Recently, data augmentation techniques aiming at enhancing data diversity have demonstrated proven performance in improving the generalization ability of learned policies.…

2022

E-MAPP: Efficient Multi-Agent Reinforcement Learning with Parallel Program Guidance

NeurIPS 2022accept

A critical challenge in multi-agent reinforcement learning(MARL) is for multiple agents to efficiently accomplish complex, long-horizon tasks. The agents often have difficulties in cooperating on common goals, dividing complex tasks, and planning through several stages to make progress. We propose t…

Cited by 6SourcePDFScholar
2022

Learning Vision-Guided Quadrupedal Locomotion End-to-End with Cross-Modal Transformers

ICLR 2022spotlight

We propose to address quadrupedal locomotion tasks using Reinforcement Learning (RL) with a Transformer-based model that learns to combine proprioceptive information and high-dimensional depth sensor inputs. While learning-based locomotion has made great advances using RL, most methods still rely on…

2022

Phasic Self-Imitative Reduction for Sparse-Reward Goal-Conditioned Reinforcement Learning

ICML 2022spotlight

It has been a recent trend to leverage the power of supervised learning (SL) towards more effective reinforcement learning (RL) methods. We propose a novel phasic solution by alternating online RL and offline SL for tackling sparse-reward goal-conditioned problems. In the online phase, we perform RL…

Cited by 22SourcePDFScholar
2022

Plan Better Amid Conservatism: Offline Multi-Agent Reinforcement Learning with Actor Rectification

ICML 2022spotlight

Conservatism has led to significant progress in offline reinforcement learning (RL) where an agent learns from pre-collected datasets. However, as many real-world scenarios involve interaction among multiple agents, it is important to resolve offline RL in the multi-agent setting. Given the recent s…

2022

Pre-Trained Image Encoder for Generalizable Visual Reinforcement Learning

NeurIPS 2022accept

Learning generalizable policies that can adapt to unseen environments remains challenging in visual Reinforcement Learning (RL). Existing approaches try to acquire a robust representation via diversifying the appearances of in-domain observations for better generalization. Limited by the specific ob…

Cited by 85SourcePDFScholar
2022

See, Hear, and Feel: Smart Sensory Fusion for Robotic Manipulation

CoRL 2022poster

Humans use all of their senses to accomplish different tasks in everyday activities. In contrast, existing work on robotic manipulation mostly relies on one, or occasionally two modalities, such as vision and touch. In this work, we systematically study how visual, auditory, and tactile perception c…

Cited by 65SourceScholar
2022

Towards Learning to Play Piano with Dexterous Hands and Touch

IROS 2022poster

As Liszt once said “(a virtuoso) must call up scent and blossom, and breathe the breath of life”, a virtuoso plays the piano with passion, poetry, and extraordinary technical ability. Hence, piano playing, being a task that is quintessentially human, becomes a hallmark for roboticians and artificial…

Cited by 37SourceScholar
2021

Discovering Diverse Multi-Agent Strategic Behavior via Reward Randomization

ICLR 2021poster

We propose a simple, general and effective technique, Reward Randomization for discovering diverse strategic policies in complex multi-agent games. Combining reward randomization and policy gradient, we derive a new algorithm, Reward-Randomized Policy Gradient (RPG). RPG is able to discover a set of…

Cited by 63SourcePDFScholar
2021

Multi-Person 3D Motion Prediction with Multi-Range Transformers

NeurIPS 2021poster

We propose a novel framework for multi-person 3D motion trajectory prediction. Our key observation is that a human's action and behaviors may highly depend on the other persons around. Thus, instead of predicting each human pose trajectory in isolation, we introduce a Multi-Range Transformers model…

2021

NovelD: A Simple yet Effective Exploration Criterion

NeurIPS 2021poster

Efficient exploration under sparse rewards remains a key challenge in deep reinforcement learning. Previous exploration methods (e.g., RND) have achieved strong results in multiple hard tasks. However, if there are multiple novel areas to explore, these methods often focus quickly on one without suf…

2021

PyTouch: A Machine Learning Library for Touch Processing

ICRA 2021poster

With the increased availability of rich tactile sensors, there is an an equally proportional need for open-source and integrated software capable of efficiently and effectively processing raw touch measurements into high-level signals that can be used for control and decision-making. In this paper,…

Cited by 28SourcecodeScholar
2021

Solving Compositional Reinforcement Learning Problems via Task Reduction

ICLR 2021poster

We propose a novel learning paradigm, Self-Imitation via Reduction (SIR), for solving compositional reinforcement learning problems. SIR is based on two core ideas: task reduction and self-imitation. Task reduction tackles a hard-to-solve task by actively reducing it to an easier task whose solution…

2021

Synthesizing Long-Term 3D Human Motion and Interaction in 3D Scenes

CVPR 2021poster

Synthesizing 3D human motion plays an important role in many graphics applications as well as understanding human activity. While many efforts have been made on generating realistic and natural human motion, most approaches neglect the importance of modeling human-scene interactions and affordances.…

Cited by 145PDFScholar
2021

Zero-shot Policy Learning with Spatial Temporal Reward Decomposition on Contingency-aware Observation

ICRA 2021poster

It is a long-standing challenge to enable an intelligent agent to learn in one environment and generalize to an unseen environment without further data collection and finetuning. In this paper, we consider a zero shot generalization problem setup that complies with biological intelligent agents’ lea…

Cited by 1SourceScholar
2020

Hierarchical Style-based Networks for Motion Synthesis

ECCV 2020poster

Generating diverse and natural behaviors is one of the long-standing goals for creating intelligent characters in the animated world. In this paper, we propose an unsupervised method for generating long-range, diverse and plausible behaviors to achieve a specific goal location. Our proposed method l…

Cited by 35SourcePDFScholar
2020

Learning Self-Correctable Policies and Value Functions from Demonstrations with Negative Sampling

ICLR 2020poster

Imitation learning, followed by reinforcement learning algorithms, is a promising paradigm to solve complex control tasks sample-efficiently. However, learning from demonstrations often suffers from the covariate shift problem, which results in cascading errors of the learned policy. We introduce a…

Cited by 21SourceScholar
2019

Algorithmic Framework for Model-based Deep Reinforcement Learning with Theoretical Guarantees

ICLR 2019poster

Model-based reinforcement learning (RL) is considered to be a promising approach to reduce the sample complexity that hinders model-free RL. However, the theoretical understanding of such methods has been rather limited. This paper introduces a novel algorithmic framework for designing and analyzing…

2019

Disentangling Propagation and Generation for Video Prediction

ICCV 2019poster

A dynamic scene has two types of elements: those that move fluidly and can be predicted from previous frames, and those which are disoccluded (exposed) and cannot be extrapolated. Prior approaches to video prediction typically learn either to warp or to hallucinate future pixels, but not both. In th…

Cited by 118PDFScholar
2017

End-To-End Learning of Driving Models From Large-Scale Video Datasets

CVPR 2017oral

Robust perception-action models should be learned from training data with diverse visual appearances and realistic behaviors, yet current approaches to deep visuomotor policy learning have been generally limited to in-situ models learned from a single vehicle or simulation environment. We advocate l…

Cited by 1126PDFcodeScholar