Improving Generalization for Abstract Reasoning Tasks Using Disentangled Feature Representations
auto - encoding variational bayes .