PyG-SSL简化图自监督学习,让新手也能快速复现主流算法。
PyG-SSL: A Graph Self-Supervised Learning Toolkit
- 基于PyTorch构建统一框架,整合数据加载与训练流程
- 提供10+主流图SSL算法在多个数据集上的最佳参数配置
- 配套教程和标准化评估,提升结果可复现性
图自监督学习(Graph SSL)近年来成为研究热点。通过无标签数据中的预训练任务,模型能有效学习图的拓扑结构与属性,从而提升性能、泛化能力与鲁棒性。然而现有方法实现复杂,评估指标不一,复现困难,阻碍了该领域的发展。为此,我们推出基于PyTorch的图自监督学习工具包PyG-SSL,兼容多种深度学习后端。该工具包提供统一的数据加载、超参配置、模型训练与下游任务评估流程,并配备针对不同图数据集的主流图SSL算法的最佳参数及入门教程,显著降低使用门槛,促进研究复现与推广。项目主页:https://github.com/iDEA-iSAIL-Lab-UIUC/pyg-ssl。
原文摘要 · Abstract (English)
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.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。