We gratefully acknowledge support from
the Simons Foundation and member institutions.
Full-text links:

Download:

Current browse context:

cs.SI

Change to browse by:

References & Citations

DBLP - CS Bibliography

Bookmark

(what is this?)
CiteULike logo BibSonomy logo Mendeley logo del.icio.us logo Digg logo Reddit logo

Computer Science > Social and Information Networks

Title: CogDL: An Extensive Toolkit for Deep Learning on Graphs

Abstract: Graph representation learning aims to learn low-dimensional node embeddings for graphs. It is used in several real-world applications such as social network analysis and large-scale recommender systems. In this paper, we introduce CogDL, an extensive research toolkit for deep learning on graphs that allows researchers and developers to easily conduct experiments and build applications. It provides standard training and evaluation for the most important tasks in the graph domain, including node classification, link prediction, graph classification, and other graph tasks. For each task, it offers implementations of state-of-the-art models. The models in our toolkit are divided into two major parts, graph embedding methods and graph neural networks. Most of the graph embedding methods learn node-level or graph-level representations in an unsupervised way and preserves the graph properties such as structural information, while graph neural networks capture node features and work in semi-supervised or self-supervised settings. All models implemented in our toolkit can be easily reproducible for leaderboard results. Most models in CogDL are developed on top of PyTorch, and users can leverage the advantages of PyTorch to implement their own models. Furthermore, we demonstrate the effectiveness of CogDL for real-world applications in AMiner, which is a large academic database and system.
Subjects: Social and Information Networks (cs.SI); Machine Learning (cs.LG); Machine Learning (stat.ML)
Cite as: arXiv:2103.00959 [cs.SI]
  (or arXiv:2103.00959v1 [cs.SI] for this version)

Submission history

From: Yukuo Cen [view email]
[v1] Mon, 1 Mar 2021 12:35:16 GMT (335kb,D)
[v2] Fri, 23 Jul 2021 03:29:00 GMT (280kb,D)
[v3] Thu, 3 Mar 2022 13:23:37 GMT (4639kb,D)
[v4] Mon, 17 Apr 2023 10:44:27 GMT (593kb,D)

Link back to: arXiv, form interface, contact.