text: Generalization performance (F1 score) of differrent architectures for the reward function in the IMAGINE setting. MA denotes an architecture based on deep sets posessing the object-centered bias; FA and FC denote flat, non-object-centered baselines. Stars indicate significant difference.