Link Prediction on Large Graphs with Variational Autoencoders

The proposed method is a variational autoencoder, with graph convolutional layers in the encoder part.
Dataset statistics.
Link prediction results, with PR-AUC score as metric.

--

--

--

Deep Learning and AI solutions from Budapest University of Technology and Economics. http://smartlab.tmit.bme.hu/

Love podcasts or audiobooks? Learn on the go with our new app.

Recommended from Medium

Classification and Regression Problems in Machine Learning

Colab & CocaCola = Heaven

Skin in the game — Deep Learning based skin disease classifier (Part 1)

Harvesting’s Automated Credit Risk Scoring System for lenders serving rural agriculture…

8 Petuum Papers Accepted to NeurIPS 2018

Awesome Machine Learning Engineering Tools

How to Monitor an AI-based Application

Deploying a Model

Get the Medium app

A button that says 'Download on the App Store', and if clicked it will lead you to the iOS App store
A button that says 'Get it on, Google Play', and if clicked it will lead you to the Google Play store
SmartLab AI

SmartLab AI

Deep Learning and AI solutions from Budapest University of Technology and Economics. http://smartlab.tmit.bme.hu/

More from Medium

Pruning for Deep Neural Networks — Techniques to Prune Image and Language Models

Convolutional Autoencoders and image denoising.

pix2pix GAN for Generating Maps given Satellite Images using PyTorch

DistilBERT — distilled version of BERT