paper-with-me

홈 › Papers

Rethinking Tokenized Graph Transformers for Node Classification

2025-02-12 · Jinsong Chen, Chenyang Li, Gaichao Li, John E. Hopcroft, Kun He

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.

📄 PDF Abstract BibTeX arXiv:2502.08101

Code (0)

등록된 구현이 없습니다.

Tasks

ClassificationNode ClassificationRepresentation Learning

Methods 이 논문이 사용한 방법론

Attention 설명 없음
Linear Layer A Linear Layer is a projection $\mathbf{XW + b}$.
Multi-Head Attention 설명 없음
Position-Wise Feed-Forward Layer 설명 없음
Adam 설명 없음
Softmax The Softmax output function transforms a previous layer's output into a vector of probabilities. It is commonly used for multiclass classification. Given an input vector $x$…
Absolute Position Encodings Absolute Position Encodings are a type of position embeddings for [Transformer-based models] where positional encodings are…
Dropout Dropout is a regularization technique for neural networks that drops a unit (along with connections) at training time with a specified probability $p$ (a common value is…

Similar Papers 제목 키워드 기반

Leveraging Contrastive Learning for Enhanced Node Representations in Tokenized Graph Transformers

2024-06-27 · Jinsong Chen, Hanpeng Liu, John E. Hopcroft, Kun He

While tokenized graph Transformers have demonstrated strong performance in node classification tasks, their reliance on a limited subset of nodes with high similarity scores for constructing token sequences overlooks val…

Contrastive LearningNode Classification

NTFormer: A Composite Node Tokenized Graph Transformer for Node Classification

2024-06-27 · Jinsong Chen, Siyu Jiang, Kun He

Recently, the emerging graph Transformers have made significant advancements for node classification on graphs. In most graph Transformers, a crucial step involves transforming the input graph into token sequences as the…

Node ClassificationRepresentation Learning

NAGphormer: A Tokenized Graph Transformer for Node Classification in Large Graphs

2022-06-10 · Jinsong Chen, Kaiyuan Gao, Gaichao Li, Kun He

The graph Transformer emerges as a new architecture and has shown superior performance on various graph mining tasks. In this work, we observe that existing graph Transformers treat nodes as independent tokens and constr…

Graph LearningGraph MiningNode Classification

Curve Your Attention: Mixed-Curvature Transformers for Graph Representation Learning

2023-09-08 · Sungjun Cho, Seunghyuk Cho, Sungwoo Park, Hankook Lee 외

Real-world graphs naturally exhibit hierarchical or cyclical structures that are unfit for the typical Euclidean space. While there exist graph neural networks that leverage hyperbolic or spherical spaces to learn repres…

Graph ReconstructionGraph Representation LearningNode ClassificationRepresentation Learning

Tokenized Graph Transformer with Neighborhood Augmentation for Node Classification in Large Graphs

2023-05-22 · Jinsong Chen, Chang Liu, Kaiyuan Gao, Gaichao Li 외

Graph Transformers, emerging as a new architecture for graph representation learning, suffer from the quadratic complexity on the number of nodes when handling large graphs. To this end, we propose a Neighborhood Aggrega…

Data AugmentationGraph Representation LearningNode ClassificationRepresentation Learning