Share your thoughts, 1 month free Claude Pro on usSee more
WorkDL logo mark

Turning Tabular Foundation Models into Graph Foundation Models

About

While foundation models have revolutionized fields such as natural language processing and computer vision, their potential in graph machine learning remains 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. While many works on GFMs have 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 (TFMs) like TabPFNv2 and LimiX, we propose G2T-FM, a simple framework that allows tabular foundation models to be applied to graph node-level tasks. Specifically, G2T-FM augments the original node features with neighborhood feature aggregation, adds structural embeddings, and then applies a TFM to the constructed node representations. Even in the in-context learning setting, our model achieves strong results when combined with a strong TFM, outperforming both prior GFMs and well-tuned GNNs trained from scratch. Moreover, after finetuning, G2T-FM consistently surpasses well-tuned GNN baselines, often by a significant margin. In summary, our paper reveals the potential of a previously overlooked direction: utilizing tabular foundation models for graph machine learning tasks.

Dmitry Eremeev, Gleb Bazhenov, Oleg Platonov, Artem Babenko, Liudmila Prokhorenkova• 2025

Related benchmarks

TaskDatasetResultRank
Node ClassificationPubmed
Accuracy90.94
902
Node Classificationamazon-ratings
Accuracy46.09
354
Node Classificationquestions
Accuracy22.98
79
Node-level classificationFacebook
Accuracy92.83
65
Binary Classificationtolokers 2 (RL (Random Low))
AP61.58
21
Binary Classificationcity-reviews (RL)
Average Precision80.15
21
Aggregate RankingGraphLand Suite Aggregate (RL (Random Low))
AR (cls)5
19
Binary Classificationartnet-exp (RL Random Low)
AP0.4996
19
Node Classificationamazon-ratings 10%/10%/80%
Average Precision46.05
18
Node Classificationpubmed 10%/10%/80% split
Accuracy90.88
18
Showing 10 of 29 rows

Other info

Follow for update