Leveraging Contrastive Learning for Enhanced Node Representations in Tokenized Graph Transformers
Jinsong Chen, Hanpeng Liu, John E. Hopcroft, Kun He
Abstract
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 valuable information from other nodes, hindering their ability to fully harness graph information for learning optimal node representations. To address this limitation, we propose a novel graph Transformer called GCFormer. Unlike previous approaches, GCFormer develops a hybrid token generator to create two types of token sequences, positive and negative, to capture diverse graph information. And a tailored Transformer-based backbone is adopted to learn meaningful node representations from these generated token sequences. Additionally, GCFormer introduces contrastive learning to extract valuable information from both positive and negative token sequences, enhancing the quality of learned node representations. Extensive experimental results across various datasets, including homophily and heterophily graphs, demonstrate the superiority of GCFormer in node classification, when compared to representative graph neural networks (GNNs) and graph Transformers.
Ask about this paper
Your agent reads all of it.
Lune indexed this paper to the last equation, along with the top-tier papers that cite it. Ask a question and the answer quotes them.
Your agent calls
Luneget_paper_fulltext
Free to start. No credit card required.
Terminal
Install the CLIlune papers fulltext d8b1b0f1-4fb2-4679-8a93-f8f76cb73a85Cited by top-tier papers14
- FedSSP: Federated Graph Learning with Spectral Knowledge and Personalized PreferenceZihan Tan, Guancheng Wan, Wenke Huang, Mang YeNeurIPS 2024 · 40 citations
- Rethinking Tokenized Graph Transformers for Node ClassificationJinsong Chen, Chenyang Li, Gaichao Li, John E. Hopcroft et al.NeurIPS 2025 · 8 citations
- A Closer Look at Graph Transformers: Cross-Aggregation and BeyondJiaming Zhuo, Ziyi Ma, Yintong Lu, Yuwei Liu et al.NeurIPS 2025 · 4 citations
- Unifying and Enhancing Graph Transformers via a Hierarchical Mask FrameworkYujie Xing, Xiao Wang, Bin Wu, Hai Huang et al.NeurIPS 2025 · 2 citations
- DUALFormer: Dual Graph TransformerJiaming Zhuo, Yuwei Liu, Yintong Lu, Ziyi Ma et al.ICLR 2025
Builds on26
- Big Bird: Transformers for Longer SequencesManzil Zaheer, Guru Guruganesh, Kumar Avinava Dubey, Joshua Ainslie et al.NeurIPS 2020 · 3,159 citations
- Graph Contrastive Learning with AugmentationsYuning You, Tianlong Chen, Yongduo Sui, Ting Chen et al.NeurIPS 2020 · 3,042 citations
- Reformer: The Efficient TransformerNikita Kitaev, Lukasz Kaiser, Anselm LevskayaICLR 2020 · 2,878 citations
- Do Transformers Really Perform Badly for Graph Representation?Chengxuan Ying, Tianle Cai, Shengjie Luo, Shuxin Zheng et al.NeurIPS 2021 · 1,632 citations
- Measuring and Relieving the Over-Smoothing Problem for Graph Neural Networks from the Topological ViewDeli Chen, Yankai Lin, Wei Li, Peng Li et al.AAAI 2020 · 1,353 citations
Related papers
- Tokenphormer: Structure-aware Multi-token Graph Transformer for Node ClassificationZijie Zhou, Zhaoqi Lu, Xuekai Wei, Rongqin Chen et al.AAAI 2025 · 5 citations
- NAGphormer: A Tokenized Graph Transformer for Node Classification in Large GraphsJinsong Chen, Kaiyuan Gao, Gaichao Li, Kun HeICLR 2023 · 22 citations
- GNN-Transformer Cooperative Architecture for Trustworthy Graph Contrastive LearningJianqing Liang, Xinkai Wei, Min Chen, Zhiqiang Wang et al.AAAI 2025 · 1 citation
- VecFormer: Towards Efficient and Generalizable Graph Transformer with Graph Token AttentionJingbo Zhou, Jun Xia, Siyuan Li, Yunfan Liu et al.WWW 2026
- Heterformer: Transformer-based Deep Node Representation Learning on Heterogeneous Text-Rich NetworksBowen Jin, Yu Zhang, Qi Zhu, Jiawei HanKDD 2023 · 27 citations
