Instruct: Identify the main and secondary category of Arxiv papers based on the titles and abstracts
Query: Compositional Multi-Object Reinforcement Learning with Linear Relation
  Networks   Although reinforcement learning has seen remarkable progress over the last
years, solving robust dexterous object-manipulation tasks in multi-object
settings remains a challenge. In this paper, we focus on models that can learn
manipulation tasks in fixed multi-object settings and extrapolate this skill
zero-shot without any drop in performance when the number of objects changes.
We consider the generic task of bringing a specific cube out of a set to a goal
position. We find that previous approaches, which primarily leverage attention
and graph neural network-based architectures, do not generalize their skills
when the number of input objects changes while scaling as $K^2$. We propose an
alternative plug-and-play module based on relational inductive biases to
overcome these limitations. Besides exceeding performances in their training
environment, we show that our approach, which scales linearly in $K$, allows
agents to extrapolate and generalize zero-shot to any new object number.

22
The categories of this paper are: chao-dyn
26
The categories of this paper are: bayes-an
25
The categories of this paper are: patt-sol
9
The categories of this paper are: nlin
0
The categories of this paper are: hep-ph
4
The categories of this paper are: cond-mat
13
The categories of this paper are: quant-ph
3
The categories of this paper are: physics
8
The categories of this paper are: hep-ex
2
The categories of this paper are: math
36
The categories of this paper are: cmp-lg
18
The categories of this paper are: econ
33
The categories of this paper are: plasm-ph
14
The categories of this paper are: math-ph
37
The categories of this paper are: supr-con
19
The categories of this paper are: eess