johnoooooo / knowledgegraphembeddingsviapairedrelationvectors_pairre

PairRE

Code for paper PairRE: Knowledge Graph Embeddings via Paired Relation Vectors.

This implementation of PairRE for Open Graph Benchmak datasets (ogbl-wikikg and ogbl-biokg) is based on OGB. Thanks for their contributions.

Setup

To run the code, you need the following dependencies:

• Pytorch 1.4.0
• ogb 1.2.2 for wikikg
• ogb 1.2.4 for wikikg2

Results

The results of PairRE on ogbl-wikikg, ogbl-biokg and ogbl-wikikg2 are as follows.

ogbl-wikikg

	#Dim	#Parameters	Hardware	Test MRR	Valid MRR
PairRE	100	250,167,400	16GB GPU	0.4912 ± 0.004	0.5013 ± 0.004
PairRE	200	500,334,800	16GB GPU	0.5289 ± 0.003	0.5529 ± 0.001

ogbl-biokg

	#Dim	#Parameters	Hardware	Test MRR	Valid MRR
PairRE	2000	187,750,000	16GB GPU	0.8164 ± 0.0005	0.8172 ± 0.0005

ogbl-wikikg2

	#Dim	#Parameters	Hardware	Test MRR	Valid MRR
PairRE	100	250,167,400	16GB GPU	0.4849 ± 0.003	0.4941 ± 0.004
PairRE	200	500,334,800	16GB GPU	0.5208 ± 0.003	0.5423 ± 0.002

Running the code

ogbl-wikikg

cd wikikg && sh examples.sh

ogbl-biokg

cd biokg && sh examples.sh

ogbl-wikikg2

Please update ogb package to version 1.2.4. The hyperparameters are same to the experiments in ogbl-wikikg.

cd wikikg && sh examples.sh

The details of the optional hyperparameters can be found in examples.sh.

PairRE for FB15k, FB15k-237 and DB100k

These experiments are based on this repository. We add the score function of PairRE to this repository directly.

Commands to reproduce the reuslts are:

FB15k-237

bash run.sh train PairRE FB15k-237 0 0 1024 256 1500 6.0 1.0 0.00005 100000 16 -dr

FB15k:

bash run.sh train PairRE FB15k 0 0 1024 256 1500 15.0 1.0 0.0001 150000 16 -dr

DB100k:

bash run.sh train PairRE DB100K 0 0 1024 256 500 9.0 1.0 0.00005 150000 16 -dr