Neural Task Graphs: Generalizing to Unseen Tasks from a Single Video Demonstration
Not on the map. Why?
Abstract (arXiv)
Our goal is to generate a policy to complete an unseen task given just a single video demonstration of the task in a given domain. We hypothesize that to successfully generalize to unseen complex tasks from a single video demonstration, it is necessary to explicitly incorporate the compositional structure of the tasks into the model. To this end, we propose Neural Task Graph (NTG) Networks, which use conjugate task graph as the intermediate representation to modularize both the video demonstration and the derived policy. We empirically show NTG achieves inter-task generalization on two complex tasks: Block Stacking in BulletPhysics and Object Collection in AI2-THOR. NTG improves data efficiency with visual input as well as achieve strong generalization without the need for dense hierarchical supervision. We further show that similar performance trends hold when applied to real-world data. We show that NTG can effectively predict task structure on the JIGSAWS surgical dataset and generalize to unseen tasks.
Key facts. Cited by 60 papers in the map (data through 7 Oct 2026). First posted on arXiv 10 Jul 2018.
Builds on
No reference list yet.
Built on by
Citing papers per year: 2018: 1; 2019: 3; 2020: 1; 2021: 14; 2022: 16; 2023: 14; 2024: 5; 2025: 4; 2026: 2.
60 papers.
Sort:
By area:
- 2022Do As I Can, Not As I Say: Grounding Language in Robotic Affordancescited by 2,302
- 2022BC-Z: Zero-Shot Task Generalization with Robotic Imitation Learningcited by 517
- 2023RH20T: A Comprehensive Robotic Dataset for Learning Diverse Skills in One-Shotcited by 173
- 2022You Only Demonstrate Once: Category-Level Manipulation from Single Visual Demonstrationcited by 79
- 2024OAKINK2: A Dataset of Bimanual Hands-Object Manipulation in Complex Task Completioncited by 78
- 2021Augmenting Reinforcement Learning with Behavior Primitives for Diverse Manipulation Taskscited by 75
- 2023Grounded Decoding: Guiding Text Generation with Grounded Models for Embodied Agentscited by 66
- 2021UMPNet: Universal Manipulation Policy Network for Articulated Objectscited by 64
- 2021Learning to Synthesize Programs as Interpretable and Generalizable Policiescited by 59
- 2022TASKOGRAPHY: Evaluating robot task planning over large 3D scene graphscited by 59
- 2021SORNet: Spatial Object-Centric Representations for Sequential Manipulationcited by 48
- 2021Towards More Generalizable One-shot Visual Imitation Learningcited by 48
38 more
- 2021NeRP: Neural Rearrangement Planning for Unknown Objectscited by 33
- 2022Generalizing Goal-Conditioned Reinforcement Learning with Variational Causal Reasoningcited by 33
- 2018One-Shot Hierarchical Imitation Learning of Compound Visuomotor Taskscited by 32
- 2022LEAGUE: Guided Skill Learning and Abstraction for Long-Horizon Manipulationcited by 32
- 2019Few-Shot Bayesian Imitation Learning with Logical Program Policiescited by 30
- 2021Predicting Stable Configurations for Semantic Placement of Novel Objectscited by 23
- 2022Policy Architectures for Compositional Generalization in Controlcited by 23
- 2023Universal Visual Decomposer: Long-Horizon Manipulation Made Easycited by 23
- 2020Scalable Multi-Task Imitation Learning with Autonomous Improvementcited by 21
- 2021Efficient and Interpretable Robot Manipulation with Graph Neural Networkscited by 21
- 2022Generating Executable Action Plans with Environmentally-Aware Language Modelscited by 21
- 2024A Survey of Optimization-based Task and Motion Planning: From Classical To Learning Approachescited by 19
- 2022STAP: Sequencing Task-Agnostic Policiescited by 17
- 2022Cross-Domain Transfer via Semantic Skill Imitationcited by 17
- 2025Learning Compositional Behaviors from Demonstration and Languagecited by 15
- 2023Multimodal Subtask Graph Generation from Instructional Videoscited by 13
- 2023United We Stand, Divided We Fall: UnityGraph for Unsupervised Procedure Learning from Videoscited by 13
- 2021Example-Driven Model-Based Reinforcement Learning for Solving Long-Horizon Visuomotor Taskscited by 12
- 2023EC^2: Emergent Communication for Embodied Controlcited by 10
- 2019Continuous Relaxation of Symbolic Planner for One-Shot Imitation Learningcited by 8
- 2023Action Dynamics Task Graphs for Learning Plannable Representations of Procedural Taskscited by 8
- 2024PRIME: Scaffolding Manipulation Tasks with Behavior Primitives for Data-Efficient Imitation Learningcited by 8
- 2023Masked Diffusion with Task-awareness for Procedure Planning in Instructional Videoscited by 7
- 2019Motion Reasoning for Goal-Based Imitation Learningcited by 6
- 2021TANGO: Commonsense Generalization in Predicting Tool Interactions for Mobile Manipulatorscited by 6
- 2022Learning Sensorimotor Primitives of Sequential Manipulation Tasks from Visual Demonstrationscited by 6
- 2023Neural Constraint Satisfaction: Hierarchical Abstraction for Combinatorial Generalization in Object Rearrangementcited by 6
- 2025Chain-of-Modality: Learning Manipulation Programs from Multimodal Human Videos with Vision-Language-Modelscited by 6
- 2021Discovering Generalizable Skills via Automated Generation of Diverse Taskscited by 5
- 2022Zero-Shot Policy Transfer with Disentangled Task Representation of Meta-Reinforcement Learningcited by 5
- 2021Learning from demonstrations with SACR2: Soft Actor-Critic with Reward Relabelingcited by 4
- 2023TOD-Flow: Modeling the Structure of Task-Oriented Dialoguescited by 4
- 2022Learning Parameterized Task Structure for Generalization to Unseen Entitiescited by 3
- 2022Fast Inference and Transfer of Compositional Task Structures for Few-shot Task Generalizationcited by 3
- 2023Moving Forward by Moving Backward: Embedding Action Impact over Action Semanticscited by 3
- 2023Evaluating Robustness of Visual Representations for Object Assembly Task Requiring Spatio-Geometrical Reasoningcited by 3
- 2024Learning to Recover from Plan Execution Errors during Robot Manipulation: A Neuro-symbolic Approachcited by 3
- 2021Projection: A Mechanism for Human-like Reasoning in Artificial Intelligencecited by 2
50 most cited of 60 All 60 (JSON)
- 2026VLA-Corrector: Stage-Aware Observable State Understanding for Prompt-Based Closed-Loop Recovery of Vision-Language-Action Policies6 Sep 2026
- 2026See and Switch: Vision-Based Branching for Interactive Robot-Skill Programming9 Mar 2026
- 2025ManiLong-Shot: Interaction-Aware One-Shot Imitation Learning for Long-Horizon Manipulation18 Dec 2025
- 2025Learning-based Cooperative Robotic Paper Wrapping: A Unified Control Policy with Residual Force Control5 Nov 2025
- 2025Learning Compositional Behaviors from Demonstration and Language28 May 2025
- 2025Chain-of-Modality: Learning Manipulation Programs from Multimodal Human Videos with Vision-Language-Models17 Apr 2025
- 2024Box2Flow: Instance-based Action Flow Graphs from Videos30 Aug 2024
- 2024Learning to Recover from Plan Execution Errors during Robot Manipulation: A Neuro-symbolic Approach29 May 2024
- 2024A Survey of Optimization-based Task and Motion Planning: From Classical To Learning Approaches3 Apr 2024
- 2024OAKINK2: A Dataset of Bimanual Hands-Object Manipulation in Complex Task Completion28 Mar 2024
- 2024PRIME: Scaffolding Manipulation Tasks with Behavior Primitives for Data-Efficient Imitation Learning1 Mar 2024
- 2023TOD-Flow: Modeling the Structure of Task-Oriented Dialogues7 Dec 2023
38 more
- 2023United We Stand, Divided We Fall: UnityGraph for Unsupervised Procedure Learning from Videos6 Nov 2023
- 2023Robot Skill Generalization via Keypoint Integrated Soft Actor-Critic Gaussian Mixture Models23 Oct 2023
- 2023Evaluating Robustness of Visual Representations for Object Assembly Task Requiring Spatio-Geometrical Reasoning15 Oct 2023
- 2023Universal Visual Decomposer: Long-Horizon Manipulation Made Easy12 Oct 2023
- 2023Compositional Servoing by Recombining Demonstrations6 Oct 2023
- 2023Masked Diffusion with Task-awareness for Procedure Planning in Instructional Videos14 Sep 2023
- 2023RH20T: A Comprehensive Robotic Dataset for Learning Diverse Skills in One-Shot2 Jul 2023
- 2023Moving Forward by Moving Backward: Embedding Action Impact over Action Semantics24 Apr 2023
- 2023EC^2: Emergent Communication for Embodied Control19 Apr 2023
- 2023Neural Constraint Satisfaction: Hierarchical Abstraction for Combinatorial Generalization in Object Rearrangement20 Mar 2023
- 2023Grounded Decoding: Guiding Text Generation with Grounded Models for Embodied Agents1 Mar 2023
- 2023Multimodal Subtask Graph Generation from Instructional Videos17 Feb 2023
- 2023Action Dynamics Task Graphs for Learning Plannable Representations of Procedural Tasks11 Jan 2023
- 2022Cross-Domain Transfer via Semantic Skill Imitation14 Dec 2022
- 2022LEAGUE: Guided Skill Learning and Abstraction for Long-Horizon Manipulation23 Oct 2022
- 2022STAP: Sequencing Task-Agnostic Policies21 Oct 2022
- 2022Generating Executable Action Plans with Environmentally-Aware Language Models10 Oct 2022
- 2022Zero-Shot Policy Transfer with Disentangled Task Representation of Meta-Reinforcement Learning1 Oct 2022
- 2022Generalizing Goal-Conditioned Reinforcement Learning with Variational Causal Reasoning19 Jul 2022
- 2022TASKOGRAPHY: Evaluating robot task planning over large 3D scene graphs11 Jul 2022
- 2022Fast Inference and Transfer of Compositional Task Structures for Few-shot Task Generalization25 May 2022
- 2022Do As I Can, Not As I Say: Grounding Language in Robotic Affordances4 Apr 2022
- 2022Learning Parameterized Task Structure for Generalization to Unseen Entities28 Mar 2022
- 2022Policy Architectures for Compositional Generalization in Control10 Mar 2022
- 2022Learning Sensorimotor Primitives of Sequential Manipulation Tasks from Visual Demonstrations8 Mar 2022
- 2022BC-Z: Zero-Shot Task Generalization with Robotic Imitation Learning4 Feb 2022
- 2022You Only Demonstrate Once: Category-Level Manipulation from Single Visual Demonstration30 Jan 2022
- 2022Learning Two-Step Hybrid Policy for Graph-Based Interpretable Reinforcement Learning21 Jan 2022
- 2022STIR2: Reward Relabelling for combined Reinforcement and Imitation Learning on sparse-reward tasks11 Jan 2022
- 2021Learning from demonstrations with SACR2: Soft Actor-Critic with Reward Relabeling27 Oct 2021
- 2021Towards More Generalizable One-shot Visual Imitation Learning26 Oct 2021
- 2021Augmenting Reinforcement Learning with Behavior Primitives for Diverse Manipulation Tasks7 Oct 2021
- 2021Example-Driven Model-Based Reinforcement Learning for Solving Long-Horizon Visuomotor Tasks21 Sep 2021
- 2021UMPNet: Universal Manipulation Policy Network for Articulated Objects13 Sep 2021
- 2021SORNet: Spatial Object-Centric Representations for Sequential Manipulation8 Sep 2021
- 2021Video2Skill: Adapting Events in Demonstration Videos to Skills in an Environment using Cyclic MDP Homomorphisms8 Sep 2021
- 2021Learning to Synthesize Programs as Interpretable and Generalizable Policies31 Aug 2021
- 2021Predicting Stable Configurations for Semantic Placement of Novel Objects26 Aug 2021
50 newest of 60 All 60 (JSON)
- 2021Towards More Generalizable One-shot Visual Imitation Learningcited by 48
- 2018One-Shot Hierarchical Imitation Learning of Compound Visuomotor Taskscited by 32
- 2019Few-Shot Bayesian Imitation Learning with Logical Program Policiescited by 30
- 2020Scalable Multi-Task Imitation Learning with Autonomous Improvementcited by 21
- 2025Learning Compositional Behaviors from Demonstration and Languagecited by 15
- 2019Continuous Relaxation of Symbolic Planner for One-Shot Imitation Learningcited by 8
- 2024PRIME: Scaffolding Manipulation Tasks with Behavior Primitives for Data-Efficient Imitation Learningcited by 8
- 2019Motion Reasoning for Goal-Based Imitation Learningcited by 6
- 2021Learning from demonstrations with SACR2: Soft Actor-Critic with Reward Relabelingcited by 4
- 2022STIR2: Reward Relabelling for combined Reinforcement and Imitation Learning on sparse-reward taskscited by 1
- 2025ManiLong-Shot: Interaction-Aware One-Shot Imitation Learning for Long-Horizon Manipulationcited by 0
- 2025ManiLong-Shot: Interaction-Aware One-Shot Imitation Learning for Long-Horizon Manipulation18 Dec 2025
- 2025Learning Compositional Behaviors from Demonstration and Language28 May 2025
- 2024PRIME: Scaffolding Manipulation Tasks with Behavior Primitives for Data-Efficient Imitation Learning1 Mar 2024
- 2022STIR2: Reward Relabelling for combined Reinforcement and Imitation Learning on sparse-reward tasks11 Jan 2022
- 2021Learning from demonstrations with SACR2: Soft Actor-Critic with Reward Relabeling27 Oct 2021
- 2021Towards More Generalizable One-shot Visual Imitation Learning26 Oct 2021
- 2020Scalable Multi-Task Imitation Learning with Autonomous Improvement25 Feb 2020
- 2019Motion Reasoning for Goal-Based Imitation Learning13 Nov 2019
- 2019Continuous Relaxation of Symbolic Planner for One-Shot Imitation Learning16 Aug 2019
- 2019Few-Shot Bayesian Imitation Learning with Logical Program Policies12 Apr 2019
- 2018One-Shot Hierarchical Imitation Learning of Compound Visuomotor Tasks25 Oct 2018
- 2022Do As I Can, Not As I Say: Grounding Language in Robotic Affordancescited by 2,302
- 2023RH20T: A Comprehensive Robotic Dataset for Learning Diverse Skills in One-Shotcited by 173
- 2022You Only Demonstrate Once: Category-Level Manipulation from Single Visual Demonstrationcited by 79
- 2024OAKINK2: A Dataset of Bimanual Hands-Object Manipulation in Complex Task Completioncited by 78
- 2021Augmenting Reinforcement Learning with Behavior Primitives for Diverse Manipulation Taskscited by 75
- 2021UMPNet: Universal Manipulation Policy Network for Articulated Objectscited by 64
- 2021Learning to Synthesize Programs as Interpretable and Generalizable Policiescited by 59
- 2022TASKOGRAPHY: Evaluating robot task planning over large 3D scene graphscited by 59
- 2021SORNet: Spatial Object-Centric Representations for Sequential Manipulationcited by 48
- 2021NeRP: Neural Rearrangement Planning for Unknown Objectscited by 33
- 2022LEAGUE: Guided Skill Learning and Abstraction for Long-Horizon Manipulationcited by 32
- 2021Predicting Stable Configurations for Semantic Placement of Novel Objectscited by 23
12 most cited of 44 All 44 (JSON)
- 2026See and Switch: Vision-Based Branching for Interactive Robot-Skill Programming9 Mar 2026
- 2025Learning-based Cooperative Robotic Paper Wrapping: A Unified Control Policy with Residual Force Control5 Nov 2025
- 2025Chain-of-Modality: Learning Manipulation Programs from Multimodal Human Videos with Vision-Language-Models17 Apr 2025
- 2024Box2Flow: Instance-based Action Flow Graphs from Videos30 Aug 2024
- 2024Learning to Recover from Plan Execution Errors during Robot Manipulation: A Neuro-symbolic Approach29 May 2024
- 2024A Survey of Optimization-based Task and Motion Planning: From Classical To Learning Approaches3 Apr 2024
- 2024OAKINK2: A Dataset of Bimanual Hands-Object Manipulation in Complex Task Completion28 Mar 2024
- 2023TOD-Flow: Modeling the Structure of Task-Oriented Dialogues7 Dec 2023
- 2023United We Stand, Divided We Fall: UnityGraph for Unsupervised Procedure Learning from Videos6 Nov 2023
- 2023Robot Skill Generalization via Keypoint Integrated Soft Actor-Critic Gaussian Mixture Models23 Oct 2023
- 2023Evaluating Robustness of Visual Representations for Object Assembly Task Requiring Spatio-Geometrical Reasoning15 Oct 2023
- 2023Universal Visual Decomposer: Long-Horizon Manipulation Made Easy12 Oct 2023
12 newest of 44 All 44 (JSON)