Learning Plannable Representations with Causal InfoGAN. - Thanard Kurutach; Aviv Tamar; Ge Yang 003; Stuart J. Russell; Pieter Abbeel (NeurIPS, 2018)

The Importance of Sampling inMeta-Reinforcement Learning. - Bradly C. Stadie; Ge Yang 003; Rein Houthooft; Xi Chen 022; Yan Duan; Yuhuai Wu; Pieter Abbeel; Ilya Sutskever (NeurIPS, 2018)

Some Considerations on Learning to Explore via Meta-Reinforcement Learning. - Bradly C. Stadie; Ge Yang 003; Rein Houthooft; Xi Chen 022; Yan Duan; Yuhuai Wu; Pieter Abbeel; Ilya Sutskever (CoRR, 2018)

Plan2Vec: Unsupervised Representation Learning by Latent Plans. - Ge Yang 003; Amy Zhang 001; Ari S. Morcos; Joelle Pineau; Pieter Abbeel; Roberto Calandra (L4DC, 2020)

Learning Task Informed Abstractions. - Xiang Fu 005; Ge Yang 003; Pulkit Agrawal 001; Tommi S. Jaakkola (ICML, 2021)

World Model as a Graph: Learning Latent Landmarks for Planning. - Lunjun Zhang; Ge Yang 003; Bradly C. Stadie (ICML, 2021)

Invariance Through Inference. - Takuma Yoneda; Ge Yang 003; Matthew R. Walter; Bradly C. Stadie (CoRR, 2021)

Bi-linear Value Networks for Multi-goal Reinforcement Learning. - Zhang-Wei Hong; Ge Yang 003; Pulkit Agrawal 001 (ICLR, 2022)

Overcoming The Spectral Bias of Neural Value Approximation. - Ge Yang 003; Anurag Ajay; Pulkit Agrawal 001 (ICLR, 2022)

Invariance Through Latent Alignment. - Takuma Yoneda; Ge Yang 003; Matthew R. Walter; Bradly C. Stadie (Robotics: Science and Systems, 2022)

Bilinear value networks. - Zhang-Wei Hong; Ge Yang 003; Pulkit Agrawal 001 (CoRR, 2022)

Compositional Sculpting of Iterative Generative Processes. - Timur Garipov; Sebastiaan De Peuter; Ge Yang 003; Vikas Garg 001; Samuel Kaski; Tommi S. Jaakkola (NeurIPS, 2023)

To the Noise and Back: Diffusion for Shared Autonomy. - Takuma Yoneda; Luzhe Sun; Ge Yang 003; Bradly C. Stadie; Matthew R. Walter (Robotics: Science and Systems, 2023)

Rapid locomotion via reinforcement learning. - Gabriel B. Margolis; Ge Yang 003; Kartik Paigwar; Tao Chen 046; Pulkit Agrawal 001 (Int. J. Robotics Res., 2024)