Transformers are Adaptable Task Planners
Not on the map. Why?
Abstract (arXiv)
Every home is different, and every person likes things done in their particular way. Therefore, home robots of the future need to both reason about the sequential nature of day-to-day tasks and generalize to user's preferences. To this end, we propose a Transformer Task Planner(TTP) that learns high-level actions from demonstrations by leveraging object attribute-based representations. TTP can be pre-trained on multiple preferences and shows generalization to unseen preferences using a single demonstration as a prompt in a simulated dishwasher loading task. Further, we demonstrate real-world dish rearrangement using TTP with a Franka Panda robotic arm, prompted using a single human demonstration.
Key facts. Cited by 15 papers in the map (data through 7 Oct 2026). Builds on 60 of its 74 references that are in the map. First posted on arXiv 6 Jul 2022.
Builds on
60 of 74 references.
References · 60
- 2017Attention Is All You Needcited by 59,932
- 2018BERT: Pre-training of Deep Bidirectional Transformers for Language Understandingcited by 40,204
- 2020Language Models are Few-Shot Learnerscited by 32,199
- 2019Exploring the Limits of Transfer Learning with a Unified Text-to-Text Transformercited by 14,491
- 2017Proximal Policy Optimization Algorithmscited by 14,441
- 2021Masked Autoencoders Are Scalable Vision Learnerscited by 7,339
- 2017Mask R-CNNcited by 7,259
- 2019BART: Denoising Sequence-to-Sequence Pre-training for Natural Language Generation, Translation, and Comprehensioncited by 5,874
- 2021Zero-Shot Text-to-Image Generationcited by 4,276
- 2017Deep reinforcement learning from human preferencescited by 3,846
- 2021Pre-train, Prompt, and Predict: A Systematic Survey of Prompting Methods in Natural Language Processingcited by 2,222
- 2021Decision Transformer: Reinforcement Learning via Sequence Modelingcited by 1,528
38 more
- 2021Multitask Prompted Training Enables Zero-Shot Task Generalizationcited by 1,462
- 2019The Replica Dataset: A Digital Replica of Indoor Spacescited by 885
- 2020Object-Centric Learning with Slot Attentioncited by 865
- 2018QT-Opt: Scalable Deep Reinforcement Learning for Vision-Based Robotic Manipulationcited by 730
- 2018Open3D: A Modern Library for 3D Data Processingcited by 717
- 2021P-Tuning v2: Prompt Tuning Can Be Comparable to Fine-tuning Universally Across Scales and Taskscited by 686
- 2020It's Not Just Size That Matters: Small Language Models Are Also Few-Shot Learnerscited by 668
- 2021Offline Reinforcement Learning as One Big Sequence Modeling Problemcited by 636
- 2018On Evaluation of Embodied Navigation Agentscited by 526
- 2021Habitat 2.0: Training Home Assistants to Rearrange their Habitatcited by 501
- 2019Normalized Object Coordinate Space for Category-Level 6D Object Pose and Size Estimationcited by 490
- 2021RMA: Rapid Motor Adaptation for Legged Robotscited by 430
- 2019MASS: Masked Sequence to Sequence Pre-training for Language Generationcited by 345
- 2020Transporter Networks: Rearranging the Visual World for Robotic Manipulationcited by 342
- 2020Integrated Task and Motion Planningcited by 281
- 2020Accelerating Reinforcement Learning with Learned Skill Priorscited by 220
- 2016Modular Multitask Reinforcement Learning with Policy Sketchescited by 200
- 2019Deep Dynamics Models for Learning Dexterous Manipulationcited by 200
- 2022Prompting Decision Transformer for Few-Shot Policy Generalizationcited by 143
- 2020Parrot: Data-Driven Behavioral Priors for Reinforcement Learningcited by 136
- 2019Self-Supervised Correspondence in Visuomotor Policy Learningcited by 106
- 2021B-Pref: Benchmarking Preference-Based Reinforcement Learningcited by 106
- 2019Self-supervised 6D Object Pose Estimation for Robot Manipulationcited by 100
- 2017Intel RealSense Stereoscopic Depth Camerascited by 82
- 2016Multi-view Self-supervised Deep Learning for 6D Pose Estimation in the Amazon Picking Challengecited by 70
- 2019KETO: Learning Keypoint Representations for Tool Manipulationcited by 68
- 2020Learning RGB-D Feature Embeddings for Unseen Object Instance Segmentationcited by 64
- 2022Reward Uncertainty for Exploration in Preference-based Reinforcement Learningcited by 60
- 2021StructFormer: Learning Spatial Structure for Language-Guided Semantic Rearrangement of Novel Objectscited by 57
- 2022Models of human preference for learning reward functionscited by 54
- 2021SORNet: Spatial Object-Centric Representations for Sequential Manipulationcited by 48
- 2020Deep Visual Reasoning: Learning to Predict Action Sequences for Task and Motion Planning from an Initial Scene Imagecited by 45
- 2021PlaTe: Visually-Grounded Planning with Transformers in Procedural Taskscited by 45
- 2021Skill Preferences: Learning to Extract and Execute Robotic Skills from Human Feedbackcited by 34
- 2022AdaPrompt: Adaptive Model Training for Prompt-based NLPcited by 30
- 2021My House, My Rules: Learning Tidying Preferences with Graph Neural Networkscited by 27
- 2021Differentiable Spatial Planning using Transformerscited by 25
- 2017Beating Atari with Natural Language Guided Reinforcement Learningcited by 24
50 most cited of 60 All 60 (JSON)
Built on by
Citing papers per year: 2022: 2; 2023: 7; 2024: 4; 2025: 1; 2026: 1.
15 papers.
Sort:
By area:
- 2022DALL-E-Bot: Introducing Web-Scale Diffusion Models to Roboticscited by 105
- 2023Toward General-Purpose Robots via Foundation Models: A Survey and Meta-Analysiscited by 99
- 2024LLM3:Large Language Model-based Task and Motion Planning with Motion Failure Reasoningcited by 63
- 2024Vid2Robot: End-to-end Video-conditioned Policy Learning with Cross-Attention Transformerscited by 47
- 2022On Transforming Reinforcement Learning by Transformer: The Development Trajectorycited by 35
- 2023Transfer Knowledge from Natural Language to Electrocardiography: Can We Detect Cardiovascular Disease Through Language Models?cited by 21
- 2023Dream2Real: Zero-Shot 3D Object Rearrangement with Vision-Language Modelscited by 16
- 2023MMSum: A Dataset for Multimodal Summarization and Thumbnail Generation of Videoscited by 12
- 2025ADAPT: Actively Discovering and Adapting to Preferences for any Taskcited by 5
- 2023SceneScore: Learning a Cost Function for Object Arrangementcited by 4
- 2023LLM-SAP: Large Language Models Situational Awareness Based Planningcited by 4
- 2024Robot Behavior Personalization from Sparse User Feedbackcited by 2
3 more
- 2023Knolling Bot: Teaching Robots the Human Notion of Tidinesscited by 0
- 2024PACA: Perspective-Aware Cross-Attention Representation for Zero-Shot Scene Rearrangementcited by 0
- 2026Offline Decision Transformers for Neural Combinatorial Optimization: Surpassing Heuristics on the Traveling Salesman Problemcited by 0
- 2026Offline Decision Transformers for Neural Combinatorial Optimization: Surpassing Heuristics on the Traveling Salesman Problem26 Mar 2026
- 2025ADAPT: Actively Discovering and Adapting to Preferences for any Task5 Apr 2025
- 2024PACA: Perspective-Aware Cross-Attention Representation for Zero-Shot Scene Rearrangement29 Oct 2024
- 2024Robot Behavior Personalization from Sparse User Feedback25 Oct 2024
- 2024Vid2Robot: End-to-end Video-conditioned Policy Learning with Cross-Attention Transformers19 Mar 2024
- 2024LLM3:Large Language Model-based Task and Motion Planning with Motion Failure Reasoning18 Mar 2024
- 2023LLM-SAP: Large Language Models Situational Awareness Based Planning26 Dec 2023
- 2023Toward General-Purpose Robots via Foundation Models: A Survey and Meta-Analysis14 Dec 2023
- 2023Dream2Real: Zero-Shot 3D Object Rearrangement with Vision-Language Models7 Dec 2023
- 2023SceneScore: Learning a Cost Function for Object Arrangement14 Nov 2023
- 2023Knolling Bot: Teaching Robots the Human Notion of Tidiness6 Oct 2023
- 2023MMSum: A Dataset for Multimodal Summarization and Thumbnail Generation of Videos7 Jun 2023
3 more
- 2023Transfer Knowledge from Natural Language to Electrocardiography: Can We Detect Cardiovascular Disease Through Language Models?21 Jan 2023
- 2022On Transforming Reinforcement Learning by Transformer: The Development Trajectory29 Dec 2022
- 2022DALL-E-Bot: Introducing Web-Scale Diffusion Models to Robotics5 Oct 2022
- 2023Toward General-Purpose Robots via Foundation Models: A Survey and Meta-Analysiscited by 99
- 2024Vid2Robot: End-to-end Video-conditioned Policy Learning with Cross-Attention Transformerscited by 47
- 2022On Transforming Reinforcement Learning by Transformer: The Development Trajectorycited by 35
- 2023Transfer Knowledge from Natural Language to Electrocardiography: Can We Detect Cardiovascular Disease Through Language Models?cited by 21
- 2023MMSum: A Dataset for Multimodal Summarization and Thumbnail Generation of Videoscited by 12
- 2023SceneScore: Learning a Cost Function for Object Arrangementcited by 4
- 2023LLM-SAP: Large Language Models Situational Awareness Based Planningcited by 4
- 2024Robot Behavior Personalization from Sparse User Feedbackcited by 2
- 2023Knolling Bot: Teaching Robots the Human Notion of Tidinesscited by 0
- 2024Robot Behavior Personalization from Sparse User Feedback25 Oct 2024
- 2024Vid2Robot: End-to-end Video-conditioned Policy Learning with Cross-Attention Transformers19 Mar 2024
- 2023LLM-SAP: Large Language Models Situational Awareness Based Planning26 Dec 2023
- 2023Toward General-Purpose Robots via Foundation Models: A Survey and Meta-Analysis14 Dec 2023
- 2023SceneScore: Learning a Cost Function for Object Arrangement14 Nov 2023
- 2023Knolling Bot: Teaching Robots the Human Notion of Tidiness6 Oct 2023
- 2023MMSum: A Dataset for Multimodal Summarization and Thumbnail Generation of Videos7 Jun 2023
- 2023Transfer Knowledge from Natural Language to Electrocardiography: Can We Detect Cardiovascular Disease Through Language Models?21 Jan 2023
- 2022On Transforming Reinforcement Learning by Transformer: The Development Trajectory29 Dec 2022