CogDL: A Comprehensive Library for Graph Deep Learning
arXiv:2103.00959
Abstract
Graph neural networks (GNNs) have attracted tremendous attention from the graph learning community in recent years. It has been widely adopted in various real-world applications from diverse domains, such as social networks and biological graphs. The research and applications of graph deep learning present new challenges, including the sparse nature of graph data, complicated training of GNNs, and non-standard evaluation of graph tasks. To tackle the issues, we present CogDL, a comprehensive library for graph deep learning that allows researchers and practitioners to conduct experiments, compare methods, and build applications with ease and efficiency. In CogDL, we propose a unified design for the training and evaluation of GNN models for various graph tasks, making it unique among existing graph learning libraries. By utilizing this unified trainer, CogDL can optimize the GNN training loop with several training techniques, such as mixed precision training. Moreover, we develop efficient sparse operators for CogDL, enabling it to become the most competitive graph library for efficiency. Another important CogDL feature is its focus on ease of use with the aim of facilitating open and reproducible research of graph learning. We leverage CogDL to report and maintain benchmark results on fundamental graph tasks, which can be reproduced and directly used by the community.
arXiv commentAccepted to WWW 2023. Website: https://github.com/THUDM/cogdl
References in corpus (20)
- Efficient Estimation of Word Representations in Vector Space
- PyTorch: An Imperative Style, High-Performance Deep Learning Library
- Inductive Representation Learning on Large Graphs
- Fast Graph Representation Learning with PyTorch Geometric
- Complex Embeddings for Simple Link Prediction
- Representation Learning on Graphs with Jumping Knowledge Networks
- RotatE: Knowledge Graph Embedding by Relational Rotation in Complex Space
- Deep Graph Library: A Graph-Centric, Highly-Performant Package for Graph Neural Networks
- DropEdge: Towards Deep Graph Convolutional Networks on Node Classification
- Pitfalls of Graph Neural Network Evaluation
- graph2vec: Learning Distributed Representations of Graphs
- Graph Transformer Networks
- Representation Learning for Attributed Multiplex Heterogeneous Network
- Simple and Deep Graph Convolutional Networks
- Contrastive Multi-View Representation Learning on Graphs
- Scaling Graph Neural Networks with Approximate PageRank
- Benchmarking Graph Neural Networks
- InfoGraph: Unsupervised and Semi-supervised Graph-Level Representation Learning via Mutual Information Maximization
- A Fair Comparison of Graph Neural Networks for Graph Classification
- GE-SpMM: General-purpose Sparse Matrix-Matrix Multiplication on GPUs for Graph Neural Networks