Goto

Collaborating Authors

 efficient multi-object reinforcement learning


Learning Dynamic Attribute-factored World Models for Efficient Multi-object Reinforcement Learning

Neural Information Processing Systems

In many reinforcement learning tasks, the agent has to learn to interact with many objects of different types and generalize to unseen combinations and numbers of objects. Often a task is a composition of previously learned tasks (e.g.


Learning Dynamic Attribute-factored World Models for Efficient Multi-object Reinforcement Learning

Neural Information Processing Systems

In many reinforcement learning tasks, the agent has to learn to interact with many objects of different types and generalize to unseen combinations and numbers of objects. Often a task is a composition of previously learned tasks (e.g. Recent works have shown the benefits of object-factored representations and hierarchical abstractions for improving sample efficiency in these settings. On the other hand, these methods do not fully exploit the benefits of factorization in terms of object attributes. In this paper, we address this opportunity and introduce the Dynamic Attribute FacTored RL (DAFT-RL) framework.