To appear in NeurIPS 2019.

Abstract Reasoning with Distracting Features

(Kecheng Zheng, Wei Wei and Zheng-jun Zha, 2019)

To download the dataset, please check chizhang's project page.

For details, please check our paper.

Important

• Python 2.7
• PyTorch
• CUDA and cuDNN

python main.py --net <model name> --datapath <path to the dataset> --rl False --typeloss False

#TODO

1. The code for PGM dataset
2. The long-term features (state) in teacher model

If you find the paper and/or the code helpful, please cite us.

@inproceedings{zheng2019abstract,
    title={Abstract Reasoning with Distracting Features},
    author={Kecheng Zheng and Zheng-jun Zha and Wei Wei},
    booktitle={Advances in Neural Information Processing Systems},
    year={2019}}
}

• Wild Relational Network

Please feel free to discuss paper/code through issues or emails.

Apache License 2.0