Computer Science ›› 2022, Vol. 49 ›› Issue (9): 76-82.doi: 10.11896/jsjkx.210900078

• Database & Big Data & Data Science • Previous Articles     Next Articles

Author’s Academic Behavior Prediction Based on Heterogeneous Network Representation Learning

HUANG Li1, ZHU Yan1, LI Chun-ping2   

  1. 1 School of Computing and Artificial Intelligence,Southwest Jiaotong University,Chengdu 611756,China
    2 School of Software,Tsinghua University,Beijing 100091,China
  • Received:2021-09-10 Revised:2022-01-25 Online:2022-09-15 Published:2022-09-09
  • About author:HUANG Li,born in 1996,postgraduate.Her main research interests include representation learning,data mining and link prediction.
    ZHU Yan,born in 1965,Ph.D,professor,Ph.D supervisor,is a member of China Computer Federation.Her main research interests include data mining,Web anomaly and intelligent analysis.
  • Supported by:
    Sichuan Province Science and Technology Project(2019YFSY0032).

Abstract: The author's academic behavior prediction aims to mine the behavioral relationships of authors from heterogeneous academic networks to promote scientific research cooperation and produce high-level and high-quality research results.Most of the existing methods of node representation learning do not consider the semantic feature,content feature,global structure of the node,etc.It is difficult to effectively learn the low-dimensional characteristics of the node in the network.In order to effectively integrate the multi-dimensional features and global structure of nodes,a heterogeneous network representation learning method(HNEMA) that integrates BiLSTM,attention mechanism and clustering algorithm is proposed to improve the predictive effect of author's academic behavior.HNEMA first integrates the multi-dimensional features of nodes based on BiLSTM and attention mechanism,aggregates the same type of neighbors on the same meta-path or different meta-paths,and then aggregates the multi-dimensional features of all neighbors of the node to be characterized.Based on this,a clustering algorithm is used to capture the global features of the node,so as to comprehensively and effectively learn the low-dimensional characteristics of the node.On the basis of comprehensive feature learning,logistic regression classifier is used to predict author's academic behavior.Validation experiments on three public datasets show that HNEMA has a certain degree of improvement in AUC and F1 indicators compared to other methods.

Key words: Heterogeneous network, Network representation learning, Link prediction, Meta-path

CLC Number: 

  • TP183
[1]KUMAR A,SINGH S S,SINGH K,et al.Link prediction techniques,applications,and performance:A survey[J/OL].Physica A:Statistical Mechanics and its Applications,2020,553.https://doi.org/10.1016/j.physa.
[2]SHI C,SUN Y.Research progress of heterogeneous networkrepresentation learning[J].Communications of the CCF,2018,14(3):16-20.
[3]DONG Y,CHAWLA N V,SWAMI A.Metapath2vec:Scalable representation learning for heterogeneous networks [C]//Proceedings of the 23rd ACM SIGKDD international conference on knowledge discovery and data mining.2017:135-144.
[4]WANG X,JI H,SHI C,et al.Heterogeneous graph attention network[C]//The World Wide Web Conference.2019:2022-2032.
[5]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.
[6]PEROZZI B,AL-RFOU R,SKIENA S.DeepWalk:online lear-ning of social representations [C]//Proc of the 20th ACM SIGKDD International Conference on Knowledge Discovery and Data Mining.New York:ACM Press.2014:701-710.
[7]GROVER A,LESKOVEC J.Node2vec:Scalable feature learning for networks [C]//Proceedings of the 22nd ACM SIGKDD International Conference on Knowledge Discovery and Data Mining.New York:ACM Press.2016:855-864.
[8]WANG X,SHI C,HU B,et al.Heterogeneous information network embedding for recommendation[J].IEEE Transactions on Knowledge and Data Engineering,2018,31(2):357-370.
[9]ZHANG Y,SHI C.Hyperbolic heterogeneous information network embedding[J].Proceedings of the AAAI Conference on Artificial Intelligence,2019,33(1):5337-5344.
[10]HU Z,DONG Y,WANG K.Heterogeneous graph transformer [C]//Proceedings of The Web Conference.2020:2704-2710.
[11]CAO M,MA X,XU M,et al.Heterogeneous information network embedding with meta-path based on graph attention networks [C]//International Conference on Artificial Neural Networks.2019:622-634.
[12]SANKAR A,ZHANG X,CHANG K C.Meta-gnn:metagraph neural network for semi-supervised learning in attributed heterogeneous information networks[C]//Proceedings of the 2019 IEEE/ACM International Conference on Advances in Social Networks Analysis and Mining.2019:137-144.
[13]ZHOU S,BU J,WANG X,et al.Hahe:Hierarchical attentive heterogeneous information network embedding[J].arXiv:1902.01475,2019.
[14]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:2331-2341.
[15]HU B,FANG Y,SHI C.Adversarial learning on heterogeneous information networks[C]//Proceedings of the 25th ACM SIGKDD International Conference on Knowledge Discovery & Data Mining.2019:120-129.
[16]LU Y,SHI C,HU L,et al.Relation structure-aware heterogeneous information network embedding[J].Proceedings of the AAAI Conference on Artificial Intelligence,2019,33(1):4456-4463.
[17]HAMILTON W L,YING R,LESKOVEC J.Inductive representation learning on large graphs[C]//Proceedings of the 31st International Conference on Neural Information Processing Systems.2017:1025-1035.
[18]LE Q,MIKOLOV T.Distributed representations of sentences and documents[C]//International conference on machine learning.PMLR,2014:1188-1196.
[19]ROZEMBERCZKI B,DAVIES R,SARKAR R,et al.Gemsec:Graph embedding with self clustering [C]//Proceedings of the 2019 IEEE/ACM International Conference on Advances in Social Networks Analysis and Mining.2019:65-72.
[20]ZHANG C,SWAMI A,CHAWLA N V.Shne:Representation learning for semantic-associated heterogeneous networks [C]//Proceedings of the Twelfth ACM International Conference on Web Search and Data Mining.2019:690-698.
[1] SONG Jie, LIANG Mei-yu, XUE Zhe, DU Jun-ping, KOU Fei-fei. Scientific Paper Heterogeneous Graph Node Representation Learning Method Based onUnsupervised Clustering Level [J]. Computer Science, 2022, 49(9): 64-69.
[2] LYU Xiao-feng, ZHAO Shu-liang, GAO Heng-da, WU Yong-liang, ZHANG Bao-qi. Short Texts Feautre Enrichment Method Based on Heterogeneous Information Network [J]. Computer Science, 2022, 49(9): 92-100.
[3] LI Yong, WU Jing-peng, ZHANG Zhong-ying, ZHANG Qiang. Link Prediction for Node Featureless Networks Based on Faster Attention Mechanism [J]. Computer Science, 2022, 49(4): 43-48.
[4] ZHAO Xue-lei, JI Xin-sheng, LIU Shu-xin, LI Ying-le, LI Hai-tao. Link Prediction Method for Directed Networks Based on Path Connection Strength [J]. Computer Science, 2022, 49(2): 216-222.
[5] PU Shi, ZHAO Wei-dong. Community Detection Algorithm for Dynamic Academic Network [J]. Computer Science, 2022, 49(1): 89-94.
[6] JIANG Zong-li, FAN Ke, ZHANG Jin-li. Generative Adversarial Network and Meta-path Based Heterogeneous Network Representation Learning [J]. Computer Science, 2022, 49(1): 133-139.
[7] ZHAO Jin-long, ZHAO Zhong-ying. Recommendation Algorithm Based on Heterogeneous Information Network Embedding and Attention Neural Network [J]. Computer Science, 2021, 48(8): 72-79.
[8] CHEN Heng, WANG Wei-mei, LI Guan-yu, SHI Yi-ming. Knowledge Graph Completion Model Using Quaternion as Relational Rotation [J]. Computer Science, 2021, 48(5): 225-231.
[9] HU Xin-tong, SHA Chao-feng, LIU Yan-jun. Post-processing Network Embedding Algorithm with Random Projection and Principal Component Analysis [J]. Computer Science, 2021, 48(5): 124-129.
[10] CHENG Yun-fei, TIAN Hong-xin, LIU Zu-jun. Collaborative Optimization of Joint User Association and Power Control in NOMA Heterogeneous Network [J]. Computer Science, 2021, 48(3): 269-274.
[11] GONG Zhui-fei, WEI Chuan-jia. Link Prediction of Complex Network Based on Improved AdaBoost Algorithm [J]. Computer Science, 2021, 48(3): 158-162.
[12] LI Xin-chao, LI Pei-feng, ZHU Qiao-ming. Directed Network Representation Method Based on Hierarchical Structure Information [J]. Computer Science, 2021, 48(2): 100-104.
[13] FU Kun, ZHAO Xiao-meng, FU Zi-tong, GAO Jin-hui, MA Hao-ran. Deep Network Representation Learning Method on Incomplete Information Networks [J]. Computer Science, 2021, 48(12): 212-218.
[14] GONG Zhui-fei, WEI Chuan-jia. Complex Network Link Prediction Method Based on Topology Similarity and XGBoost [J]. Computer Science, 2021, 48(12): 226-230.
[15] HUANG Shou-meng. Heterogeneous Network Link Prediction Model Based on Supervised Learning [J]. Computer Science, 2021, 48(11A): 111-116.
Viewed
Full text


Abstract

Cited

  Shared   
  Discussed   
No Suggested Reading articles found!