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.
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
The results of PairRE on ogbl-wikikg, ogbl-biokg and ogbl-wikikg2 are as follows.
| #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 |
| #Dim | #Parameters | Hardware | Test MRR | Valid MRR | |
|---|---|---|---|---|---|
| PairRE | 2000 | 187,750,000 | 16GB GPU | 0.8164 ± 0.0005 | 0.8172 ± 0.0005 |
| #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 |
cd wikikg && sh examples.sh
cd biokg && sh examples.shPlease update ogb package to version 1.2.4. The hyperparameters are same to the experiments in ogbl-wikikg.
cd wikikg && sh examples.shThe details of the optional hyperparameters can be found in examples.sh.
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 -drFB15k:
bash run.sh train PairRE FB15k 0 0 1024 256 1500 15.0 1.0 0.0001 150000 16 -drDB100k:
bash run.sh train PairRE DB100K 0 0 1024 256 500 9.0 1.0 0.00005 150000 16 -dr