Schema Networks: Zero-shot Transfer with a Generative Causal Model of Intuitive Physics
Not on the map. Why?
Abstract (arXiv)
The recent adaptation of deep neural network-based methods to reinforcement learning and planning domains has yielded remarkable progress on individual tasks. Nonetheless, progress on task-to-task transfer remains limited. In pursuit of efficient and robust generalization, we introduce the Schema Network, an object-oriented generative physics simulator capable of disentangling multiple causes of events and reasoning backward through causes to achieve goals. The richly structured architecture of the Schema Network can learn the dynamics of an environment directly from data. We compare Schema Networks with Asynchronous Advantage Actor-Critic and Progressive Networks on a suite of Breakout variations, reporting results on training efficiency and zero-shot generalization, consistently demonstrating faster, more robust learning and better transfer. We argue that generalizing from limited data and learning causal relationships are essential abilities on the path toward generally intelligent systems.
Key facts. Cited by 57 papers in the map (data through 7 Oct 2026). First posted on arXiv 14 Jun 2017.
Builds on
No reference list yet.
Built on by
Citing papers per year: 2018: 1; 2019: 1; 2020: 0; 2021: 15; 2022: 7; 2023: 15; 2024: 7; 2025: 6; 2026: 5.
57 papers.
Sort:
By area:
- 2021A Survey of Zero-shot Generalisation in Deep Reinforcement Learningcited by 164
- 2024WorldCoder, a Model-Based LLM Agent: Building World Models by Writing Code and Interacting with the Environmentcited by 99
- 2021A Survey on Interpretable Reinforcement Learningcited by 70
- 2023An Investigation into Pre-Training Object-Centric Representations for Reinforcement Learningcited by 49
- 2025What Has a Foundation Model Found? Using Inductive Bias to Probe for World Modelscited by 46
- 2021Abstract Spatial-Temporal Reasoning via Probabilistic Abduction and Executioncited by 43
- 2021Learning Neuro-Symbolic Relational Transition Models for Bilevel Planningcited by 43
- 2021Human-Level Reinforcement Learning through Theory-Based Modeling, Exploration, and Planningcited by 43
- 2023A Survey on Causal Reinforcement Learningcited by 43
- 2019Few-Shot Bayesian Imitation Learning with Logical Program Policiescited by 30
- 2021Learning Algebraic Representation for Systematic Generalization in Abstract Reasoningcited by 27
- 2022Symmetry-Based Representations for Artificial and Biological General Intelligencecited by 26
38 more
- 2025SIMA 2: A Generalist Embodied Agent for Virtual Worldscited by 26
- 2025PoE-World: Compositional World Modeling with Products of Programmatic Expertscited by 25
- 2023Causal Reinforcement Learning: A Surveycited by 22
- 2021Capturing the objects of vision with neural networkscited by 21
- 2023Explore to Generalize in Zero-Shot RLcited by 20
- 2021HALMA: Humanlike Abstraction Learning Meets Affordance in Rapid Problem Solvingcited by 15
- 2022On the link between conscious function and general intelligence in humans and machinescited by 15
- 2022FAIR for AI: An interdisciplinary and international community building perspectivecited by 13
- 2025Synthesizing world models for bilevel planningcited by 13
- 2021Discovering User-Interpretable Capabilities of Black-Box Planning Agentscited by 10
- 2021Theoretically Principled Deep RL Acceleration via Nearest Neighbor Function Approximationcited by 10
- 2018LIFT: Reinforcement Learning in Computer Systems by Learning From Demonstrationscited by 8
- 2023Conceptual Reinforcement Learning for Language-Conditioned Taskscited by 8
- 2023PDSketch: Integrated Planning Domain Programming and Learningcited by 7
- 2024From Frege to chatGPT: Compositionality in language, cognition, and deep neural networkscited by 7
- 2026OPINE-World: Programmatic World Modeling with Ontology-error-Prioritized Interactive Exploration for ARC-AGI-3cited by 7
- 2022Object-Category Aware Reinforcement Learningcited by 6
- 2023Granger Causal Interaction Skill Chainscited by 5
- 2023PushWorld: A benchmark for manipulation planning with tools and movable obstaclescited by 3
- 2023DetermiNet: A Large-Scale Diagnostic Dataset for Complex Visually-Grounded Referencing using Determinerscited by 3
- 2026World Models: A Comprehensive Survey of Architectures, Methodologies, Reasoning Paradigms, and Applicationscited by 3
- 2021Spatially Invariant Unsupervised 3D Object-Centric Learning and Scene Decompositioncited by 2
- 2023Q-Cogni: An Integrated Causal Reinforcement Learning Frameworkcited by 2
- 2023A Review of Symbolic, Subsymbolic and Hybrid Methods for Sequential Decision Makingcited by 2
- 2023Procedural Content Generation via Knowledge Transformation (PCG-KT)cited by 2
- 2026SafeGen-LLM: Enhancing Safety Generalization in Task Planning for Robotic Systemscited by 2
- 2021A Classification of Artificial Intelligence Systems for Mathematics Educationcited by 1
- 2023The Construction of Reality in an AI: A Reviewcited by 1
- 2023The Role of Diverse Replay for Generalisation in Reinforcement Learningcited by 1
- 2023Efficient Open-world Reinforcement Learning via Knowledge Distillation and Autonomous Rule Discoverycited by 1
- 2024Learning to Play Video Games with Intuitive Physics Priorscited by 1
- 2024Training on Irrelevant States Implies Data Augmentation: Generalization in Contextual MDPscited by 1
- 2025How Ensembles of Distilled Policies Improve Generalisation in Reinforcement Learningcited by 1
- 2025Causality in the human niche: lessons for machine learningcited by 1
- 2021Probabilistic Programming Bots in Intuitive Physics Game Playcited by 0
- 2021Plug and Play, Model-Based Reinforcement Learningcited by 0
- 2021Feature-Attending Recurrent Modules for Generalization in Reinforcement Learningcited by 0
- 2022Does DQN really learn? Exploring adversarial training schemes in Pongcited by 0
50 most cited of 57 All 57 (JSON)
- 2026Robust Successor Features25 Sep 2026
- 2026Object-Centric Environment Modeling for Agentic Tasks3 Jul 2026
- 2026OPINE-World: Programmatic World Modeling with Ontology-error-Prioritized Interactive Exploration for ARC-AGI-31 Jul 2026
- 2026World Models: A Comprehensive Survey of Architectures, Methodologies, Reasoning Paradigms, and Applications28 May 2026
- 2026SafeGen-LLM: Enhancing Safety Generalization in Task Planning for Robotic Systems27 Feb 2026
- 2025SIMA 2: A Generalist Embodied Agent for Virtual Worlds4 Dec 2025
- 2025What Has a Foundation Model Found? Using Inductive Bias to Probe for World Models9 Jul 2025
- 2025Causality in the human niche: lessons for machine learning13 Jun 2025
- 2025How Ensembles of Distilled Policies Improve Generalisation in Reinforcement Learning22 May 2025
- 2025PoE-World: Compositional World Modeling with Products of Programmatic Experts16 May 2025
- 2025Synthesizing world models for bilevel planning26 Mar 2025
- 2024Training on Irrelevant States Implies Data Augmentation: Generalization in Contextual MDPs4 Oct 2024
38 more
- 2024Learning to Play Video Games with Intuitive Physics Priors20 Sep 2024
- 2024Value Internalization: Learning and Generalizing from Social Reward19 Jul 2024
- 2024Explore-Go: Leveraging Exploration for Generalisation in Deep Reinforcement Learning12 Jun 2024
- 2024From Frege to chatGPT: Compositionality in language, cognition, and deep neural networks24 May 2024
- 2024WorldCoder, a Model-Based LLM Agent: Building World Models by Writing Code and Interacting with the Environment19 Feb 2024
- 2024Inductive Models for Artificial Intelligence Systems are Insufficient without Good Explanations17 Jan 2024
- 2023Efficient Open-world Reinforcement Learning via Knowledge Distillation and Autonomous Rule Discovery24 Nov 2023
- 2023DetermiNet: A Large-Scale Diagnostic Dataset for Complex Visually-Grounded Referencing using Determiners7 Sep 2023
- 2023Causal Reinforcement Learning: A Survey4 Jul 2023
- 2023Granger Causal Interaction Skill Chains15 Jun 2023
- 2023The Role of Diverse Replay for Generalisation in Reinforcement Learning9 Jun 2023
- 2023Explore to Generalize in Zero-Shot RL5 Jun 2023
- 2023Procedural Content Generation via Knowledge Transformation (PCG-KT)1 May 2023
- 2023A Review of Symbolic, Subsymbolic and Hybrid Methods for Sequential Decision Making20 Apr 2023
- 2023PDSketch: Integrated Planning Domain Programming and Learning9 Mar 2023
- 2023Conceptual Reinforcement Learning for Language-Conditioned Tasks9 Mar 2023
- 2023Q-Cogni: An Integrated Causal Reinforcement Learning Framework26 Feb 2023
- 2023A Survey on Causal Reinforcement Learning10 Feb 2023
- 2023An Investigation into Pre-Training Object-Centric Representations for Reinforcement Learning9 Feb 2023
- 2023The Construction of Reality in an AI: A Review3 Feb 2023
- 2023PushWorld: A benchmark for manipulation planning with tools and movable obstacles24 Jan 2023
- 2022Hippocampus-Inspired Cognitive Architecture (HICA) for Operant Conditioning16 Dec 2022
- 2022Reward-Predictive Clustering7 Nov 2022
- 2022Object-Category Aware Reinforcement Learning13 Oct 2022
- 2022FAIR for AI: An interdisciplinary and international community building perspective30 Sep 2022
- 2022On the link between conscious function and general intelligence in humans and machines24 Mar 2022
- 2022Does DQN really learn? Exploring adversarial training schemes in Pong20 Mar 2022
- 2022Symmetry-Based Representations for Artificial and Biological General Intelligence17 Mar 2022
- 2021A Survey on Interpretable Reinforcement Learning24 Dec 2021
- 2021Feature-Attending Recurrent Modules for Generalization in Reinforcement Learning15 Dec 2021
- 2021Learning Algebraic Representation for Systematic Generalization in Abstract Reasoning25 Nov 2021
- 2021A Survey of Zero-shot Generalisation in Deep Reinforcement Learning18 Nov 2021
- 2021Theoretically Principled Deep RL Acceleration via Nearest Neighbor Function Approximation9 Oct 2021
- 2021Capturing the objects of vision with neural networks7 Sep 2021
- 2021Plug and Play, Model-Based Reinforcement Learning20 Aug 2021
- 2021Discovering User-Interpretable Capabilities of Black-Box Planning Agents28 Jul 2021
- 2021Human-Level Reinforcement Learning through Theory-Based Modeling, Exploration, and Planning27 Jul 2021
- 2021A Classification of Artificial Intelligence Systems for Mathematics Education13 Jul 2021
50 newest of 57 All 57 (JSON)
- 2024WorldCoder, a Model-Based LLM Agent: Building World Models by Writing Code and Interacting with the Environmentcited by 99
- 2025What Has a Foundation Model Found? Using Inductive Bias to Probe for World Modelscited by 46
- 2025PoE-World: Compositional World Modeling with Products of Programmatic Expertscited by 25
- 2025Synthesizing world models for bilevel planningcited by 13
- 2026OPINE-World: Programmatic World Modeling with Ontology-error-Prioritized Interactive Exploration for ARC-AGI-3cited by 7
- 2026World Models: A Comprehensive Survey of Architectures, Methodologies, Reasoning Paradigms, and Applicationscited by 3
- 2026OPINE-World: Programmatic World Modeling with Ontology-error-Prioritized Interactive Exploration for ARC-AGI-31 Jul 2026
- 2026World Models: A Comprehensive Survey of Architectures, Methodologies, Reasoning Paradigms, and Applications28 May 2026
- 2025What Has a Foundation Model Found? Using Inductive Bias to Probe for World Models9 Jul 2025
- 2025PoE-World: Compositional World Modeling with Products of Programmatic Experts16 May 2025
- 2025Synthesizing world models for bilevel planning26 Mar 2025
- 2024WorldCoder, a Model-Based LLM Agent: Building World Models by Writing Code and Interacting with the Environment19 Feb 2024
- 2025SIMA 2: A Generalist Embodied Agent for Virtual Worldscited by 26
- 2026Object-Centric Environment Modeling for Agentic Taskscited by 0
- 2026Object-Centric Environment Modeling for Agentic Tasks3 Jul 2026
- 2025SIMA 2: A Generalist Embodied Agent for Virtual Worlds4 Dec 2025
- 2021A Survey of Zero-shot Generalisation in Deep Reinforcement Learningcited by 164
- 2021A Survey on Interpretable Reinforcement Learningcited by 70
- 2023An Investigation into Pre-Training Object-Centric Representations for Reinforcement Learningcited by 49
- 2021Abstract Spatial-Temporal Reasoning via Probabilistic Abduction and Executioncited by 43
- 2021Learning Neuro-Symbolic Relational Transition Models for Bilevel Planningcited by 43
- 2021Human-Level Reinforcement Learning through Theory-Based Modeling, Exploration, and Planningcited by 43
- 2023A Survey on Causal Reinforcement Learningcited by 43
- 2021Learning Algebraic Representation for Systematic Generalization in Abstract Reasoningcited by 27
- 2022Symmetry-Based Representations for Artificial and Biological General Intelligencecited by 26
- 2023Causal Reinforcement Learning: A Surveycited by 22
- 2021Capturing the objects of vision with neural networkscited by 21
- 2023Explore to Generalize in Zero-Shot RLcited by 20
12 most cited of 46 All 46 (JSON)
- 2026Robust Successor Features25 Sep 2026
- 2026SafeGen-LLM: Enhancing Safety Generalization in Task Planning for Robotic Systems27 Feb 2026
- 2025Causality in the human niche: lessons for machine learning13 Jun 2025
- 2025How Ensembles of Distilled Policies Improve Generalisation in Reinforcement Learning22 May 2025
- 2024Training on Irrelevant States Implies Data Augmentation: Generalization in Contextual MDPs4 Oct 2024
- 2024Learning to Play Video Games with Intuitive Physics Priors20 Sep 2024
- 2024Explore-Go: Leveraging Exploration for Generalisation in Deep Reinforcement Learning12 Jun 2024
- 2024From Frege to chatGPT: Compositionality in language, cognition, and deep neural networks24 May 2024
- 2024Inductive Models for Artificial Intelligence Systems are Insufficient without Good Explanations17 Jan 2024
- 2023Efficient Open-world Reinforcement Learning via Knowledge Distillation and Autonomous Rule Discovery24 Nov 2023
- 2023DetermiNet: A Large-Scale Diagnostic Dataset for Complex Visually-Grounded Referencing using Determiners7 Sep 2023
- 2023Causal Reinforcement Learning: A Survey4 Jul 2023
12 newest of 46 All 46 (JSON)