Computer Science ›› 2023, Vol. 50 ›› Issue (10): 146-155.doi: 10.11896/jsjkx.221000063

• Artificial Intelligence • Previous Articles     Next Articles

Knowledge Enhanced Relationship Prediction Model for Enterprise Entities

WANG Jiaqi1, LI Wengen1, GUAN Jihong1, XING Ting2, WEI Xiaomin2, SHAO Bingqing2, FU Chongjie2   

  1. 1 College of Electronic and Information Engineering,Tongji University,Shanghai 201804,China
    2 Beijing Shangqi Digital Technology Co.,Ltd.,Beijing 100084,China
  • Received:2022-10-10 Revised:2023-02-20 Online:2023-10-10 Published:2023-10-10
  • About author:WANG Jiaqi,born in 2000,doctoral student,is a member of China Compu-ter Federation.His main research intere-sts include knowledge graph and data mining.GUAN Jihong,born in 1969,Ph.D,professor,Ph.D supervisor,is a member of China Computer Federation.Her main research interests include machine learning and bioinformatics.
  • Supported by:
    National Natural Science Foundation of China(U1936205,62202336) and Shanghai Soft Science Research Program(22692194100).

Abstract: With the development of knowledge graphs,a variety of industrial knowledge graphs have come into being.However,these industrial knowledge graphs lack sufficient relationships among enterprises,such as up-down stream relationship,supply relationship,cooperation and competition relationship,which greatly affects their applications.Most existing methods for predicting the enterprise entity relationships focus on the fact triples and cannot fully utilize multiple perspectives such as enterprise descriptions and associated entity descriptions.To solve this problem,KERP,a knowledge enhanced relationship prediction model for enterprise entities is proposed.The model first improves enterprise features representations using a multi-view entity feature lear-ning module,then uses graph attention network to obtain higher-order semantic representations of entities and fuses lower-order semantic representations learned by TransR for knowledge enhancement,and finally predicts enterprise entity relationships by a convolutional decoder ConvE.Experimental results on the new energy automobile industrial knowledge graph show that KERP has better results in predicting the relationships between enterprises with a improvement of 6.7%in terms of F1 value compared with the existing models.Generalization is also evaluated on multiple datasets,and the experimental results demonstrate that KERP has good generality for generalized entity relationship prediction tasks.

Key words: Industrial knowledge graph, Enterprise entity relationship, Knowledge completion, Link prediction, Knowledge enhancement

CLC Number: 

  • TP391
[1]CHEN Z,WANG Y,ZHAO B,et al.Knowledge Graph Completion:A Review[J].IEEE Access,2020,8:192435-192456.
[2]LI F L,CHEN H,XU G,et al.AliMeKG:Domain knowledge graph construction and application in e-commerce[C]//Procee-dings of the 29th ACM International Conference on Information &Knowledge Management.2020:2581-2588.
[3]MOON C,JONES P,SAMATOVA N F.Learning Entity Type Embeddings for Knowledge Graph Completion[C]//Proceedings of the 2017 ACM on Conference on Information and Knowledge Management.2017:2215-2218.
[4]WANG J J,LIU J G,LI Z K.Research on Partnership of SupplyChain Based on Complex Network[J].Chinese Journal of Systems Science,2021,29(3):110-115130.
[5]LU Z Z,CHEN Q.Link Prediction of Enterprise Cooperation Relationship in Dynamic Supply Chain Network[J].Computer Engineering and Applications,2022,58(2):265-273.
[6]LIN Y,LIU Z,SUN M,et al.Learning entity and relation embeddings for knowledge graph completion[C]//Twenty-ninth AAAI Conference on Artificial Intelligence.2015.
[7]DETTMERS T,MINERVINI P,STENETORP P,et al.Convolutional 2d knowledge graph embeddings[C]//Proceedings of the AAAI Conference on Artificial Intelligence.2018.
[8]ZHANG J Z,HU Y M.Uncovering the Mechanism of Co-Authorship Network Evolution by Link Prediction[J].Information Science,2017,35(7):75-81.
[9]LU Z G,CHEN Q.Discovering Potential Partners via Projec-tion-Based Link Prediction in the Supply Chain Network[J].International Journal of Computational Intelligence Systems,2020,13(1):1253-1264.
[10]XIE M,WANG T,JIANG Q,et al.Higher-Order NetworkStructure Embedding in Supply Chain Partner Link Prediction[C]//CCF Conference on Computer Supported Cooperative Work and Social Computing.Singapore:Springer,2019:3-17.
[11]KERSTING K,RAEDT L D.Adaptive Bayesian logic programs[C]//International Conference on Inductive Logic Programming.Berlin:Springer,2001:104-117.
[12]LAO N,COHEN W W.Relational retrieval using a combination of path-constrained random walks[J].Machine Learning,2010,81(1):53-67.
[13]WANG Q,MAO Z,WANG B,et al.Knowledge graph embedding:A survey of approaches and applications[J].IEEE Tran-sactions on Knowledge and Data Engineering,2017,29(12):2724-2743.
[14]BORDES A,USUNIER N,GARCIA-DURAN A,et al.Translating embeddings for modeling multi-relational data[C]//Proceedings of the 26th International Conference on Neural Information Processing Systems-Volume 2.2013:2787-2795.
[15]JI G,HE S,XU L,et al.Knowledge graph embedding via dy-namic mapping matrix[C]//Proceedings of the 53rd Annual Meeting of the Association for Computational Linguistics and the 7th International Joint Conference on Natural Language Processing(volume 1:Long papers).2015:687-696.
[16]NICKEL M,TRESP V,KRIEGEL H P.A three-way model for collective learning on multi-relational data[C]//ICML.2011.
[17]TROUILLON T,WELBL J,RIEDEL S,et al.Complex embeddings for simple link prediction[C]//International Conference on Machine Learning.PMLR,2016:2071-2080.
[18]GU J,WANG Z,KUEN J,et al.Recent advances in convolu-tional neural networks[J].Pattern Recognition,2018,77:354-377.
[19]NGUYEN D Q,NGUYEN T D,NGUYEN D Q,et al.A novel embedding model for knowledge base completion based on con-volutional neural network[J].arXiv:1712.02121,2017.
[20]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.
[21]SHANG C,TANG Y,HUANG J,et al.End-to-end structure-aware convolutional networks for knowledge base completion[C]//Proceedings of the AAAI Conference on Artificial Intelligence.2019:3060-3067.
[22]NATHANI D,CHAUHAN J,SHARMA C,et al.Learning attention-based embeddings for relation prediction in knowledge graphs[J].arXiv:1906.01195,2019.
[23]XIAO H,HUANG M,MENG L,et al.SSP:semantic space projection for knowledge graph embedding with text descriptions[C]//Thirty-First AAAI Conference on Artificial Intelligence.2017.
[24]XIE R,LIU Z,JIA J,et al.Representation learning of knowledge graphs with entity descriptions[C]//Proceedings of the AAAI Conference on Artificial Intelligence.2016.
[25]ZHANG Z,CAO L,CHEN X,et al.Representation learning of knowledge graphs with entity attributes[J].IEEE Access,2020,8:7435-7441.
[26]LIN Z F,OU S Y.Research on Relation Prediction in Know-ledge Graphs by Fusing Structure and Text Features[J].Libraryand Information Service,2020,64(21):99-110.
[27]ZHAI S P,WANG S H,SHANG D R,etal.An adaptive model for knowledge representation with entity description[J].Journal of Chinese Information Processing,2021,35(01):43-53.
[28]BLEI D M,NG A Y,JORDAN M I.Latent dirichlet allocation[J].Journal of Machine Learning Research,2003,3(Jan):993-1022.
[29]SUN X L,ZHUANG W H,LI B,et al.Research on Characteristics and Cooperation Prediction of Industry-University-Research lnstitute Collaboration Based on Patents inRegional Equipment Manufacturing Industry[J].Science and Management,2020,40(1):31-40.
[30]MIKOLOV T,CHEN K,CORRADO G,et al.Efficient estimation of word representations in vector space[J].arXiv:1301.3781,2013.
[31]VELICKOVIC P,CUCURULL G,CASANOVA A,et al.Graph Attention Networks[J].arXiv:1710.10903,2017.
[32]SUN Z,VASHISHTH S,SANYAL S,et al.A Re-evaluation of Knowledge Graph Completion Methods[C]//Proceedings of the 58th Annual Meeting of the Association for Computational Linguistics.2020:5516-5522.
[33]XU B,XU Y,LIANG J,et al.CN-DBpedia:A never-ending Chinese knowledge extraction system[C]//International Confe-rence on Industrial,Engineering and Other Applications of Applied Intelligent Systems.Springer,2017:428-438.
[34]KINGMA D P,BA J.Adam:A method for stochastic optimization[J].arXiv:1412.6980,2014.
[1] TANG Shaosai, SHEN Derong, KOU Yue, NIE Tiezheng. Link Prediction Model on Temporal Knowledge Graph Based on Bidirectionally Aggregating Neighborhoods and Global Aware [J]. Computer Science, 2023, 50(8): 177-183.
[2] JIANG Linpu, CHEN Kejia. Self-supervised Dynamic Graph Representation Learning Approach Based on Contrastive Prediction [J]. Computer Science, 2023, 50(7): 207-212.
[3] LIANG Mingxuan, WANG Shi, ZHU Junwu, LI Yang, GAO Xiang, JIAO Zhixiang. Survey of Knowledge-enhanced Natural Language Generation Research [J]. Computer Science, 2023, 50(6A): 220200120-8.
[4] LI Yang, TANG Jiqiang, ZHU Junwu, LIANG Mingxuan, GAO Xiang. Aspect-based Sentiment Analysis Based on Prompt and Knowledge Enhancement [J]. Computer Science, 2023, 50(6A): 220300279-7.
[5] LI Shujing, HUANG Zengfeng. Mixed-curve for Link Completion of Multi-relational Heterogeneous Knowledge Graphs [J]. Computer Science, 2023, 50(4): 172-180.
[6] WANG Jingbin, LAI Xiaolian, LIN Xinyu, YANG Xinyi. Context-aware Temporal Knowledge Graph Completion Based on Relation Constraints [J]. Computer Science, 2023, 50(3): 23-33.
[7] 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.
[8] HUANG Li, ZHU Yan, LI Chun-ping. Author’s Academic Behavior Prediction Based on Heterogeneous Network Representation Learning [J]. Computer Science, 2022, 49(9): 76-82.
[9] 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.
[10] 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.
[11] 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.
[12] 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.
[13] GONG Zhui-fei, WEI Chuan-jia. Link Prediction of Complex Network Based on Improved AdaBoost Algorithm [J]. Computer Science, 2021, 48(3): 158-162.
[14] 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.
[15] 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.
Viewed
Full text


Abstract

Cited

  Shared   
  Discussed   
No Suggested Reading articles found!