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

PyTorch implementation of Decoupling Value and Policy for Generalization in Reinforcement Learning

A Python module for the generation and training of an entry-level feedforward neural network.

Chess reinforcement learning by AlphaGo Zero methods.

This repo is to be freely used by ML devs to check the GAN performances without coding from scratch.

Implementing DropPath/StochasticDepth in PyTorch

The official implementation for "FQ-ViT: Fully Quantized Vision Transformer without Retraining".

MarcoPolo is a clustering-free approach to the exploration of bimodally expressed genes along with group information in single-cell RNA-seq data

The missing CMake project initializer

Line-level Handwritten Text Recognition (HTR) system implemented with TensorFlow.

Unicorn can be used for performance analyses of highly configurable systems with causal reasoning

RARA: Zero-shot Sim2Real Visual Navigation with Following Foreground Cues

SC-GlowTTS: an Efficient Zero-Shot Multi-Speaker Text-To-Speech Model

PyTorch implementation of ARM-Net: Adaptive Relation Modeling Network for Structured Data.

Deep learning with TensorFlow and earth observation data.

Official PyTorch code for WACV 2022 paper "CFLOW-AD: Real-Time Unsupervised Anomaly Detection with Localization via Conditional Normalizing Flows"

Code of Classification Saliency-Based Rule for Visible and Infrared Image Fusion

This repository comes with the paper "On the Robustness of Counterfactual Explanations to Adverse Perturbations"

Solution of Kaggle competition: Sartorius - Cell Instance Segmentation

This repository provides a basic implementation of our GCPR 2021 paper "Learning Conditional Invariance through Cycle Consistency"

Mae segmentation - Reproduction of semantic segmentation using masked autoencoder (mae)