Improving Generalization for Abstract Reasoning Tasks Using Disentangled Feature Representations
A simple neural network module for relational reasoning