Causal Object-Centric Models for Planning with Monte Carlo Tree Search

Rodion Vakhitov, Leonid Ugadiarov, Alexey Skrynnik +1

We introduce COMET (Causal Object-centric Model for Efficient Tree search), a model-based reinforcement learning algorithm that performs Monte Carlo Tree Search in a slot-structure…