Official implementation of NeurIPS'21: Implicit SVD for Graph Representation Learning

Last update: Jan 08, 2023

Related tags

Overview

isvd

Official implementation of NeurIPS'21: Implicit SVD for Graph Representation Learning

If you find this code useful, you may cite us as:

@inproceedings{haija2021isvd,
  author={Sami Abu-El-Haija AND Hesham Mostafa AND Marcel Nassar AND Valentino Crespi AND Greg Ver Steeg AND Aram Galstyan},
  title={Implicit SVD for Graph Representation Learning},
  booktitle={Advances in Neural Information Processing Systems},
  year={2021},
}

To run link prediction on Stanford SNAP and node2vec datasets:

To embed with rank-32 SVD:

python3 run_snap_linkpred.py --dataset_name=ppi --dim=32
python3 run_snap_linkpred.py --dataset_name=ca-AstroPh --dim=32
python3 run_snap_linkpred.py --dataset_name=ca-HepTh --dim=32
python3 run_snap_linkpred.py --dataset_name=soc-facebook --dim=32

To embed with rank 256 on half of the training edges, determine "best rank" based on the remaining half, then re-run sVD with the best rank on all of training: (note: negative dim causes this logic):

python3 run_snap_linkpred.py --dataset_name=ppi --dim=-256
python3 run_snap_linkpred.py --dataset_name=ca-AstroPh --dim=-256
python3 run_snap_linkpred.py --dataset_name=ca-HepTh --dim=-256
python3 run_snap_linkpred.py --dataset_name=soc-facebook --dim=-256

To run semi-supervised node classification on Planetoid datasets

You must first download the planetoid dataset as:

mkdir -p ~/data
cd ~/data
git clone [email protected]:kimiyoung/planetoid.git

Afterwards, you may navigate back to this directory and run our code as:

python3 run_planetoid.py --dataset=ind.citeseer
python3 run_planetoid.py --dataset=ind.cora
python3 run_planetoid.py --dataset=ind.pubmed

To run link prediction on Stanford OGB DDI

python3 ogb_linkpred_sing_val_net.py

Note the above will download the dataset from Stanford. If you already have it, you may symlink it into directory dataset

To run link prediction on Stanford OGB ArXiv

As our code imports gttf, you must first clone it onto the repo:

git clone [email protected]:isi-usc-edu/gttf.git

Afterwards, you may run as:

python3 final_obgn_mixed_device.py --funetune_device='gpu:0'

Note the above will download the dataset from Stanford. If you already have it, you may symlink it into directory dataset. You may skip the finetune_device argument if you do not have a GPU installed.

Official implementation of NeurIPS'21: Implicit SVD for Graph Representation Learning

Related tags

Overview

isvd

To run link prediction on Stanford SNAP and node2vec datasets:

To run semi-supervised node classification on Planetoid datasets

To run link prediction on Stanford OGB DDI

To run link prediction on Stanford OGB ArXiv

Owner

Sami Abu-El-Haija

CNN Based Meta-Learning for Noisy Image Classification and Template Matching

Code for Fully Context-Aware Image Inpainting with a Learned Semantic Pyramid

The aim of this project is to build an AI bot that can play the Wordle game, or more generally Squabble

Keras implementation of "One pixel attack for fooling deep neural networks" using differential evolution on Cifar10 and ImageNet

retweet 4 satoshi ⚡️

Dense Contrastive Learning (DenseCL) for self-supervised representation learning, CVPR 2021.

Iris prediction model is used to classify iris species created julia's DecisionTree, DataFrames, JLD2, PlotlyJS and Statistics packages.

Code to accompany the paper "Finding Bipartite Components in Hypergraphs", which is published in NeurIPS'21.

Code for "Continuous-Time Meta-Learning with Forward Mode Differentiation" (ICLR 2022)

NVIDIA Deep Learning Examples for Tensor Cores

a general-purpose Transformer based vision backbone

[ICCV'21] Official implementation for the paper Social NCE: Contrastive Learning of Socially-aware Motion Representations

Official Chainer implementation of GP-GAN: Towards Realistic High-Resolution Image Blending (ACMMM 2019, oral)

Group Fisher Pruning for Practical Network Compression(ICML2021)

Summary Explorer is a tool to visually explore the state-of-the-art in text summarization.

Official implementation of CATs: Cost Aggregation Transformers for Visual Correspondence NeurIPS'21

This program creates a formatted excel file which highlights the undervalued stock according to Graham's number.

Angora is a mutation-based fuzzer. The main goal of Angora is to increase branch coverage by solving path constraints without symbolic execution.

Active Offline Policy Selection With Python

This project uses ViT to perform image classification tasks on DATA set CIFAR10.