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

NTSFormer: A Self-Teaching Graph Transformer for Multimodal Isolated Cold-Start Node Classification

About

Isolated cold-start node classification on multimodal graphs is challenging because such nodes have no edges and often have missing modalities (e.g., absent text or image features). Existing methods address structural isolation by degrading graph learning models to multilayer perceptrons (MLPs) for isolated cold-start inference, using a teacher model (with graph access) to guide the MLP. However, this results in limited model capacity in the student, which is further challenged when modalities are missing. In this paper, we propose Neighbor-to-Self Graph Transformer (NTSFormer), a unified Graph Transformer framework that jointly tackles the isolation and missing-modality issues via a self-teaching paradigm. Specifically, NTSFormer uses a cold-start attention mask to simultaneously make two predictions for each node: a "student" prediction based only on self information (i.e., the node's own features), and a "teacher" prediction incorporating both self and neighbor information. This enables the model to supervise itself without degrading to an MLP, thereby fully leveraging the Transformer's capacity to handle missing modalities. To handle diverse graph information and missing modalities, NTSFormer performs a one-time multimodal graph pre-computation that converts structural and feature data into token sequences, which are then processed by Mixture-of-Experts (MoE) Input Projection and Transformer layers for effective fusion. Experiments on public datasets show that NTSFormer achieves superior performance for multimodal isolated cold-start node classification.

Jun Hu, Yufei He, Yuan Li, Bryan Hooi, Bingsheng He• 2025

Related benchmarks

TaskDatasetResultRank
Node ClassificationMovies
Accuracy56.37
47
Modal RetrievalEle-fashion
MRR92.88
31
Node ClusteringRedditS
NMI85.81
31
Link PredictionBili Dance
MRR38.73
27
Node ClassificationGrocery
Accuracy81.85
21
Node ClusteringToys
NMI49.42
17
G2TextFlickr30K
BLEU-48.41
17
G2ImageSemArt
CLIP Similarity (CLIP-S)62.88
17
Link PredictionDY
MRR72.15
17
Node ClassificationGoodreads
Accuracy71.19
14
Showing 10 of 14 rows

Other info

Follow for update