BISCUIT: Causal Representation Learning from Binary Interactions
Description
This repository contains the datasets from the paper "BISCUIT: Causal Representation Learning from Binary Interactions" (link) by Phillip Lippe, Sara Magliacane, Sindy Löwe, Yuki M. Asano, Taco Cohen, Efstratios Gavves.
iTHOR Embodied AI - The Embodied AI dataset is generated with the iTHOR simulator. We use the default kitchen environment, FloorPlan10, and position the robot in front of the kitchen counter. The robot interacts with different objects in the environment, including a Microwave, cabinet, stove, and an egg. For more details on the dataset, see our GitHub repository and the appendix of our paper.
CausalWorld - The CausalWorld environment implements a tri-finger robot which can interact with a cube in the center of a stage. We additionally introduce causal variables for the friction parameters of the stage, floor and cube, as well as color changes. For more details on the dataset as well as the code to generate this dataset, see our GitHub repository and the appendix of our paper.
Voronoi - The Voronoi benchmark allows for creating causal systems with arbitrary number of causal variables and causal graphs. We provide datasets with 6 and 9 variables, as well as systems with minimal number of interactions. For details, see our GitHub repository and our paper.
Files
causal_world.zip
Additional details
References
- Lippe, P., Magliacane, S., Löwe, S., Asano, Y. M., Cohen, T., & Gavves, E. (2023). BISCUIT: Causal Representation Learning from Binary Interactions. In: Uncertainty in Artificial Intelligence (UAI), 2023.