Turning Tabular Foundation Models into Graph Foundation Models
By: Dmitry Eremeev , Gleb Bazhenov , Oleg Platonov and more
Potential Business Impact:
Helps computers understand complex networks better.
While foundation models have revolutionized such fields as natural language processing and computer vision, their application and potential within graph machine learning remain largely unexplored. One of the key challenges in designing graph foundation models (GFMs) is handling diverse node features that can vary across different graph datasets. Although many works on GFMs have been focused exclusively on text-attributed graphs, the problem of handling arbitrary features of other types in GFMs has not been fully addressed. However, this problem is not unique to the graph domain, as it also arises in the field of machine learning for tabular data. In this work, motivated by the recent success of tabular foundation models like TabPFNv2, we propose G2T-FM, a simple graph foundation model that employs TabPFNv2 as a backbone. Specifically, G2T-FM augments the original node features with neighborhood feature aggregation, adds structural embeddings, and then applies TabPFNv2 to the constructed node representations. Even in a fully in-context regime, our model achieves strong results, significantly outperforming publicly available GFMs and performing on par with well-tuned GNNs trained from scratch. Moreover, after finetuning, G2T-FM surpasses well-tuned GNN baselines, highlighting the potential of the proposed approach. More broadly, our paper reveals a previously overlooked direction of utilizing tabular foundation models for graph machine learning tasks.
Similar Papers
Of Graphs and Tables: Zero-Shot Node Classification with Tabular Foundation Models
Machine Learning (CS)
Makes computers understand networks like tables.
Robust Tabular Foundation Models
Machine Learning (CS)
Makes AI better at learning from data.
Generalization Can Emerge in Tabular Foundation Models From a Single Table
Machine Learning (CS)
Teaches computers to learn from just one example.