OCAtari: Object-Centric Atari 2600 Reinforcement Learning Environments
Not on the map. Why?
Abstract (arXiv)
Cognitive science and psychology suggest that object-centric representations of complex scenes are a promising step towards enabling efficient abstract reasoning from low-level perceptual features. Yet, most deep reinforcement learning approaches only rely on pixel-based representations that do not capture the compositional properties of natural scenes. For this, we need environments and datasets that allow us to work and evaluate object-centric approaches. In our work, we extend the Atari Learning Environments, the most-used evaluation framework for deep RL approaches, by introducing OCAtari, that performs resource-efficient extractions of the object-centric states for these games. Our framework allows for object discovery, object representation learning, as well as object-centric RL. We evaluate OCAtari's detection capabilities and resource efficiency. Our source code is available at github.com/k4ntz/OC_Atari.
Key facts. Cited by 30 papers in the map (data through 14 Oct 2026). Builds on 56 of its 77 references that are in the map. First posted on arXiv 14 Jun 2023.
Builds on
56 of 77 references.
References · 56
- 2017Proximal Policy Optimization Algorithmscited by 15,215
- 2015Deep Reinforcement Learning with Double Q-learningcited by 1,745
- 2019Dream to Control: Learning Behaviors by Latent Imaginationcited by 1,653
- 2019Generalized Intersection over Union: A Metric and A Loss for Bounding Box Regressioncited by 1,368
- 2019Mastering Atari, Go, Chess and Shogi by Planning with a Learned Modelcited by 1,353
- 2020Mastering Atari with Discrete World Modelscited by 1,007
- 2018IMPALA: Scalable Distributed Deep-RL with Importance Weighted Actor-Learner Architecturescited by 883
- 2020Object-Centric Learning with Slot Attentioncited by 876
- 2017AI2-THOR: An Interactive 3D Environment for Visual AIcited by 869
- 2021Deep Reinforcement Learning at the Edge of the Statistical Precipicecited by 807
- 2017Rainbow: Combining Improvements in Deep Reinforcement Learningcited by 786
- 2017A Distributional Perspective on Reinforcement Learningcited by 765
38 more
- 2017Deep Reinforcement Learning that Matterscited by 754
- 2016Unifying Count-Based Exploration and Intrinsic Motivationcited by 639
- 2019Model-Based Reinforcement Learning for Ataricited by 502
- 2015Deep Recurrent Q-Learning for Partially Observable MDPscited by 465
- 2018VirtualHome: Simulating Household Activities via Programscited by 415
- 2019Multi-Object Representation Learning with Iterative Variational Inferencecited by 360
- 2020Improving Reproducibility in Machine Learning Research (A Report from the NeurIPS 2019 Reproducibility Program)cited by 322
- 2020Agent57: Outperforming the Atari Human Benchmarkcited by 303
- 2020Decoupling Representation Learning from Reinforcement Learningcited by 269
- 2020First return, then explorecited by 266
- 2021Conditional Object-Centric Learning from Videocited by 248
- 2019GENESIS: Generative Scene Inference and Sampling with Object-Centric Latent Representationscited by 227
- 2019Go-Explore: a New Approach for Hard-Exploration Problemscited by 212
- 2020Never Give Up: Learning Directed Exploration Strategiescited by 212
- 2016Attend, Infer, Repeat: Fast Scene Understanding with Generative Modelscited by 209
- 2021iGibson 2.0: Object-Centric Simulation for Robot Learning of Everyday Household Taskscited by 194
- 2020SPACE: Unsupervised Object-Oriented Scene Representation via Spatial Attention and Decompositioncited by 190
- 2022SAVi++: Towards End-to-End Object-Centric Learning from Real-World Videoscited by 166
- 2021Illiterate DALL-E Learns to Composecited by 147
- 2019Unsupervised State Representation Learning in Ataricited by 144
- 2020Making deep neural networks right for the right scientific reasons by interacting with their explanationscited by 142
- 2022Simple Unsupervised Object-Centric Learning for Complex and Naturalistic Videoscited by 139
- 2018Sequential Attend, Infer, Repeat: Generative Modelling of Moving Objectscited by 128
- 2021Goal Misgeneralization in Deep Reinforcement Learningcited by 123
- 2018Generalization and Regularization in DQNcited by 118
- 2021Generalization and Robustness Implications in Object-Centric Learningcited by 81
- 2019Neural Logic Reinforcement Learningcited by 53
- 2023An Investigation into Pre-Training Object-Centric Representations for Reinforcement Learningcited by 49
- 2023Interpretable and Explainable Logical Policies via Neurally Guided Symbolic Abstractioncited by 46
- 2023Read and Reap the Rewards: Learning to Play Atari with the Help of Instruction Manualscited by 45
- 2022Atari-5: Distilling the Arcade Learning Environment down to Five Gamescited by 44
- 2018Investigating Human Priors for Playing Video Gamescited by 39
- 2022Human-level Atari 200x fastercited by 28
- 2021Adaptive Rational Activations to Boost Deep Reinforcement Learningcited by 23
- 2022Compositional Multi-Object Reinforcement Learning with Linear Relation Networkscited by 19
- 2021Semantic Tracklets: An Object-Centric Representation for Visual Multi-Agent Reinforcement Learningcited by 18
- 2022Boosting Object Representation Learning via Motion and Object Continuitycited by 17
- 2019Is Deep Reinforcement Learning Really Superhuman on Atari? Leveling the playing fieldcited by 15
50 most cited of 56 All 56 (JSON)
Built on by
Citing papers per year: 2024: 11; 2025: 10; 2026: 9.
30 papers.
Sort:
By area:
- 2024Gymnasium: A Standard Interface for Reinforcement Learning Environmentscited by 46
- 2025PoE-World: Compositional World Modeling with Products of Programmatic Expertscited by 26
- 2024Interpretable Concept Bottlenecks to Align Reinforcement Learning Agentscited by 23
- 2024Uni-RLHF: Universal Platform and Benchmark Suite for Reinforcement Learning with Diverse Human Feedbackcited by 21
- 2024Pix2Code: Learning to Compose Neural Visual Concepts as Programscited by 18
- 2024Interpretable and Editable Programmatic Tree Policies for Reinforcement Learningcited by 18
- 2024End-to-End Neuro-Symbolic Reinforcement Learning with Textual Explanationscited by 17
- 2024BlendRL: A Framework for Merging Symbolic and Neural Policy Learningcited by 14
- 2024HackAtari: Atari Learning Environments for Robust and Continual Reinforcement Learningcited by 11
- 2025Object-Centric World Models from Few-Shot Annotations for Sample-Efficient Reinforcement Learningcited by 10
- 2025One Life to Learn: Inferring Symbolic World Models for Stochastic Environments from Unguided Explorationcited by 10
- 2025Learning Game-Playing Agents with Generative Code Optimizationcited by 8
18 more
- 2026Understanding the Challenges in Iterative Generative Optimization with LLMscited by 8
- 2026Hierarchical Entity-centric Reinforcement Learning with Factored Subgoal Diffusioncited by 7
- 2024EXPIL: Explanatory Predicate Invention for Learning in Gamescited by 5
- 2024LICORICE: Label-Efficient Concept-Based Interpretable Reinforcement Learningcited by 5
- 2024Interpretable end-to-end Neurosymbolic Reinforcement Learning agentscited by 4
- 2025Evaluating Interpretable Reinforcement Learning by Distilling Policies into Programscited by 3
- 2025Better Decisions through the Right Causal World Modelcited by 3
- 2025The Formalism-Implementation Gap in Reinforcement Learning Researchcited by 3
- 2025Deep Reinforcement Learning via Object-Centric Attentioncited by 2
- 2025DyMoDreamer: World Modeling with Dynamic Modulationcited by 2
- 2025Human-Allied Relational Reinforcement Learningcited by 1
- 2026Joint Learning of Hierarchical Neural Options and Abstract World Modelcited by 1
- 2026Boosting deep Reinforcement Learning using pretraining with Logical Optionscited by 0
- 2026See, Symbolize, Act: Grounding VLMs with Spatial Representations for Better Gameplaycited by 0
- 2026GRAIL: Autonomous Concept Grounding for Neuro-Symbolic Reinforcement Learningcited by 0
- 2026Language Game: Talking to Non-Human Systemscited by 0
- 2026Mind-Studio: Executable World Models with Lookahead Evaluation for Partially Observable Gamescited by 0
- 2026A Differentiable Atari VCS:A Complex, Fully Known Ground Truth for Explainable AIcited by 0
- 2026A Differentiable Atari VCS:A Complex, Fully Known Ground Truth for Explainable AI21 Jun 2026
- 2026Mind-Studio: Executable World Models with Lookahead Evaluation for Partially Observable Games14 Jun 2026
- 2026Language Game: Talking to Non-Human Systems5 May 2026
- 2026GRAIL: Autonomous Concept Grounding for Neuro-Symbolic Reinforcement Learning18 Apr 2026
- 2026Understanding the Challenges in Iterative Generative Optimization with LLMs25 Mar 2026
- 2026See, Symbolize, Act: Grounding VLMs with Spatial Representations for Better Gameplay12 Mar 2026
- 2026Boosting deep Reinforcement Learning using pretraining with Logical Options6 Mar 2026
- 2026Joint Learning of Hierarchical Neural Options and Abstract World Model2 Feb 2026
- 2026Hierarchical Entity-centric Reinforcement Learning with Factored Subgoal Diffusion2 Feb 2026
- 2025Human-Allied Relational Reinforcement Learning17 Oct 2025
- 2025The Formalism-Implementation Gap in Reinforcement Learning Research17 Oct 2025
- 2025One Life to Learn: Inferring Symbolic World Models for Stochastic Environments from Unguided Exploration14 Oct 2025
18 more
- 2025DyMoDreamer: World Modeling with Dynamic Modulation29 Sep 2025
- 2025Learning Game-Playing Agents with Generative Code Optimization27 Aug 2025
- 2025PoE-World: Compositional World Modeling with Products of Programmatic Experts16 May 2025
- 2025Better Decisions through the Right Causal World Model9 Apr 2025
- 2025Deep Reinforcement Learning via Object-Centric Attention3 Apr 2025
- 2025Evaluating Interpretable Reinforcement Learning by Distilling Policies into Programs11 Mar 2025
- 2025Object-Centric World Models from Few-Shot Annotations for Sample-Efficient Reinforcement Learning27 Jan 2025
- 2024Interpretable end-to-end Neurosymbolic Reinforcement Learning agents18 Oct 2024
- 2024BlendRL: A Framework for Merging Symbolic and Neural Policy Learning15 Oct 2024
- 2024Gymnasium: A Standard Interface for Reinforcement Learning Environments24 Jul 2024
- 2024LICORICE: Label-Efficient Concept-Based Interpretable Reinforcement Learning22 Jul 2024
- 2024EXPIL: Explanatory Predicate Invention for Learning in Games10 Jun 2024
- 2024HackAtari: Atari Learning Environments for Robust and Continual Reinforcement Learning6 Jun 2024
- 2024Interpretable and Editable Programmatic Tree Policies for Reinforcement Learning23 May 2024
- 2024End-to-End Neuro-Symbolic Reinforcement Learning with Textual Explanations19 Mar 2024
- 2024Pix2Code: Learning to Compose Neural Visual Concepts as Programs13 Feb 2024
- 2024Uni-RLHF: Universal Platform and Benchmark Suite for Reinforcement Learning with Diverse Human Feedback4 Feb 2024
- 2024Interpretable Concept Bottlenecks to Align Reinforcement Learning Agents11 Jan 2024
- 2024Gymnasium: A Standard Interface for Reinforcement Learning Environmentscited by 46
- 2024Interpretable Concept Bottlenecks to Align Reinforcement Learning Agentscited by 23
- 2024Interpretable and Editable Programmatic Tree Policies for Reinforcement Learningcited by 18
- 2024End-to-End Neuro-Symbolic Reinforcement Learning with Textual Explanationscited by 17
- 2024HackAtari: Atari Learning Environments for Robust and Continual Reinforcement Learningcited by 11
- 2026Hierarchical Entity-centric Reinforcement Learning with Factored Subgoal Diffusioncited by 7
- 2024EXPIL: Explanatory Predicate Invention for Learning in Gamescited by 5
- 2024LICORICE: Label-Efficient Concept-Based Interpretable Reinforcement Learningcited by 5
- 2024Interpretable end-to-end Neurosymbolic Reinforcement Learning agentscited by 4
- 2025Evaluating Interpretable Reinforcement Learning by Distilling Policies into Programscited by 3
- 2025The Formalism-Implementation Gap in Reinforcement Learning Researchcited by 3
- 2025Deep Reinforcement Learning via Object-Centric Attentioncited by 2
12 most cited of 14 All 14 (JSON)
- 2026Boosting deep Reinforcement Learning using pretraining with Logical Options6 Mar 2026
- 2026Hierarchical Entity-centric Reinforcement Learning with Factored Subgoal Diffusion2 Feb 2026
- 2025Human-Allied Relational Reinforcement Learning17 Oct 2025
- 2025The Formalism-Implementation Gap in Reinforcement Learning Research17 Oct 2025
- 2025Deep Reinforcement Learning via Object-Centric Attention3 Apr 2025
- 2025Evaluating Interpretable Reinforcement Learning by Distilling Policies into Programs11 Mar 2025
- 2024Interpretable end-to-end Neurosymbolic Reinforcement Learning agents18 Oct 2024
- 2024Gymnasium: A Standard Interface for Reinforcement Learning Environments24 Jul 2024
- 2024LICORICE: Label-Efficient Concept-Based Interpretable Reinforcement Learning22 Jul 2024
- 2024EXPIL: Explanatory Predicate Invention for Learning in Games10 Jun 2024
- 2024HackAtari: Atari Learning Environments for Robust and Continual Reinforcement Learning6 Jun 2024
- 2024Interpretable and Editable Programmatic Tree Policies for Reinforcement Learning23 May 2024
12 newest of 14 All 14 (JSON)
- 2025PoE-World: Compositional World Modeling with Products of Programmatic Expertscited by 26
- 2025Object-Centric World Models from Few-Shot Annotations for Sample-Efficient Reinforcement Learningcited by 10
- 2025One Life to Learn: Inferring Symbolic World Models for Stochastic Environments from Unguided Explorationcited by 10
- 2025Better Decisions through the Right Causal World Modelcited by 3
- 2025DyMoDreamer: World Modeling with Dynamic Modulationcited by 2
- 2026Joint Learning of Hierarchical Neural Options and Abstract World Modelcited by 1
- 2026Mind-Studio: Executable World Models with Lookahead Evaluation for Partially Observable Gamescited by 0
- 2026Mind-Studio: Executable World Models with Lookahead Evaluation for Partially Observable Games14 Jun 2026
- 2026Joint Learning of Hierarchical Neural Options and Abstract World Model2 Feb 2026
- 2025One Life to Learn: Inferring Symbolic World Models for Stochastic Environments from Unguided Exploration14 Oct 2025
- 2025DyMoDreamer: World Modeling with Dynamic Modulation29 Sep 2025
- 2025PoE-World: Compositional World Modeling with Products of Programmatic Experts16 May 2025
- 2025Better Decisions through the Right Causal World Model9 Apr 2025
- 2025Object-Centric World Models from Few-Shot Annotations for Sample-Efficient Reinforcement Learning27 Jan 2025
- 2024Pix2Code: Learning to Compose Neural Visual Concepts as Programscited by 18
- 2026Understanding the Challenges in Iterative Generative Optimization with LLMscited by 8
- 2026Language Game: Talking to Non-Human Systemscited by 0
- 2026A Differentiable Atari VCS:A Complex, Fully Known Ground Truth for Explainable AIcited by 0
- 2026A Differentiable Atari VCS:A Complex, Fully Known Ground Truth for Explainable AI21 Jun 2026
- 2026Language Game: Talking to Non-Human Systems5 May 2026
- 2026Understanding the Challenges in Iterative Generative Optimization with LLMs25 Mar 2026
- 2024Pix2Code: Learning to Compose Neural Visual Concepts as Programs13 Feb 2024