Wiki-CS: A Wikipedia-Based Benchmark for Graph Neural Networks

Péter Mernyei; Cătălina Cangea

Wiki-CS: A Wikipedia-Based Benchmark for Graph Neural Networks

Machine Learning 2022-01-11 v2 Social and Information Networks Machine Learning

Authors: Péter Mernyei , Cătălina Cangea

Abstract

We present Wiki-CS, a novel dataset derived from Wikipedia for benchmarking Graph Neural Networks. The dataset consists of nodes corresponding to Computer Science articles, with edges based on hyperlinks and 10 classes representing different branches of the field. We use the dataset to evaluate semi-supervised node classification and single-relation link prediction models. Our experiments show that these methods perform well on a new domain, with structural properties different from earlier benchmarks. The dataset is publicly available, along with the implementation of the data pipeline and the benchmark experiments, at https://github.com/pmernyei/wiki-cs-dataset .

Keywords

graph neural network graph theory image classification

Cite

@article{arxiv.2007.02901,
  title  = {Wiki-CS: A Wikipedia-Based Benchmark for Graph Neural Networks},
  author = {Péter Mernyei and Cătălina Cangea},
  journal= {arXiv preprint arXiv:2007.02901},
  year   = {2022}
}

Comments

Graph Representation Learning and Beyond workshop (ICML 2020); corrected incorrect metrics due to issue in experiment implementation

Wiki-CS: A Wikipedia-Based Benchmark for Graph Neural Networks

Abstract

Keywords

Cite

Comments

Related papers