Unsupervised Abstract Reasoning for Raven’s Problem Matrices

This code is the implementation of our TIP paper.

This is the first unsupervised abstract reasoning method on Raven's Progressive Matrices, it is an extention of our arxiv preprint.

Comparision with some supervised methods.

Average testing accuracy on the RAVEN, I-RAVEN, and PGM dataset

Method	Raven	I-RAVEN	PGM
CNN	36.97	13.26	33.00
ResNet50	86.26	-	42.00
DCNet (ICLR2021)	93.58	49.36	68.57
NCD (Ours)	36.99	48.22	47.62

Generalization test results on PGM dataset

Method	neutral	interpolation	extrapolation
WReN (ICML2018)	62.6	64.4	17.2
DCNet (ICLR2021)	68.6	59.7	17.8
MXGNet (ICLR2020)	89.6	84.6	18.4
NCD (Ours)	47.6	47.0	24.9

Citation

If our code is useful for your research, please cite the following papers.

@article{zhuo2021unsup,
  title={Unsupervised Abstract Reasoning for Raven’s Problem Matrices},
  author={Tao Zhuo, Qiang Huang, and Mohan Kankanhalli},
  journal={IEEE Transactions on Image Processing},
  year={2021}
}

@article{zhuo2020solving,
  title={Solving Raven's Progressive Matrices with Neural Networks},
  author={Tao Zhuo and Mohan Kankanhalli},
  journal={arXiv preprint arXiv:2002.01646},
  year={2020}
}

@inproceedings{iclr2021,  
    author={Tao Zhuo and Mohan Kankanhalli},  
    title={Effective Abstract Reasoning with Dual-Contrast Network},  
    booktitle={International Conference on Learning Representations (ICLR)},      
    year={2021}
}