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

Bridging Composite and Real: Towards End-to-end Deep Image Matting

Code for "Share With Thy Neighbors: Single-View Reconstruction by Cross-Instance Consistency" paper

Bayes-Newton—A Gaussian process library in JAX, with a unifying view of approximate Bayesian inference as variants of Newton's algorithm.

Machine Learning Framework for Operating Systems - Brings ML to Linux kernel

A Keras implementation of YOLOv4 (Tensorflow backend)

TipToiDog - Tip Toi Dog With Python

DFFNet: An IoT-perceptive Dual Feature Fusion Network for General Real-time Semantic Segmentation

Official implementation of the method ContIG, for self-supervised learning from medical imaging with genomics

Official PyTorch implementation of N-ImageNet: Towards Robust, Fine-Grained Object Recognition with Event Cameras (ICCV 2021)

Official repository for Automated Learning Rate Scheduler for Large-Batch Training (8th ICML Workshop on AutoML)

Deploy optimized transformer based models on Nvidia Triton server

This is an official implementation of the CVPR2022 paper "Blind2Unblind: Self-Supervised Image Denoising with Visible Blind Spots".

Official implementation of the paper Image Generators with Conditionally-Independent Pixel Synthesis https://arxiv.org/abs/2011.13775

Code for NeurIPS 2021 paper "Curriculum Offline Imitation Learning"

Think Big, Teach Small: Do Language Models Distil Occam’s Razor?

Weakly Supervised Learning of Instance Segmentation with Inter-pixel Relations, CVPR 2019 (Oral)

Notepy is a full-featured Notepad Python app

rastrainer is a QGIS plugin to training remote sensing semantic segmentation model based on PaddlePaddle.

Speckle-free Holography with Partially Coherent Light Sources and Camera-in-the-loop Calibration

A PyTorch-based Semi-Supervised Learning (SSL) Codebase for Pixel-wise (Pixel) Vision Tasks