LGAIAug 20, 2024

Do Neural Scaling Laws Exist on Graph Self-Supervised Learning?

arXiv:2408.11243v24 citationsh-index: 16Has Code
AI Analysis

This work addresses the feasibility of building Graph Foundation Models (GFMs) by identifying a key limitation in current graph SSL methods, which is incremental as it benchmarks existing techniques without proposing a new solution.

The study investigated whether existing graph self-supervised learning (SSL) techniques exhibit neural scaling laws, where performance improves with increased model and data sizes, as seen in NLP and CV domains, and found that no techniques showed such scaling behavior, with performance only fluctuating across scales.

Self-supervised learning~(SSL) is essential to obtain foundation models in NLP and CV domains via effectively leveraging knowledge in large-scale unlabeled data. The reason for its success is that a suitable SSL design can help the model to follow the neural scaling law, i.e., the performance consistently improves with increasing model and dataset sizes. However, it remains a mystery whether existing SSL in the graph domain can follow the scaling behavior toward building Graph Foundation Models~(GFMs) with large-scale pre-training. In this study, we examine whether existing graph SSL techniques can follow the neural scaling behavior with the potential to serve as the essential component for GFMs. Our benchmark includes comprehensive SSL technique implementations with analysis conducted on both the conventional SSL setting and many new settings adopted in other domains. Surprisingly, despite the SSL loss continuously decreasing, no existing graph SSL techniques follow the neural scaling behavior on the downstream performance. The model performance only merely fluctuates on different data scales and model scales. Instead of the scales, the key factors influencing the performance are the choices of model architecture and pretext task design. This paper examines existing SSL techniques for the feasibility of Graph SSL techniques in developing GFMs and opens a new direction for graph SSL design with the new evaluation prototype. Our code implementation is available online to ease reproducibility on https://github.com/GraphSSLScaling/GraphSSLScaling.

Foundations

The foundational work for this paper's niche, ranked by how specifically the neighbourhood builds on it — not by global fame.

Your Notes