Rethinking Tokenized Graph Transformers for Node Classification

Fuente: arXiv
Salvato in:
Dettagli Bibliografici
Autori principali: Chen, Jinsong, Li, Chenyang, Li, GaiChao, Hopcroft, John E., He, Kun
Natura: Preprint
Pubblicazione: 2025
Soggetti:
Accesso online:
Tags: Aggiungi Tag
Nessun Tag, puoi essere il primo ad aggiungerne!!
_version_ 1866912230358908928
author Chen, Jinsong
Li, Chenyang
Li, GaiChao
Hopcroft, John E.
He, Kun
author_facet Chen, Jinsong
Li, Chenyang
Li, GaiChao
Hopcroft, John E.
He, Kun
contents Node tokenized graph Transformers (GTs) have shown promising performance in node classification. The generation of token sequences is the key module in existing tokenized GTs which transforms the input graph into token sequences, facilitating the node representation learning via Transformer. In this paper, we observe that the generations of token sequences in existing GTs only focus on the first-order neighbors on the constructed similarity graphs, which leads to the limited usage of nodes to generate diverse token sequences, further restricting the potential of tokenized GTs for node classification. To this end, we propose a new method termed SwapGT. SwapGT first introduces a novel token swapping operation based on the characteristics of token sequences that fully leverages the semantic relevance of nodes to generate more informative token sequences. Then, SwapGT leverages a Transformer-based backbone to learn node representations from the generated token sequences. Moreover, SwapGT develops a center alignment loss to constrain the representation learning from multiple token sequences, further enhancing the model performance. Extensive empirical results on various datasets showcase the superiority of SwapGT for node classification.
format Preprint
id arxiv_https___arxiv_org_abs_2502_08101
institution arXiv
publishDate 2025
record_format arxiv
spellingShingle Rethinking Tokenized Graph Transformers for Node Classification
Chen, Jinsong
Li, Chenyang
Li, GaiChao
Hopcroft, John E.
He, Kun
Machine Learning
Artificial Intelligence
Node tokenized graph Transformers (GTs) have shown promising performance in node classification. The generation of token sequences is the key module in existing tokenized GTs which transforms the input graph into token sequences, facilitating the node representation learning via Transformer. In this paper, we observe that the generations of token sequences in existing GTs only focus on the first-order neighbors on the constructed similarity graphs, which leads to the limited usage of nodes to generate diverse token sequences, further restricting the potential of tokenized GTs for node classification. To this end, we propose a new method termed SwapGT. SwapGT first introduces a novel token swapping operation based on the characteristics of token sequences that fully leverages the semantic relevance of nodes to generate more informative token sequences. Then, SwapGT leverages a Transformer-based backbone to learn node representations from the generated token sequences. Moreover, SwapGT develops a center alignment loss to constrain the representation learning from multiple token sequences, further enhancing the model performance. Extensive empirical results on various datasets showcase the superiority of SwapGT for node classification.
title Rethinking Tokenized Graph Transformers for Node Classification
topic Machine Learning
Artificial Intelligence
url https://arxiv.org/abs/2502.08101