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

Last update: Dec 19, 2022

Overview

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.

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

Related tags

Overview

PairRE

Setup

Results

ogbl-wikikg

ogbl-biokg

ogbl-wikikg2

Running the code

ogbl-wikikg

ogbl-biokg

ogbl-wikikg2

Owner

Alipay

Gluon CV Toolkit

Codes for paper "Towards Diverse Paragraph Captioning for Untrimmed Videos". CVPR 2021

Implementation of our NeurIPS 2021 paper "A Bi-Level Framework for Learning to Solve Combinatorial Optimization on Graphs".

Paper list of log-based anomaly detection

Zero-Cost Proxies for Lightweight NAS

Code and models used in "MUSS Multilingual Unsupervised Sentence Simplification by Mining Paraphrases".

Code accompanying the paper on "An Empirical Investigation of Domain Generalization with Empirical Risk Minimizers" published at NeurIPS, 2021

Randomized Correspondence Algorithm for Structural Image Editing

Deep Learning and Reinforcement Learning Library for Scientists and Engineers 🔥

Count GitHub Stars ⭐

CVNets: A library for training computer vision networks

Code for DisCo: Remedy Self-supervised Learning on Lightweight Models with Distilled Contrastive Learning

Implementation of "DeepOrder: Deep Learning for Test Case Prioritization in Continuous Integration Testing".

Benchmarks for the Optimal Power Flow Problem

Code release for NeurIPS 2020 paper "Co-Tuning for Transfer Learning"

Chatbot in 200 lines of code using TensorLayer

The code of NeurIPS 2021 paper "Scalable Rule-Based Representation Learning for Interpretable Classification".

PySOT - SenseTime Research platform for single object tracking, implementing algorithms like SiamRPN and SiamMask.

Implements Stacked-RNN in numpy and torch with manual forward and backward functions

CityLearn Challenge Multi-Agent Reinforcement Learning for Intelligent Energy Management, 2020, PikaPika team