leiloong / nlil

Neural Logic Inductive Learning

Neural Logic Inductive Learning

This is the implementation of the Neural Logic Inductive Learning model (NLIL) proposed in the ICLR 2020 paper: Learn to Explain Efficiently via Neural Logic Inductive Learning. The Transformer implementation is based on this repo.

Requirements

• python 3.6+
• pytorch 1.1.0+
• numpy
• tqdm

Knowledge completion on WN18 and FB15K

You can run knowledge completion task on WN18 and FB15K with provided scripts

bash run_wn.sh
bash run_fb.sh

Object classification on Visual Genome

First, download the scene-graph dataset from the official site (click "Download Scene Graphs")

https://cs.stanford.edu/people/dorarad/gqa/download.html

Extract the files, and run the following script to generate the dataset

bash preprocess.sh path/to/the/sgraph/folder

Now you can run object classification with

bash run_gqa.sh

Reference

@inproceedings{
    yang2020learn,
    title={Learn to Explain Efficiently via Neural Logic Inductive Learning},
    author={Yuan Yang and Le Song},
    booktitle={International Conference on Learning Representations},
    year={2020},
    url={https://openreview.net/forum?id=SJlh8CEYDB}
}