← Search

David Klee

7 accepted papers

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
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

Equivariant Single View Pose Prediction Via Induced and Restriction Representations

NeurIPS 2023poster

Learning about the three-dimensional world from two-dimensional images is a fundamental problem in computer vision. An ideal neural network architecture for such tasks would leverage the fact that objects can be rotated and translated in three dimensions to make predictions about novel images. Howev…

Cited by 8SourcePDFScholar
2023

Image to Sphere: Learning Equivariant Features for Efficient Pose Prediction

ICLR 2023top-5%

Predicting the pose of objects from a single image is an important but difficult computer vision problem. Methods that predict a single point estimate do not predict the pose of objects with symmetries well and cannot represent uncertainty. Alternatively, some works predict a distribution over orien…

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