Selected Publications
Investigating the role of model-based learning in exploration and transfer (2023).
Proceedings of the International Conference on Machine Learning (ICML 2023).
Proceedings of the International Conference on Machine Learning (ICML 2023).
Procedural generalization by planning with self-supervised world models (2022).
Proceedings of the International Conference on Learning Representations (ICLR 2022).
Proceedings of the International Conference on Learning Representations (ICLR 2022).
On the role of planning in model-based deep reinforcement learning (2021).
Proceedings of the International Conference on Learning Representations (ICLR 2021).
Proceedings of the International Conference on Learning Representations (ICLR 2021).
Combining Q-Learning and Search with Amortized Value Estimates (2020).
Proceedings of the International Conference on Learning Representations (ICLR 2020).
Proceedings of the International Conference on Learning Representations (ICLR 2020).
Levels of Analysis for Machine Learning (2020).
Proceedings of the ICLR 2020 Workshop on Bridging AI and Cognitive Science.
Proceedings of the ICLR 2020 Workshop on Bridging AI and Cognitive Science.
Analogues of mental simulation and imagination in deep learning (2019).
Current Opinion in Behavioral Sciences.
Current Opinion in Behavioral Sciences.
Structured agents for physical construction (2019).
Proceedings of the International Conference on Machine Learning (ICML 2019).
Proceedings of the International Conference on Machine Learning (ICML 2019).
Relational inductive biases, deep learning, and graph networks (2018).
arXiv preprint arXiv:1806.01261.
arXiv preprint arXiv:1806.01261.
Simulation as an engine of physical scene understanding (2013).
Proceedings of the National Academy of Sciences.
Proceedings of the National Academy of Sciences.
All Publications
For the most up-to-date list, please see my Google Scholar.
(2024).
Intuitive physics as probabilistic inference.
Bayesian models of cognition: reverse engineering the mind.
(2024).
Transformers meet Neural Algorithmic Reasoners.
CVPR 2024 Multimodal Algorithmic Reasoning (MAR) Workshop.
(2023).
Beyond Temporal Credit Assignment in Reinforcement Learning.
Proceedings of the Workshop on Reincarnating Reinforcement Learning at ICLR 2023.
(2023).
Investigating the role of model-based learning in exploration and transfer.
Proceedings of the International Conference on Machine Learning (ICML 2023).
(2022).
Inverse design for fluid-structure interactions using graph network simulators.
Proceedings of the Conference on Neural Information Processing Systems (NeurIPS 2022).
(2022).
Learning causal overhypotheses through exploration in children and computational models.
Proceedings of the Conference on Causal Learning and Reasoning.
(2022).
Procedural generalization by planning with self-supervised world models.
Proceedings of the International Conference on Learning Representations (ICLR 2022).
(2021).
On the role of planning in model-based deep reinforcement learning.
Proceedings of the International Conference on Learning Representations (ICLR 2021).
(2020).
Combining Q-Learning and Search with Amortized Value Estimates.
Proceedings of the International Conference on Learning Representations (ICLR 2020).
(2020).
Divide-and-Conquer Monte Carlo Tree Search For Goal-Directed Planning.
arXiv preprint arXiv:2004.11410.
(2020).
Exploring Exploration: Comparing Children with RL Agents in Unified Environments.
Proceedings of the ICLR 2020 Workshop on Bridging AI and Cognitive Science.
(2020).
Levels of Analysis for Machine Learning.
Proceedings of the ICLR 2020 Workshop on Bridging AI and Cognitive Science.
(2019).
Analogues of mental simulation and imagination in deep learning.
Current Opinion in Behavioral Sciences.
(2019).
Object-oriented state editing for HRL.
Proceedings of the NeurIPS 2019 Workshop on Perception as Generative Reasoning.
(2019).
Structured agents for physical construction.
Proceedings of the International Conference on Machine Learning (ICML 2019).
(2018).
Relational inductive bias for physical construction in humans and machines.
Proceedings of the 40th Annual Conference of the Cognitive Science Society.
(2018).
Relational inductive biases, deep learning, and graph networks.
arXiv preprint arXiv:1806.01261.
(2017).
Discovering simple heuristics from mental simulation.
Proceedings of the 39th Annual Conference of the Cognitive Science Society.
(2017).
Metacontrol for Adaptive Imagination-Based Optimization.
Proceedings of the 5th International Conference on Learning Representations (ICLR 2017).
(2017).
Metareasoning and Mental Simulation.
PhD Thesis.
(2017).
Pragmatic-Pedagogic Value Alignment.
Proceedings of the International Symposium on Robotics Research (ISRR 2017).
(2016).
A Rejection Sampler.
The Architecture of Open Source Applications, Volume 4: 500 Lines or Less.
(2016).
Generating plans that predict themselves.
Proceedings of the 12th International Workshop on the Algorithmic Foundations of Robotics (WAFR 2016).
(2016).
Goal Inference Improves Objective and Perceived Performance in Human-Robot Collaboration.
Proceedings of the 15th International Conference on Autonomous Agents and Multiagent Systems (AAMAS 2016).
(2016).
Imagination-Based Decision Making with Physical Models in Deep Neural Networks.
Proceedings of the NeurIPS 2016 Workshop on Intuitive Physics.
(2016).
Inferring Mass in Complex Scenes by Mental Simulation.
Cognition.
(2016).
Jupyter Notebooks—A publishing format for reproducible computational workflows.
Proceedings of the 20th International Conference on Electronic Publishing.
(2015).
psiTurk: An open-source framework for conducting replicable behavioral experiments online.
Behavioral Research Methods.
(2015).
Relevant and Robust: A Response to Marcus and Davis (2013).
Psychological Science.
(2015).
Think again? The amount of mental simulation tracks uncertainty in the outcome.
Proceedings of the 37th Annual Conference of the Cognitive Science Society.
(2014).
Algorithm selection by rational metareasoning as a model of human strategy selection.
Advances in Neural Information Processing Systems 27.
(2014).
What to simulate? Inferring the right direction for mental rotation.
Proceedings of the 36th Annual Meeting of the Cognitive Science Society.
(2013).
Approximating Bayesian inference with a sparse distributed memory system.
Proceedings of the 35th Annual Conference of the Cognitive Science Society.
(2013).
Mental Rotation as Bayesian Quadrature.
Bayesian Optimization Workshop at NeurIPS 2013.
(2013).
Simulation as an engine of physical scene understanding.
Proceedings of the National Academy of Sciences.
(2012).
Physical Reasoning in Complex Scenes is Sensitive to Mass.
Master’s Thesis.
(2011).
Internal physics models guide probabilistic judgments about object dynamics.
Proceedings of the 33rd Annual Conference of the Cognitive Science Society.