Improving Generalization for Abstract Reasoning Tasks Using Disentangled Feature Representations
Representation Learning with Contrastive Predictive Coding