PyG-SSL: A Graph Self-Supervised Learning Toolkit
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
- 2020A Simple Framework for Contrastive Learning of Visual Representationscited by 11,979
- 2016Semi-Supervised Classification with Graph Convolutional Networkscited by 11,282
- 2017Graph Attention Networkscited by 8,854
- 2018Representation Learning with Contrastive Predictive Codingcited by 7,662
- 2020Graph Contrastive Learning with Augmentationscited by 1,203
- 2018Deep Graph Infomaxcited by 1,101
- 2019Strategies for Pre-training Graph Neural Networkscited by 964
- 2015Image-based Recommendations on Styles and Substitutescited by 702
- 2020Contrastive Multi-View Representation Learning on Graphscited by 651
- 2020Graph Contrastive Learning with Adaptive Augmentationcited by 519
- 2020GCC: Graph Contrastive Coding for Graph Neural Network Pre-Trainingcited by 461
- 2022GraphMAE: Self-Supervised Masked Graph Autoencoderscited by 443
38 more
- 2019InfoGraph: Unsupervised and Semi-supervised Graph-Level Representation Learning via Mutual Information Maximizationcited by 441
- 2016Learning Convolutional Neural Networks for Graphscited by 314
- 2021Graph Self-Supervised Learning: A Surveycited by 289
- 2021Graph Contrastive Learning Automatedcited by 287
- 2020Wiki-CS: A Wikipedia-Based Benchmark for Graph Neural Networkscited by 253
- 2021Large-Scale Representation Learning on Graphs via Bootstrappingcited by 184
- 2021Self-Supervised Learning of Graph Neural Networks: A Unified Reviewcited by 184
- 2021Are Graph Augmentations Necessary? Simple Graph Contrastive Learning for Recommendationcited by 173
- 2021An Empirical Study of Graph Contrastive Learningcited by 113
- 2019Unsupervised Attributed Multiplex Network Embeddingcited by 95
- 2020Model-Agnostic Counterfactual Reasoning for Eliminating Popularity Bias in Recommender Systemcited by 94
- 2021Augmentation-Free Self-Supervised Learning on Graphscited by 94
- 2021HDMI: High-order Deep Multiplex Infomaxcited by 82
- 2021Multi-Scale Contrastive Siamese Networks for Self-Supervised Graph Representation Learningcited by 70
- 2019Learning metrics for persistence-based summaries and applications for graph classificationcited by 57
- 2022Adversarial Graph Contrastive Learning with Information Regularizationcited by 38
- 2021Graph Communal Contrastive Learningcited by 34
- 2024VCR-Graphormer: A Mini-batch Graph Transformer via Virtual Connectionscited by 34
- 2021Network of Tensor Time Seriescited by 32
- 2022Augmentations in Hypergraph Contrastive Learning: Fabricated and Generativecited by 28
- 2023Ensuring User-side Fairness in Dynamic Recommender Systemscited by 27
- 2024Heterogeneous Contrastive Learning for Foundation Models and Beyondcited by 27
- 2024Towards Unified Multi-Modal Personalization: Large Vision-Language Models for Generative Recommendation and Beyondcited by 25
- 2024LLM-Forest: Ensemble Learning of LLMs with Graph-Augmented Prompts for Data Imputationcited by 25
- 2023Hierarchical Multi-Marginal Optimal Transport for Network Alignmentcited by 24
- 2020KompaRe: A Knowledge Graph Comparative Reasoning Systemcited by 20
- 2021Multiplex Graph Neural Network for Extractive Text Summarizationcited by 19
- 2021X-GOAL: Multiplex Heterogeneous Graph Prototypical Contrastive Learningcited by 19
- 2022Comprehensive Fair Meta-learned Recommender Systemcited by 18
- 2022MentorGNN: Deriving Curriculum for Pre-Training GNNscited by 18
- 2021Heterogeneous Contrastive Learningcited by 15
- 2022COIN: Co-Cluster Infomax for Bipartite Graphscited by 14
- 2024PageRank Bandits for Link Predictioncited by 14
- 2022Retrieval Based Time Series Forecastingcited by 13
- 2024Can Graph Neural Networks Learn Language with Extremely Weak Text Supervision?cited by 13
- 2024APEX2: Adaptive and Extreme Summarization for Personalized Knowledge Graphscited by 13
- 2023STERLING: Synergistic Representation Learning on Bipartite Graphscited by 12
- 2023FairGen: Towards Fair Graph Generationcited by 12
50 most cited of 62 All 62 (JSON)
Built on by
Citing papers per year: 2025: 11; 2026: 3.
14 papers.
Sort:
By area:
- 2025SelfElicit: Your Language Model Secretly Knows Where is the Relevant Evidencecited by 18
- 2025Pave Your Own Path: Graph Gradual Domain Adaptation on Fused Gromov-Wasserstein Geodesicscited by 17
- 2025Graph4MM: Weaving Multimodal Learning with Structural Informationcited by 17
- 2025Flow Matching Meets Biology and Life Science: A Surveycited by 14
- 2025Adaptive Fault Tolerance Mechanisms of Large Language Models in Cloud Computing Environmentscited by 12
- 2025Breaking Silos: Adaptive Model Fusion Unlocks Better Time Series Forecastingcited by 12
- 2025PLANETALIGN: A Comprehensive Python Library for Benchmarking Network Alignmentcited by 10
- 2025Hierarchical LoRA MoE for Efficient CTR Model Scalingcited by 10
- 2025Invariant Link Selector for Spatial-Temporal Out-of-Distribution Problemcited by 5
- 2025ClimateBench-M: A Multi-Modal Climate Data Benchmark with a Simple Generative Methodcited by 4
- 2026Mixture of Sequence: Theme-Aware Mixture-of-Experts for Long-Sequence Recommendationcited by 4
- 2025Learnable Spatial-Temporal Positional Encoding for Link Predictioncited by 3
2 more
- 2026SafeImpute: Reliable Clinical Data Imputation via Conformal Selection6 Jul 2026
- 2026Generalist Graph Anomaly Detection via Prototype-Based Distillation26 May 2026
- 2026Mixture of Sequence: Theme-Aware Mixture-of-Experts for Long-Sequence Recommendation1 Mar 2026
- 2025Graph4MM: Weaving Multimodal Learning with Structural Information19 Oct 2025
- 2025Hierarchical LoRA MoE for Efficient CTR Model Scaling12 Oct 2025
- 2025Flow Matching Meets Biology and Life Science: A Survey23 Jul 2025
- 2025Learnable Spatial-Temporal Positional Encoding for Link Prediction10 Jun 2025
- 2025Invariant Link Selector for Spatial-Temporal Out-of-Distribution Problem30 May 2025
- 2025PLANETALIGN: A Comprehensive Python Library for Benchmarking Network Alignment27 May 2025
- 2025Breaking Silos: Adaptive Model Fusion Unlocks Better Time Series Forecasting24 May 2025
- 2025Pave Your Own Path: Graph Gradual Domain Adaptation on Fused Gromov-Wasserstein Geodesics19 May 2025
- 2025ClimateBench-M: A Multi-Modal Climate Data Benchmark with a Simple Generative Method10 Apr 2025
- 2025Flow Matching Meets Biology and Life Science: A Surveycited by 14
- 2025Flow Matching Meets Biology and Life Science: A Survey23 Jul 2025
- 2025SelfElicit: Your Language Model Secretly Knows Where is the Relevant Evidencecited by 18
- 2025Pave Your Own Path: Graph Gradual Domain Adaptation on Fused Gromov-Wasserstein Geodesicscited by 17
- 2025Graph4MM: Weaving Multimodal Learning with Structural Informationcited by 17
- 2025Adaptive Fault Tolerance Mechanisms of Large Language Models in Cloud Computing Environmentscited by 12
- 2025Breaking Silos: Adaptive Model Fusion Unlocks Better Time Series Forecastingcited by 12
- 2025PLANETALIGN: A Comprehensive Python Library for Benchmarking Network Alignmentcited by 10
- 2025Invariant Link Selector for Spatial-Temporal Out-of-Distribution Problemcited by 5
- 2025ClimateBench-M: A Multi-Modal Climate Data Benchmark with a Simple Generative Methodcited by 4
- 2025Learnable Spatial-Temporal Positional Encoding for Link Predictioncited by 3
- 2026Generalist Graph Anomaly Detection via Prototype-Based Distillationcited by 0
- 2026SafeImpute: Reliable Clinical Data Imputation via Conformal Selectioncited by 0
- 2026SafeImpute: Reliable Clinical Data Imputation via Conformal Selection6 Jul 2026
- 2026Generalist Graph Anomaly Detection via Prototype-Based Distillation26 May 2026
- 2025Graph4MM: Weaving Multimodal Learning with Structural Information19 Oct 2025
- 2025Learnable Spatial-Temporal Positional Encoding for Link Prediction10 Jun 2025
- 2025Invariant Link Selector for Spatial-Temporal Out-of-Distribution Problem30 May 2025
- 2025PLANETALIGN: A Comprehensive Python Library for Benchmarking Network Alignment27 May 2025
- 2025Breaking Silos: Adaptive Model Fusion Unlocks Better Time Series Forecasting24 May 2025
- 2025Pave Your Own Path: Graph Gradual Domain Adaptation on Fused Gromov-Wasserstein Geodesics19 May 2025
- 2025ClimateBench-M: A Multi-Modal Climate Data Benchmark with a Simple Generative Method10 Apr 2025
- 2025Adaptive Fault Tolerance Mechanisms of Large Language Models in Cloud Computing Environments15 Mar 2025
- 2025SelfElicit: Your Language Model Secretly Knows Where is the Relevant Evidence12 Feb 2025