vix.ing · top · new · best · stats

Fast Graph Representation Learning with PyTorch Geometric

2019/03/06 by Matthias Fey, Jan Eric Lenssen, Fey, Matthias +1 · 1,260 citations
Computer Science · Mathematics · #Advanced Graph Neural Networks #Artificial intelligence #CUDA #Computational science #Computer science #Data Visualization and Analytics #Deep learning #Graph #Graph Theory and Algorithms #Parallel computing #Point cloud #Theoretical computer science #cs.LG #stat.ML

paper · pdf · doi:10.48550/arxiv.1903.02428

published in arXiv (Cornell University) (Cornell University) · ICLR 2019 (RLGM Workshop)

openalex publication_date 2019/03/06 · openalex created_date 2019/03/11 · arxiv created 2019/04/25 · arxiv updated 2019/04/26 · openalex updated_date 2026/07/28

Abstract

We introduce PyTorch Geometric, a library for deep learning on irregularly structured input data such as graphs, point clouds and manifolds, built upon PyTorch. In addition to general graph data structures and processing methods, it contains a variety of recently published methods from the domains of relational learning and 3D data processing. PyTorch Geometric achieves high data throughput by leveraging sparse GPU acceleration, by providing dedicated CUDA kernels and by introducing efficient mini-batch handling for input examples of different size. In this work, we present the library in detail and perform a comprehensive comparative study of the implemented methods in homogeneous evaluation scenarios.

Citations

Cited by

Related