mlsanityData to 7 Oct 2026

PyG-SSL: A Graph Self-Supervised Learning Toolkit

arXiv 2412.21151 ↗First posted 30 Dec 2024cs.LG cs.AIcited by 14

Not on the map. Why?

Abstract (arXiv)

Graph Self-Supervised Learning (SSL) has emerged as a pivotal area of research in recent years. By engaging in pretext tasks to learn the intricate topological structures and properties of graphs using unlabeled data, these graph SSL models achieve enhanced performance, improved generalization, and heightened robustness. Despite the remarkable achievements of these graph SSL methods, their current implementation poses significant challenges for beginners and practitioners due to the complex nature of graph structures, inconsistent evaluation metrics, and concerns regarding reproducibility hinder further progress in this field. Recognizing the growing interest within the research community, there is an urgent need for a comprehensive, beginner-friendly, and accessible toolkit consisting of the most representative graph SSL algorithms. To address these challenges, we present a Graph SSL toolkit named PyG-SSL, which is built upon PyTorch and is compatible with various deep learning and scientific computing backends. Within the toolkit, we offer a unified framework encompassing dataset loading, hyper-parameter configuration, model training, and comprehensive performance evaluation for diverse downstream tasks. Moreover, we provide beginner-friendly tutorials and the best hyper-parameters of each graph SSL algorithm on different graph datasets, facilitating the reproduction of results. The GitHub repository of the library is https://github.com/iDEA-iSAIL-Lab-UIUC/pyg-ssl.

Key facts. Cited by 14 papers in the map (data through 7 Oct 2026). Builds on 62 of its 113 references that are in the map. First posted on arXiv 30 Dec 2024.

Builds on

62 of 113 references.

References · 62

  1. 2020A Simple Framework for Contrastive Learning of Visual Representationscited by 11,979
  2. 2016Semi-Supervised Classification with Graph Convolutional Networkscited by 11,282
  3. 2017Graph Attention Networkscited by 8,854
  4. 2018Representation Learning with Contrastive Predictive Codingcited by 7,662
  5. 2020Graph Contrastive Learning with Augmentationscited by 1,203
  6. 2018Deep Graph Infomaxcited by 1,101
  7. 2019Strategies for Pre-training Graph Neural Networkscited by 964
  8. 2015Image-based Recommendations on Styles and Substitutescited by 702
  9. 2020Contrastive Multi-View Representation Learning on Graphscited by 651
  10. 2020Graph Contrastive Learning with Adaptive Augmentationcited by 519
  11. 2020GCC: Graph Contrastive Coding for Graph Neural Network Pre-Trainingcited by 461
  12. 2022GraphMAE: Self-Supervised Masked Graph Autoencoderscited by 443
38 more
  1. 2019InfoGraph: Unsupervised and Semi-supervised Graph-Level Representation Learning via Mutual Information Maximizationcited by 441
  2. 2016Learning Convolutional Neural Networks for Graphscited by 314
  3. 2021Graph Self-Supervised Learning: A Surveycited by 289
  4. 2021Graph Contrastive Learning Automatedcited by 287
  5. 2020Wiki-CS: A Wikipedia-Based Benchmark for Graph Neural Networkscited by 253
  6. 2021Large-Scale Representation Learning on Graphs via Bootstrappingcited by 184
  7. 2021Self-Supervised Learning of Graph Neural Networks: A Unified Reviewcited by 184
  8. 2021Are Graph Augmentations Necessary? Simple Graph Contrastive Learning for Recommendationcited by 173
  9. 2021An Empirical Study of Graph Contrastive Learningcited by 113
  10. 2019Unsupervised Attributed Multiplex Network Embeddingcited by 95
  11. 2020Model-Agnostic Counterfactual Reasoning for Eliminating Popularity Bias in Recommender Systemcited by 94
  12. 2021Augmentation-Free Self-Supervised Learning on Graphscited by 94
  13. 2021HDMI: High-order Deep Multiplex Infomaxcited by 82
  14. 2021Multi-Scale Contrastive Siamese Networks for Self-Supervised Graph Representation Learningcited by 70
  15. 2019Learning metrics for persistence-based summaries and applications for graph classificationcited by 57
  16. 2022Adversarial Graph Contrastive Learning with Information Regularizationcited by 38
  17. 2021Graph Communal Contrastive Learningcited by 34
  18. 2024VCR-Graphormer: A Mini-batch Graph Transformer via Virtual Connectionscited by 34
  19. 2021Network of Tensor Time Seriescited by 32
  20. 2022Augmentations in Hypergraph Contrastive Learning: Fabricated and Generativecited by 28
  21. 2023Ensuring User-side Fairness in Dynamic Recommender Systemscited by 27
  22. 2024Heterogeneous Contrastive Learning for Foundation Models and Beyondcited by 27
  23. 2024Towards Unified Multi-Modal Personalization: Large Vision-Language Models for Generative Recommendation and Beyondcited by 25
  24. 2024LLM-Forest: Ensemble Learning of LLMs with Graph-Augmented Prompts for Data Imputationcited by 25
  25. 2023Hierarchical Multi-Marginal Optimal Transport for Network Alignmentcited by 24
  26. 2020KompaRe: A Knowledge Graph Comparative Reasoning Systemcited by 20
  27. 2021Multiplex Graph Neural Network for Extractive Text Summarizationcited by 19
  28. 2021X-GOAL: Multiplex Heterogeneous Graph Prototypical Contrastive Learningcited by 19
  29. 2022Comprehensive Fair Meta-learned Recommender Systemcited by 18
  30. 2022MentorGNN: Deriving Curriculum for Pre-Training GNNscited by 18
  31. 2021Heterogeneous Contrastive Learningcited by 15
  32. 2022COIN: Co-Cluster Infomax for Bipartite Graphscited by 14
  33. 2024PageRank Bandits for Link Predictioncited by 14
  34. 2022Retrieval Based Time Series Forecastingcited by 13
  35. 2024Can Graph Neural Networks Learn Language with Extremely Weak Text Supervision?cited by 13
  36. 2024APEX2: Adaptive and Extreme Summarization for Personalized Knowledge Graphscited by 13
  37. 2023STERLING: Synergistic Representation Learning on Bipartite Graphscited by 12
  38. 2023FairGen: Towards Fair Graph Generationcited by 12

50 most cited of 62 All 62 (JSON)

Built on by

Cited by year (2026 to 7 Oct)

Citing papers per year: 2025: 11; 2026: 3.

14 papers.

Sort:

By area: