← Search

Dian Wang

24 accepted papers

2026

Generalizable Hierarchical Skill Learning via Object-Centric Representation

RA-L 2026

We present Generalizable Hierarchical Skill Learning (GSL), a novel framework for hierarchical policy learning that improves policy generalization and sample efficiency in robot manipulation. One core idea of GSL is to use object-centric skills as an interface that bridges the high-level vision-lang

Cited by 3SourceScholar
2026

HoMMI: Learning Whole-Body Mobile Manipulation from Human Demonstrations

RSS 2026poster

We present Whole-Body Mobile Manipulation Interface (HoMMI), a data collection and policy learning framework that learns whole-body mobile manipulation directly from robot-free human demonstrations. We augment UMI interfaces with egocentric sensing to capture the global context required for mobile m…

Cited by 0SourceScholar
2026

RAVEN: End-to-end Equivariant Robot Learning with RGB Cameras

ICLR 2026poster

Recent work has shown that equivariant policy networks can achieve strong performance on robot manipulation tasks with limited human demonstrations. However, existing equivariant methods typically require structured inputs, such as 3D point clouds or top-down camera views, which prevents their use…

Cited by 0SourceScholar
2025

3D Equivariant Visuomotor Policy Learning via Spherical Projection

NeurIPS 2025spotlight

Equivariant models have recently been shown to improve the data efficiency of diffusion policy by a significant margin. However, prior work that explored this direction focused primarily on point cloud inputs generated by multiple cameras fixed in the workspace. This type of point cloud input is not…

Cited by 0SourceScholar
2025

A Practical Guide for Incorporating Symmetry in Diffusion Policy

NeurIPS 2025poster

Recently, equivariant neural networks for policy learning have shown promising improvements in sample efficiency and generalization, however, their wide adoption faces substantial barriers due to implementation complexity. Equivariant architectures typically require specialized mathematical formulat…

Cited by 0SourceScholar
2025

Hierarchical Equivariant Policy via Frame Transfer

ICML 2025poster

Recent advances in hierarchical policy learning highlight the advantages of decomposing systems into high-level and low-level agents, enabling efficient long-horizon reasoning and precise fine-grained control. However, the interface between these hierarchy levels remains underexplored, and existing…

Cited by 2SourcePDFScholar
2025

Learning Efficient and Robust Language-Conditioned Manipulation Using Textual-Visual Relevancy and Equivariant Language Mapping

RA-L 2025

Controlling robots through natural language is pivotal for enhancing human-robot collaboration and synthesizing complex robot behaviors. Recent works that are trained on large robot datasets show impressive generalization abilities. However, such pretrained methods are (1) often fragile to unseen sc

Cited by 7SourcecodeScholar
2025

Match Policy: A Simple Pipeline from Point Cloud Registration to Manipulation Policies

ICRA 2025

Many manipulation tasks require the robot to rearrange objects relative to one another. Such tasks can be described as a sequence of relative poses between parts of a set of rigid bodies. In this work, we propose Match Policy, a simple but novel pipeline for solving high-precision pick and place tas

Cited by 5SourcecodeScholar
2024

Equivariant Diffusion Policy

CoRL 2024poster

Recent work has shown diffusion models are an effective approach to learning the multimodal distributions arising from demonstration data in behavior cloning. However, a drawback of this approach is the need to learn a denoising function, which is significantly more complex than learning an explicit…

Cited by 26SourcecodeScholar
2024

Fourier Transporter: Bi-Equivariant Robotic Manipulation in 3D

ICLR 2024poster

Many complex robotic manipulation tasks can be decomposed as a sequence of pick and place actions. Training a robotic agent to learn this sequence over many different starting conditions typically requires many iterations or demonstrations, especially in 3D environments. In this work, we propose Fou…

Cited by 24SourcePDFScholar
2024

IMAGINATION POLICY: Using Generative Point Cloud Models for Learning Manipulation Policies

CoRL 2024poster

Humans can imagine goal states during planning and perform actions to match those goals. In this work, we propose IMAGINATION POLICY, a novel multi-task key-frame policy network for solving high-precision pick and place tasks. Instead of learning actions directly, IMAGINATION POLICY generates point…

Cited by 7SourceScholar
2024

OrbitGrasp: SE(3)-Equivariant Grasp Learning

CoRL 2024poster

While grasp detection is an important part of any robotic manipulation pipeline, reliable and accurate grasp detection in $\\mathrm{SE}(3)$ remains a research challenge. Many robotics applications in unstructured environments such as the home or warehouse would benefit a lot from better grasp perfor…

Cited by 13SourcecodeScholar
2023

A General Theory of Correct, Incorrect, and Extrinsic Equivariance

NeurIPS 2023poster

Although equivariant machine learning has proven effective at many tasks, success depends heavily on the assumption that the ground truth function is symmetric over the entire domain matching the symmetry in an equivariant neural network. A missing piece in the equivariant learning literature is the…

Cited by 10SourcePDFScholar
2023

Edge Grasp Network: A Graph-Based SE(3)-invariant Approach to Grasp Detection

ICRA 2023poster

Given point cloud input, the problem of 6-DoF grasp pose detection is to identify a set of hand poses in SE(3) from which an object can be successfully grasped. This important problem has many practical applications. Here we propose a novel method and neural network model that enables better grasp s…

Cited by 39SourcecodeScholar
2023

Equivariant Reinforcement Learning under Partial Observability

CoRL 2023poster

Incorporating inductive biases is a promising approach for tackling challenging robot learning domains with sample-efficient solutions. This paper identifies partially observable domains where symmetries can be a useful inductive bias for efficient learning. Specifically, by encoding the equivarianc…

Cited by 15SourceScholar
2023

SEIL: Simulation-augmented Equivariant Imitation Learning

ICRA 2023poster

In robotic manipulation, acquiring samples is extremely expensive because it often requires interacting with the real world. Traditional image-level data augmentation has shown the potential to improve sample efficiency in various machine learning tasks. However, image-level data augmentation is ins…

Cited by 20SourceScholar
2023

The Surprising Effectiveness of Equivariant Models in Domains with Latent Symmetry

ICLR 2023top-25%

Extensive work has demonstrated that equivariant neural networks can significantly improve sample efficiency and generalization by enforcing an inductive bias in the network architecture. These applications typically assume that the domain symmetry is fully described by explicit transformations of t…

Cited by 34SourcePDFScholar
2022

Leveraging Fully Observable Policies for Learning under Partial Observability

CoRL 2022poster

Reinforcement learning in partially observable domains is challenging due to the lack of observable state information. Thankfully, learning offline in a simulator with such state information is often possible. In particular, we propose a method for partially observable reinforcement learning that us…

Cited by 31SourcecodeScholar
2022

Sample Efficient Grasp Learning Using Equivariant Models

RSS 2022poster

In planar grasp detection, the goal is to learn a function from an image of a scene onto a set of feasible grasp poses in SE(2). In this paper, we recognize that the optimal grasp function is SE(2)-equivariant and can be modeled using an equivariant convolutional neural network. As a result, we are…