Learning Neuro-Symbolic Relational Transition Models for Bilevel Planning
Not on the map. Why?
Abstract (arXiv)
In robotic domains, learning and planning are complicated by continuous state spaces, continuous action spaces, and long task horizons. In this work, we address these challenges with Neuro-Symbolic Relational Transition Models (NSRTs), a novel class of models that are data-efficient to learn, compatible with powerful robotic planning methods, and generalizable over objects. NSRTs have both symbolic and neural components, enabling a bilevel planning scheme where symbolic AI planning in an outer loop guides continuous planning with neural models in an inner loop. Experiments in four robotic planning domains show that NSRTs can be learned after only tens or hundreds of training episodes, and then used for fast planning in new tasks that require up to 60 actions and involve many more objects than were seen during training. Video: https://tinyurl.com/chitnis-nsrts
Key facts. Cited by 43 papers in the map (data through 7 Oct 2026). Builds on 38 of its 75 references that are in the map. First posted on arXiv 28 May 2021.
Builds on
38 of 75 references.
References · 38
- 2014Adam: A Method for Stochastic Optimizationcited by 36,960
- 2020Offline Reinforcement Learning: Tutorial, Review, and Perspectives on Open Problemscited by 1,873
- 2019Dream to Control: Learning Behaviors by Latent Imaginationcited by 1,598
- 2018Relational inductive biases, deep learning, and graph networkscited by 1,341
- 2015Fast and Accurate Deep Network Learning by Exponential Linear Units (ELUs)cited by 1,061
- 2020Mastering Atari with Discrete World Modelscited by 990
- 2017AI2-THOR: An Interactive 3D Environment for Visual AIcited by 850
- 2018Deep Reinforcement Learning in a Handful of Trials using Probabilistic Dynamics Modelscited by 748
- 2016Interaction Networks for Learning about Objects, Relations and Physicscited by 503
- 2017Neural Network Dynamics for Model-Based Deep Reinforcement Learning with Model-Free Fine-Tuningcited by 288
- 2020Integrated Task and Motion Planningcited by 281
- 2020Accelerating Reinforcement Learning with Learned Skill Priorscited by 220
26 more
- 2020OPAL: Offline Primitive Discovery for Accelerating Offline Reinforcement Learningcited by 164
- 2020Parrot: Data-Driven Behavioral Priors for Reinforcement Learningcited by 136
- 2016A Compositional Object-Based Approach to Learning Physical Dynamicscited by 119
- 2018SDRL: Interpretable and Data-efficient Deep Reinforcement Learning Leveraging Symbolic Planningcited by 77
- 2019Online Replanning in Belief Space for Partially Observable Task and Motion Problemscited by 60
- 2020Long-Horizon Visual Planning with Goal-Conditioned Hierarchical Predictorscited by 60
- 2017Classical Planning in Deep Latent Space: Bridging the Subsymbolic-Symbolic Boundarycited by 58
- 2017Schema Networks: Zero-shot Transfer with a Generative Causal Model of Intuitive Physicscited by 57
- 2020On the role of planning in model-based deep reinforcement learningcited by 57
- 2021Learning Symbolic Operators for Task and Motion Planningcited by 57
- 2018PEORL: Integrating Symbolic Planning and Hierarchical Reinforcement Learning for Robust Decision-Makingcited by 51
- 2020Learning compositional models of robot skills for task and motion planningcited by 47
- 2020Deep Visual Reasoning: Learning to Predict Action Sequences for Task and Motion Planning from an Initial Scene Imagecited by 45
- 2020Planning with Learned Object Importance in Large Problem Instances using Graph Neural Networkscited by 38
- 2018Learning Features and Abstract Actions for Computing Generalized Planscited by 35
- 2020Generalized Planning With Deep Reinforcement Learningcited by 35
- 2017Combining Model-Based and Model-Free Updates for Trajectory-Centric Reinforcement Learningcited by 27
- 2020Broadly-Exploring, Local-Policy Trees for Long-Horizon Task Planningcited by 26
- 2019Unsupervised Grounding of Plannable First-Order Logic Representation from Imagescited by 24
- 2020Understanding the Power and Limitations of Teaching with Imperfect Knowledgecited by 23
- 2020SPOTTER: Extending Symbolic Planning Operators through Targeted Reinforcement Learningcited by 23
- 2018Composable Planning with Attributescited by 21
- 2020Divide-and-Conquer Monte Carlo Tree Search For Goal-Directed Planningcited by 19
- 2018Learning sparse relational transition modelscited by 11
- 2020Planning from Pixels in Atari with Learned Symbolic Representationscited by 11
- 2019What Should I Do Now? Marrying Reinforcement Learning and Symbolic Planningcited by 9
Built on by
Citing papers per year: 2021: 1; 2022: 7; 2023: 10; 2024: 9; 2025: 8; 2026: 8.
43 papers.
Sort:
By area:
- 2022ProcTHOR: Large-Scale Embodied AI Using Procedural Generationcited by 369
- 2023Text2Motion: From Natural Language Instructions to Feasible Planscited by 223
- 2022Learning Neuro-Symbolic Skills for Bilevel Planningcited by 55
- 2022Predicate Invention for Bilevel Planningcited by 49
- 2024VisualPredicator: Learning Abstract World Models with Neuro-Symbolic Predicates for Robot Planningcited by 37
- 2025Neuro-Symbolic AI in 2024: A Systematic Reviewcited by 36
- 2022LEAGUE: Guided Skill Learning and Abstraction for Long-Horizon Manipulationcited by 32
- 2024A Survey of Optimization-based Task and Motion Planning: From Classical To Learning Approachescited by 19
- 2024Trust the PRoC3S: Solving Long-Horizon Robotics Problems with LLMs and Constraint Satisfactioncited by 19
- 2022Learning Efficient Abstract Planning Models that Choose What to Predictcited by 17
- 2023Embodied Lifelong Learning for Task and Motion Planningcited by 17
- 2024Imperative Learning: A Self-supervised Neuro-Symbolic Learning Framework for Robot Autonomycited by 16
31 more
- 2023Primitive Skill-based Robot Learning from Human Evaluative Feedbackcited by 15
- 2021Trajectory Prediction with Linguistic Representationscited by 13
- 2022Goal-Space Planning with Subgoal Modelscited by 12
- 2024From Pixels to Predicates: Learning Symbolic World Models via Pretrained Vision-Language Modelscited by 11
- 2023Embodied Active Learning of Relational State Abstractions for Bilevel Planningcited by 8
- 2023NOIR: Neural Signal Operated Intelligent Robots for Everyday Activitiescited by 8
- 2024LogiCity: Advancing Neuro-Symbolic AI with Abstract Urban Simulationcited by 8
- 2025ExoPredicator: Learning Abstract Models of Dynamic Worlds for Robot Planningcited by 8
- 2023PDSketch: Integrated Planning Domain Programming and Learningcited by 7
- 2023Egocentric Planning for Scalable Embodied Task Achievementcited by 7
- 2023Robot Task Planning Based on Large Language Model Representing Knowledge with Directed Graph Structurescited by 7
- 2025Neuro-Symbolic Imitation Learning: Discovering Symbolic Abstractions for Skill Learningcited by 4
- 2022Learning Temporally Extended Skills in Continuous Domains as Symbolic Actions for Planningcited by 3
- 2023Discovering Predictive Relational Object Symbols with Symbolic Attentive Layerscited by 3
- 2024Learning Planning Abstractions from Languagecited by 3
- 2025Curiosity-Driven Imagination: Discovering Plan Operators and Learning Associated Policies for Open-World Adaptationcited by 3
- 2025SLAP: Shortcut Learning for Abstract Planningcited by 3
- 2026Kintsugi: Learning Policies by Repairing Executable Knowledge Basescited by 3
- 2023Learning Type-Generalized Actions for Symbolic Planningcited by 2
- 2025Few-Shot Neuro-Symbolic Imitation Learning for Long-Horizon Planning and Actingcited by 2
- 2025Neuro-Symbolic AI for Cybersecurity: State of the Art, Challenges, and Opportunitiescited by 2
- 2026Learning Lifted Action Models from Unsupervised Visual Tracescited by 2
- 2024Neuro-Symbolic Skill Discovery for Conditional Multi-Level Planningcited by 1
- 2024A New View on Planning in Online Reinforcement Learningcited by 0
- 2025NOIR 2.0: Neural Signal Operated Intelligent Robots for Everyday Activitiescited by 0
- 2026Build on Priors: Vision--Language--Guided Neuro-Symbolic Imitation Learning for Data-Efficient Real-World Robot Manipulationcited by 0
- 2026Learning Bilevel Policies over Symbolic World Models for Long-Horizon Planningcited by 0
- 2026Neuro-Symbolic Learning for Long-Horizon Task Planning Under Complex Logical Constraintscited by 0
- 2026NeSy-CSA: A Neuro-Symbolic Framework for Open-Ended Critical Scenario Attributioncited by 0
- 2026Learning Action Models with Conditional and Quantified Effects via Uncertainty-Guided Explorationcited by 0
- 2026EMPIRIC: Experiment-Driven Learning of Residual World Models for Robot Planningcited by 0
- 2026EMPIRIC: Experiment-Driven Learning of Residual World Models for Robot Planning28 Sep 2026
- 2026Learning Action Models with Conditional and Quantified Effects via Uncertainty-Guided Exploration31 Aug 2026
- 2026NeSy-CSA: A Neuro-Symbolic Framework for Open-Ended Critical Scenario Attribution4 Jul 2026
- 2026Neuro-Symbolic Learning for Long-Horizon Task Planning Under Complex Logical Constraints5 Jun 2026
- 2026Learning Bilevel Policies over Symbolic World Models for Long-Horizon Planning15 May 2026
- 2026Kintsugi: Learning Policies by Repairing Executable Knowledge Bases10 May 2026
- 2026Learning Lifted Action Models from Unsupervised Visual Traces21 Apr 2026
- 2026Build on Priors: Vision--Language--Guided Neuro-Symbolic Imitation Learning for Data-Efficient Real-World Robot Manipulation4 Apr 2026
- 2025NOIR 2.0: Neural Signal Operated Intelligent Robots for Everyday Activities25 Nov 2025
- 2025SLAP: Shortcut Learning for Abstract Planning2 Nov 2025
- 2025ExoPredicator: Learning Abstract Models of Dynamic Worlds for Robot Planning30 Sep 2025
- 2025Neuro-Symbolic AI for Cybersecurity: State of the Art, Challenges, and Opportunities8 Sep 2025
31 more
- 2025Few-Shot Neuro-Symbolic Imitation Learning for Long-Horizon Planning and Acting29 Aug 2025
- 2025Neuro-Symbolic Imitation Learning: Discovering Symbolic Abstractions for Skill Learning27 Mar 2025
- 2025Curiosity-Driven Imagination: Discovering Plan Operators and Learning Associated Policies for Open-World Adaptation6 Mar 2025
- 2025Neuro-Symbolic AI in 2024: A Systematic Review9 Jan 2025
- 2024From Pixels to Predicates: Learning Symbolic World Models via Pretrained Vision-Language Models31 Dec 2024
- 2024LogiCity: Advancing Neuro-Symbolic AI with Abstract Urban Simulation1 Nov 2024
- 2024VisualPredicator: Learning Abstract World Models with Neuro-Symbolic Predicates for Robot Planning30 Oct 2024
- 2024Neuro-Symbolic Skill Discovery for Conditional Multi-Level Planning13 Oct 2024
- 2024Imperative Learning: A Self-supervised Neuro-Symbolic Learning Framework for Robot Autonomy23 Jun 2024
- 2024Trust the PRoC3S: Solving Long-Horizon Robotics Problems with LLMs and Constraint Satisfaction8 Jun 2024
- 2024A New View on Planning in Online Reinforcement Learning3 Jun 2024
- 2024Learning Planning Abstractions from Language6 May 2024
- 2024A Survey of Optimization-based Task and Motion Planning: From Classical To Learning Approaches3 Apr 2024
- 2023NOIR: Neural Signal Operated Intelligent Robots for Everyday Activities2 Nov 2023
- 2023Discovering Predictive Relational Object Symbols with Symbolic Attentive Layers2 Sep 2023
- 2023Learning Type-Generalized Actions for Symbolic Planning9 Aug 2023
- 2023Primitive Skill-based Robot Learning from Human Evaluative Feedback28 Jul 2023
- 2023Embodied Lifelong Learning for Task and Motion Planning13 Jul 2023
- 2023Robot Task Planning Based on Large Language Model Representing Knowledge with Directed Graph Structures8 Jun 2023
- 2023Egocentric Planning for Scalable Embodied Task Achievement2 Jun 2023
- 2023Text2Motion: From Natural Language Instructions to Feasible Plans21 Mar 2023
- 2023PDSketch: Integrated Planning Domain Programming and Learning9 Mar 2023
- 2023Embodied Active Learning of Relational State Abstractions for Bilevel Planning8 Mar 2023
- 2022LEAGUE: Guided Skill Learning and Abstraction for Long-Horizon Manipulation23 Oct 2022
- 2022Learning Efficient Abstract Planning Models that Choose What to Predict16 Aug 2022
- 2022Learning Temporally Extended Skills in Continuous Domains as Symbolic Actions for Planning11 Jul 2022
- 2022Learning Neuro-Symbolic Skills for Bilevel Planning21 Jun 2022
- 2022ProcTHOR: Large-Scale Embodied AI Using Procedural Generation14 Jun 2022
- 2022Goal-Space Planning with Subgoal Models6 Jun 2022
- 2022Predicate Invention for Bilevel Planning17 Mar 2022
- 2021Trajectory Prediction with Linguistic Representations19 Oct 2021
- 2023Text2Motion: From Natural Language Instructions to Feasible Planscited by 223
- 2024VisualPredicator: Learning Abstract World Models with Neuro-Symbolic Predicates for Robot Planningcited by 37
- 2024From Pixels to Predicates: Learning Symbolic World Models via Pretrained Vision-Language Modelscited by 11
- 2025ExoPredicator: Learning Abstract Models of Dynamic Worlds for Robot Planningcited by 8
- 2026Learning Bilevel Policies over Symbolic World Models for Long-Horizon Planningcited by 0
- 2026EMPIRIC: Experiment-Driven Learning of Residual World Models for Robot Planningcited by 0
- 2026EMPIRIC: Experiment-Driven Learning of Residual World Models for Robot Planning28 Sep 2026
- 2026Learning Bilevel Policies over Symbolic World Models for Long-Horizon Planning15 May 2026
- 2025ExoPredicator: Learning Abstract Models of Dynamic Worlds for Robot Planning30 Sep 2025
- 2024From Pixels to Predicates: Learning Symbolic World Models via Pretrained Vision-Language Models31 Dec 2024
- 2024VisualPredicator: Learning Abstract World Models with Neuro-Symbolic Predicates for Robot Planning30 Oct 2024
- 2023Text2Motion: From Natural Language Instructions to Feasible Plans21 Mar 2023
- 2024Imperative Learning: A Self-supervised Neuro-Symbolic Learning Framework for Robot Autonomycited by 16
- 2025Neuro-Symbolic Imitation Learning: Discovering Symbolic Abstractions for Skill Learningcited by 4
- 2025Few-Shot Neuro-Symbolic Imitation Learning for Long-Horizon Planning and Actingcited by 2
- 2026Build on Priors: Vision--Language--Guided Neuro-Symbolic Imitation Learning for Data-Efficient Real-World Robot Manipulationcited by 0
- 2026Build on Priors: Vision--Language--Guided Neuro-Symbolic Imitation Learning for Data-Efficient Real-World Robot Manipulation4 Apr 2026
- 2025Few-Shot Neuro-Symbolic Imitation Learning for Long-Horizon Planning and Acting29 Aug 2025
- 2025Neuro-Symbolic Imitation Learning: Discovering Symbolic Abstractions for Skill Learning27 Mar 2025
- 2024Imperative Learning: A Self-supervised Neuro-Symbolic Learning Framework for Robot Autonomy23 Jun 2024
- 2022ProcTHOR: Large-Scale Embodied AI Using Procedural Generationcited by 369
- 2022Learning Neuro-Symbolic Skills for Bilevel Planningcited by 55
- 2022Predicate Invention for Bilevel Planningcited by 49
- 2025Neuro-Symbolic AI in 2024: A Systematic Reviewcited by 36
- 2022LEAGUE: Guided Skill Learning and Abstraction for Long-Horizon Manipulationcited by 32
- 2024A Survey of Optimization-based Task and Motion Planning: From Classical To Learning Approachescited by 19
- 2024Trust the PRoC3S: Solving Long-Horizon Robotics Problems with LLMs and Constraint Satisfactioncited by 19
- 2022Learning Efficient Abstract Planning Models that Choose What to Predictcited by 17
- 2023Embodied Lifelong Learning for Task and Motion Planningcited by 17
- 2021Trajectory Prediction with Linguistic Representationscited by 13
- 2022Goal-Space Planning with Subgoal Modelscited by 12
- 2023Embodied Active Learning of Relational State Abstractions for Bilevel Planningcited by 8
12 most cited of 31 All 31 (JSON)
- 2026Learning Action Models with Conditional and Quantified Effects via Uncertainty-Guided Exploration31 Aug 2026
- 2026NeSy-CSA: A Neuro-Symbolic Framework for Open-Ended Critical Scenario Attribution4 Jul 2026
- 2026Neuro-Symbolic Learning for Long-Horizon Task Planning Under Complex Logical Constraints5 Jun 2026
- 2026Kintsugi: Learning Policies by Repairing Executable Knowledge Bases10 May 2026
- 2026Learning Lifted Action Models from Unsupervised Visual Traces21 Apr 2026
- 2025NOIR 2.0: Neural Signal Operated Intelligent Robots for Everyday Activities25 Nov 2025
- 2025SLAP: Shortcut Learning for Abstract Planning2 Nov 2025
- 2025Neuro-Symbolic AI for Cybersecurity: State of the Art, Challenges, and Opportunities8 Sep 2025
- 2025Curiosity-Driven Imagination: Discovering Plan Operators and Learning Associated Policies for Open-World Adaptation6 Mar 2025
- 2025Neuro-Symbolic AI in 2024: A Systematic Review9 Jan 2025
- 2024Neuro-Symbolic Skill Discovery for Conditional Multi-Level Planning13 Oct 2024
- 2024Trust the PRoC3S: Solving Long-Horizon Robotics Problems with LLMs and Constraint Satisfaction8 Jun 2024
12 newest of 31 All 31 (JSON)