activity
20162026
most citedCategorical Reparameterization with Gumbel-Softmax

3.3k citations · 5.1k across the 19 of their papers we have counts for

collaborators

20 papers

cs.AI2026

MemWM: Memory-Augmented Text-Based World Model

Yujun Wang, Tao Zhang, Jinhe Bi +9

World models are increasingly used to support planning in agents by predicting how environment states evolve in response to agent actions. Yet fluent next-state predictions can sti…

cs.AI2022★ 27 cited

Multi-Game Decision Transformers

Kuang-Huei Lee, Ofir Nachum, Mengjiao Yang +8

A longstanding goal of the field of AI is a method for learning a highly capable, generalist agent from diverse experience. In the subfields of vision and language, this was largel…

cs.RO2022★ 524 cited

Do As I Can, Not As I Say: Grounding Language in Robotic Affordances

Michael Ahn, Anthony Brohan, Noah Brown +42

Large language models can encode a wealth of semantic knowledge about the world. Such knowledge could be extremely useful to robots aiming to act upon high-level, temporally extend…

cs.RO2022★ 3 cited

Practical Imitation Learning in the Real World via Task Consistency Loss

Mohi Khansari, Daniel Ho, Yuqing Du +6

Recent work in visual end-to-end learning for robotics has shown the promise of imitation learning across a variety of tasks. Such approaches are expensive both because they requir…

cs.RO2022★ 1 cited

Bayesian Imitation Learning for End-to-End Mobile Manipulation

Yuqing Du, Daniel Ho, Alexander A. Alemi +2

In this work we investigate and demonstrate benefits of a Bayesian approach to imitation learning from multiple sensor inputs, as applied to the task of opening office doors with a…

cs.RO2022★ 90 cited

BC-Z: Zero-Shot Task Generalization with Robotic Imitation Learning

Eric Jang, Alex Irpan, Mohi Khansari +5

In this paper, we study the problem of enabling a vision-based robotic manipulation system to generalize to novel tasks, a long-standing challenge in robot learning. We approach th…