Fast Graph Representation Learning with PyTorch Geometric
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.
Code (6)
Tasks
GPUGraph ClassificationGraph Representation LearningNode ClassificationRelational ReasoningRepresentation LearningSimilar Papers 제목 키워드 기반
Tiered Graph Autoencoders with PyTorch Geometric for Molecular Graphs
Tiered latent representations and latent spaces for molecular graphs provide a simple but effective way to explicitly represent and utilize groups (e.g., functional groups), which consist of the atom (node) tier, the gro…
Transfer LearningPyTorch Geometric High Order: A Unified Library for High Order Graph Neural Network
We introduce PyTorch Geometric High Order (PyGHO), a library for High Order Graph Neural Networks (HOGNNs) that extends PyTorch Geometric (PyG). Unlike ordinary Message Passing Neural Networks (MPNNs) that exchange messa…
Graph Neural NetworkGNNAutoScale: Scalable and Expressive Graph Neural Networks via Historical Embeddings
We present GNNAutoScale (GAS), a framework for scaling arbitrary message-passing GNNs to large graphs. GAS prunes entire sub-trees of the computation graph by utilizing historical embeddings from prior training iteration…
GPUPyTorch Geometric Signed Directed: A Software Package on Graph Neural Networks for Signed and Directed Graphs
Networks are ubiquitous in many real-world applications (e.g., social networks encoding trust/distrust relationships, correlation networks arising from time series data). While many networks are signed or directed, or bo…
Time SeriesTime Series AnalysisNeurally boosted supervised spectral clustering
Network embedding methods compute geometric representations of graphs that render various prediction problems amenable to machine learning techniques. Spectral network embeddings are based on the computation of eigenve…
ClusteringNetwork EmbeddingNode Classification