Geometric Deep Learning Extension Library for PyTorch

PyTorch Geometric is a geometric deep learning extension library for PyTorch.

It consists of various methods for deep learning on graphs and other irregular structures, also known as geometric deep learning, from a variety of published papers. In addition, it consists of an easy-to-use mini-batch loader, a large number of common benchmark datasets (based on simple interfaces to create your own), and helpful transforms, both for learning on arbitrary graphs as well as on 3D meshes or point clouds.

To install Pytorch Geometric go to Instalation Guide

PyTorch Geometric through self-contained examples. At its core, PyTorch Geometric provides the following main features:

• Data handling of graphs
• Common benchmark datasets
• Mini-Batches
• Data transforms
• Learning methods on graphs

In detail, the following methods are currently implemented:

• SplineConv from Fey et al.: SplineCNN: Fast Geometric Deep Learning with Continuous B-Spline Kernels (CVPR 2018)
• GCNConv from Kipf and Welling: Semi-Supervised Classification with Graph Convolutional Networks (ICLR 2017)
• ChebConv from Defferrard et al.: Convolutional Neural Networks on Graphs with Fast Localized Spectral Filtering (NIPS 2017)
• NNConv adapted from Gilmer et al.: Neural Message Passing for Quantum Chemistry (ICML 2017)
• GATConv from Veličković et al.: Graph Attention Networks (ICLR 2018)
• AGNNProp from Kiran et al.: Attention-based Graph Neural Network for Semi-Supervised Learning
• SAGEConv from Hamilton et al.: Inductive Representation Learning on Large Graphs (NIPS 2017)
• Graclus Pooling from Dhillon et al.: Weighted Graph Cuts without Eigenvectors: A Multilevel Approach (PAMI 2007)
• Voxel Grid Pooling