Sampling Transferable Graph Neural Networks with Limited Graph Information

TL;DR

This paper proposes feature-driven subgraph sampling methods to improve the transferability of GNNs when graph structure is noisy or incomplete, by preserving spectral properties through feature and spectral alignment.

Contribution

It introduces a novel feature-based sampling approach that maintains spectral properties of graph operators, enhancing GNN transferability without relying on precise graph connectivity.

Findings

Feature homophily measure links feature statistics to spectral properties.

Trace-based sampling algorithm improves GNN transferability on benchmarks.

Maximizing Laplacian trace yields better generalization in transfer learning.

Abstract

Graph neural networks (GNNs) achieve strong performance on graph learning tasks, but training on large-scale networks remains computationally challenging. Transferability results show that GNNs with fixed weights can generalize from smaller graphs to larger ones drawn from the same family, motivating the use of sampled subgraphs to boost training efficiency. Yet most existing sampling strategies rely on reliable access to the target graph structure, which in practice may be noisy, incomplete, or unavailable prior to training. In lieu of precise connectivity information, we study feature-driven subgraph sampling for transferable GNNs, with the goal of preserving spectral properties of graph operators that control GNN expressivity. We adopt an alignment-based perspective linking node feature statistics to graph spectral structure and develop two complementary notions of feature-graph alignment. For coarse alignment, we formalize feature homophily through a Laplacian-based measure quantifying the alignment of feature principal components with graph eigenvectors, and establish a lower bound on the Laplacian trace in terms of feature statistics. This motivates a simple, non-sequential sampling algorithm that operates on the feature matrix and preserves a trace-based proxy for operator rank. For fine alignment, we assume a stationary model where the feature covariance and Laplacian share an eigenbasis, and establish that diagonal covariance entries reflect node-degree ordering under monotone filters. We empirically validate that filter monotonicity dictates the relationship between feature variance and spectral energy. On real-world benchmarks, selecting the retention rule that maximizes the Laplacian trace consistently yields GNNs with superior transferability and reduced generalization gaps.