Graph-Structured Policy Learning for Multi-Goal Manipulation Tasks. - David Klee; Ondrej Biza; Robert Platt 001 (IROS, 2022)

Image to Icosahedral Projection for SO(3) Object Reasoning from Single-View Images. - David Klee; Ondrej Biza; Robert Platt 001; Robin Walters 001 (NeurReps, 2022)

Factored World Models for Zero-Shot Generalization in Robotic Manipulation. - Ondrej Biza; Thomas Kipf; David Klee; Robert Platt Jr. 001; Jan-Willem van de Meent; Lawson L. S. Wong (CoRR, 2022)

I2I: Image to Icosahedral Projection for SO(3) Object Reasoning from Single-View Images. - David Klee; Ondrej Biza; Robert Platt 001; Robin Walters 001 (CoRR, 2022)

Equivariant Reinforcement Learning under Partial Observability. - Hai Nguyen 001; Andrea Baisero; David Klee; Dian Wang 001; Robert Platt 001; Christopher Amato (CoRL, 2023)

Image to Sphere: Learning Equivariant Features for Efficient Pose Prediction. - David Klee; Ondrej Biza; Robert Platt 001; Robin Walters 001 (ICLR, 2023)

SEIL: Simulation-augmented Equivariant Imitation Learning. - Mingxi Jia; Dian Wang 001; Guanang Su; David Klee; Xupeng Zhu; Robin Walters 001; Robert Platt 001 (ICRA, 2023)

Equivariant Single View Pose Prediction Via Induced and Restriction Representations. - Owen Howell; David Klee; Ondrej Biza; Linfeng Zhao; Robin Walters 001 (NeurIPS, 2023)

Equivariant Single View Pose Prediction Via Induced and Restricted Representations. - Owen Howell; David Klee; Ondrej Biza; Linfeng Zhao; Robin Walters 001 (CoRR, 2023)

Equivariant Offline Reinforcement Learning. - Arsh Tangri; Ondrej Biza; Dian Wang 001; David Klee; Owen Howell; Robert Platt 001 (CoRR, 2024)

Coarse-to-Fine 3D Keyframe Transporter. - Xupeng Zhu; David Klee; Dian Wang 001; Boce Hu; Haojie Huang; Arsh Tangri; Robin Walters 001; Robert Platt 001 (CoRR, 2025)

3D Equivariant Visuomotor Policy Learning via Spherical Projection. - Boce Hu; Dian Wang 001; David Klee; Heng Tian; Xupeng Zhu; Haojie Huang; Robert Platt 001; Robin Walters 001 (CoRR, 2025)