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

Code for the Shortformer model, from the paper by Ofir Press, Noah A. Smith and Mike Lewis.

Certified Patch Robustness via Smoothed Vision Transformers

AI virtual gym is an AI program which can be used to exercise and can be used to see if we are doing the exercises

Using some basic methods to show linkages and transformations of robotic arms

Region-aware Contrastive Learning for Semantic Segmentation, ICCV 2021

ZeroGen: Efficient Zero-shot Learning via Dataset Generation

Source code of D-HAN: Dynamic News Recommendation with Hierarchical Attention Network

Image Captioning on google cloud platform based on iot

Tensorflow implementation of "Learning Deconvolution Network for Semantic Segmentation"

基于深度强化学习的原神自动钓鱼AI

Accelerated Multi-Modal MR Imaging with Transformers

This is a TensorFlow implementation for C2-Rec

Learning infinite-resolution image processing with GAN and RL from unpaired image datasets, using a differentiable photo editing model.

The software associated with a paper accepted at EMNLP 2021 titled "Open Knowledge Graphs Canonicalization using Variational Autoencoders".

Unsupervised Real-World Super-Resolution: A Domain Adaptation Perspective

Using OpenAI's CLIP to upscale and enhance images

Noise Conditional Score Networks (NeurIPS 2019, Oral)

JUSTICE: A Benchmark Dataset for Supreme Court’s Judgment Prediction

A set of Deep Reinforcement Learning Agents implemented in Tensorflow.

This repo holds codes of the ICCV21 paper: Visual Alignment Constraint for Continuous Sign Language Recognition.