Computer Science ›› 2026, Vol. 53 ›› Issue (6A): 250900103-8.doi: 10.11896/jsjkx.250900103

• Big Data & Data Science • Previous Articles     Next Articles

Optimization of HAN-based GNN-Transformer Collaborative Contrastive Learning Framework

ZHANG Zihao, WU Zezhong   

  1. Chengdu University of Information Technology,Chengdu 610000,China
  • Online:2026-06-16 Published:2026-06-12
  • About author:ZHANG Zihao,born in 2001,master.His main research interest is graph neural networks.
    WU Zezhong,born in 1970,Ph.D,professor.His main research interests include optimization theory and algorithms and swarm intelligence algorithms and applications.
  • Supported by:
    National Social Science Fund Annual Project “the Mechanism and Pathways of Digital Transformation of County-Village Logistics Distribution System in the Chengdu-Chongqing Economic Circle”(24XGLO28).

Abstract: GNNs have demonstrated strong performance in graph representation learning due to their message-passing mechanism.However,they encounter challenges such as over-smoothing and insufficient capture of multi-hop neighbor information when processing heterogeneous graphs.The GNN-Transformer collaborative contrastive learning framework(GTC) combines the local information aggregation capability of GNNs with the global information modeling capability of Transformers.This framework implements self-supervised heterogeneous graph representation learning through cross-view contrastive learning,effectively mitigating the over-smoothing problem that GNNs experience during neighbor information aggregation.This study enhances the GNN branch of the GNN-Transformer collaborative contrastive learning framework by incorporating the node-level and semantic-level attention mechanisms from the heterogeneous graph attention network(HAN).This optimization enables more effective capture of information from different node and edge types in heterogeneous graphs during neighbor aggregation.Experiments on the ACM dataset demonstrate that the improved GNN-Transformer collaborative contrastive learning framework achieves superior performance in node classification and node clustering tasks.For node classification,with 20,40,and 60 labeled nodes,the mo-del exhibits average improvements of 0.52%,3.07%,and 3.14% in AUC,macro-F1,and micro-F1 scores,respectively.In node clustering,normalized mutual information(NMI) and adjusted rand index(ARI) increase by 6.57% and 6.38%,respectively.These results confirm that HAN's hierarchical attention mechanism enables finer neighbor aggregation and metapath semantic fusion in heterogeneous graph representation learning,offering a novel approach to alleviating the over-smoothing problem in GNNs.

Key words: Graph neural networks, Heterogeneous graphs, Contrastive learning, Attention mechanism, Self-supervised

CLC Number: 

  • TP183
[1] WU Z,PAN S,CHEN F,et al.A comprehensive survey ongraph neural networks[J].IEEE Transactions on Neural Networks and Learning Systems,2020,32(1):4-24.
[2] ZHANG L Y,SUN H H,SUN Y F,et al.Review of node classification methods based on graph convolutional neural networks[J].Computer Science,2024,51(4):95-105.
[3] YOU H,DING C F,MA L R,et al.Graph Transformer techno-logy and research progress:from fundamental theory to cutting-edge applications [J].Application Research of Computers,2025,42(4):975-986.
[4] SUN Y,ZHU D,WANG Y,et al.GTC:GNN-transformer co-contrastive learning for self-supervised heterogeneous graph representation[J].Neural Networks,2025,181:106645.
[5] SCHLICHTKRULL M,KIPF T N,BLOEM P,et al.Modeling relational data with graph convolutional networks[C]//Euro-pean Semantic Web Conference.Cham:Springer,2018:593-607.
[6] WANG X,JI H,SHI C,et al.Heterogeneous graph attention network[C]//The World Wide Web Conference.2019:2022-2032.
[7] XU B B,CEN K T,HUANG J J,et al.A survey on graph con-volutional neural network[J].Chinese Journal of Computers,2020,43(5):755-780.
[8] KIPF T N,WELLING M.Semi-supervised classification withgraph convolutional networks[J].arXiv:1609.02907,2016.
[9] HAMILTON W,YING Z,LESKOVEC J.Inductive representation learning on large graphs[C]//Advances in Neural Information Processing Systems.2017.
[10] VELICKOVIC P,CUCURULL G,CASANOVA A,et al.Graph attention networks[J].Stat,2017,1050(20):10-48550.
[11] BING R,YUAN G,ZHU M,et al.Heterogeneous graph neural networks analysis:a survey of techniques,evaluations and applications[J].Artificial Intelligence Review,2023,56(8):8003-8042.
[12] DONG Y,CHAWLA N V,SWAMI A.metapath2vec:Scalablerepresentation learning for heterogeneous networks[C]//Proceedings of the 23rd ACM SIGKDD International Conference on Knowledge Discovery and Data Mining.2017:135-144.
[13] FU T,LEE W C,LEI Z.Hin2vec:Explore meta-paths in heterogeneous information networks for representation learning[C]//Proceedings of the 2017 ACM on Conference on Information and Knowledge Management.2017:1797-1806.
[14] ZHANG C,SONG D,HUANG C,et al.Heterogeneous graph neural network[C]//Proceedings of the 25th ACM SIGKDD International Conference on Knowledge Discovery & Data Mi-ning.2019:793-803.
[15] FU X,ZHANG J,MENG Z,et al.Magnn:Metapath aggregated graph neural network for heterogeneous graph embedding[C]//Proceedings of the Web Conference 2020.2020:2331-2341.
[16] LI X,LIU X P,LI W C,et al.Survey on contrastive learning research[J].Journal of Chinese Computer Systems,2023,44(4):787-797.
[17] OORD A,LI Y,VINYALS O.Representation learning with contrastive predictive coding[J].arXiv:1807.03748,2018.
[18] CEN K T,SHEN H W,CAO Q,et al.A survey on graph con-trastive learning[J].Journal of Chinese Information Processing,2023,37(5):1-2.
[19] WANG Y,SANG L,ZHANG Y,et al.Generative-contrastiveheterogeneous graph neural network[J].IEEE Transactions on Big Data,2025,11(6):3061-3073.
[20] ZHAO T,WANG Y,WANG J,et al.Hybrid-Collaborative Augmentation and Contrastive Sample Adaptive-Differential Awareness for Robust Attributed Graph Clustering[J].arXiv:2510.02731,2025.
[21] ZHAO T,WANG Y,XU S,et al.Dual-level Noise Augmentation for Graph Clustering with Triplet-wise Contrastive Lear-ning[J].Pattern Recognition,2025,172:112463.
[22] SCHROFF F,KALENICHENKO D,PHILBIN J.Facenet:A unified embedding for face recognition and clustering[C]//Proceedings of the IEEE Conference on Computer Vision and Pattern Recognition.2015:815-823.
[23] VELICˇKOVIĆ P,FEDUS W,HAMILTON W L,et al.Deep graph infomax[J].arXiv:1809.10341,2018.
[24] WANG X,LIU N,HAN H,et al.Self-supervised heterogeneous graph neural network with co-contrastive learning[C]//Proceedings of the 27th ACM SIGKDD Conference on Knowledge Discovery & Data Mining.2021:1726-1736.
[1] ZHANG Xinliang, LIU Lilong, CHEN Shangheng, CHEN Ziyang, QIAN Shengsheng. Dual-stream Heterogeneous Social Graph for Micro-video Popularity Prediction [J]. Computer Science, 2026, 53(6A): 250800073-8.
[2] ZHONG Hao, KONG Qingxuan, CAI Xianqing, LI Zhizhong, SUN Hao. Intelligent Recognition Method Based on Multimodal Feature Fusion [J]. Computer Science, 2026, 53(6A): 250700065-10.
[3] DUAN Haiying, WANG Baohui, HUANG He. Malicious Traffic Detection Method of ICMP Covert Channel Based on Baseline Features [J]. Computer Science, 2026, 53(6A): 250200069-11.
[4] LI Jie, WANG Baohui, ZHANG Jingyuan. DDoS Attack Detection Based on Attention Mechanism TCN-BiLSTM [J]. Computer Science, 2026, 53(6A): 250300060-9.
[5] ZHANG Shouyi, SHEN Qiang, GUO Yiran, WANG Hanyu. Rain and Fog Weather Object Detection Algorithm Based on Improved YOLOv8 Model [J]. Computer Science, 2026, 53(6A): 250300090-7.
[6] YANG Geer, WANG Xin, SUN Wei, WANG Xinge, HU Zhongrui, MENG Wenjun, ZHANG Junqiang, WU Xinghui, LIU Jinshan, YAN Yuming. Survey on Positional Encoding Algorithms in Deep Learning [J]. Computer Science, 2026, 53(6A): 250300107-16.
[7] WEI Wei, LI Bicheng, ZHU Zhenshui, ZUO Jun. Semantic Modeling and Co-attention Mechanism for Multimodal Sarcasm Detection Method [J]. Computer Science, 2026, 53(6A): 250400127-6.
[8] FENG Guang, LIN Jianzhong, ZHONG Ting, ZHOU Yuanhua, ZHENG Runting, LIU Tianxiang. Triple Extraction Based on Pixel Difference Convolutional Network and Attention Mechanism [J]. Computer Science, 2026, 53(6A): 250400136-10.
[9] HAN Zhigeng, FU Chunshuo. Fraud Detection Model Based on Dual-space Heterogeneous Graph Neural Network [J]. Computer Science, 2026, 53(6A): 250600050-6.
[10] CHEN Dianlong, LIU Tengbin, GAO Xiong, TIAN Zijian, ZHU Wenbing, ZOU Shun, WANG Qiang. Defect Detection of Transmission Line Fittings Based on Multiscale Feature Fusion Attention and Cross-layer Aggregation [J]. Computer Science, 2026, 53(6A): 250600110-7.
[11] SHEN Yingchun, FENG Xiaohan, LI Qian. Accurate Recognition of Dialect Based on CTC-Conformer Model [J]. Computer Science, 2026, 53(6A): 250600112-8.
[12] ZHANG Xin, CHEN Wen. CausalVulGNN:Framework for Software Vulnerability Explanation Based on Causal Inferenceand Graph Neural Networks [J]. Computer Science, 2026, 53(6): 427-436.
[13] XU Zhihong, YANG Xinlei, WANG Liqin, DONG Yongfeng, WANG Xu. Knowledge Tracing Model Based on Relational Learning Memory Network [J]. Computer Science, 2026, 53(6): 84-92.
[14] KE Changbo, LI Tianhao, ZHANG Bolei, XIAO Fu, XU Kang. Teaching Evaluation Sentiment Analysis Method Based on Capsule Network [J]. Computer Science, 2026, 53(6): 10-18.
[15] LIU Ruyi, LYU Xiaohan, MIAO Qiguang, LU Zixiang, WANG Di. Academic Early Warning Prediction Model Based on Attention Mechanism and FeatureInteraction [J]. Computer Science, 2026, 53(6): 19-29.
Viewed
Full text


Abstract

Cited

  Shared   
  Discussed   
No Suggested Reading articles found!