Mambelli, Träuble, Bauer, Schölkopf, Locatello, 2022. Compositional Multi-Object Reinforcement Learning with Linear Relation Networks.