Poster
in
Workshop: ICML 2024 Workshop on Foundation Models in the Wild
Zero-Shot Generalization of GNNs over Distinct Attribute Domains
Yangyi Shen · Beatrice Bevilacqua · Joshua Robinson · Charilaos Kanatsoulis · Jure Leskovec · Bruno Ribeiro
Keywords: [ Zero-shot ] [ attributed graphs ] [ Graph Neural Networks ] [ graph foundation models ]
There are no known graph machine learning methods that can zero-shot generalize across attributed graphs with different node attribute domains. For instance, no method can perform zero-shot link prediction by pretraining on online appliance store datasets (with node attributes such as brand, model, capacity, dimension, has ice maker, energy rating for refrigerators) and zero-shot at test on an electronics store dataset for smartphones (with attributes such as processor type, display type, storage, and battery capacity). In this work, we leverage concepts in statistical theory to design STAGE, a universally applicable approach for encoding node attributes in any GNN that facilitates such generalization. Empirically, we show that STAGE outperforms its natural baselines and can accurately make predictions when presented with completely new feature domains.