Improving link prediction models through a performance enhancement scheme:a study on semi-supervised learning and model soup

作     者：Qi Donglin Chen Shudong Du Rong Yu Yong Tong Da Qi Donglin;Chen Shudong;Du Rong;Yu Yong;Tong Da

作者机构：Intelligent Manufacturing Electronics Research and Development CenterInstitute of Microelectronics of the Chinese Academy of SciencesBeijing 100029China School of Integrated CircuitsUniversity of Chinese Academy of SciencesBeijing 100049China 

出 版 物：《The Journal of China Universities of Posts and Telecommunications》 (中国邮电高校学报（英文版）)

年 卷 期：2024年第31卷第4期

页      面：43-53页

核心收录：

学科分类：12[管理学] 1201[管理学-管理科学与工程(可授管理学、工学学位)] 081104[工学-模式识别与智能系统] 081203[工学-计算机应用技术] 08[工学] 0835[工学-软件工程] 0811[工学-控制科学与工程] 0812[工学-计算机科学与技术（可授工学、理学学位）] 

主　　题：natural language processing knowledge graph(KG) link prediction model soup semi-supervised learning 

摘      要：As an important method for knowledge graph(KG)complementation,link prediction has become a hot research topic in recent *** this paper,a performance enhancement scheme for link prediction models based on the idea of semi-supervised learning and model soup is proposed,which effectively improves the model performance on several mainstream link prediction models with small changes to their *** novel scheme consists of two main parts,one is predicting potential fact triples in the graph with semi-supervised learning strategies,the other is creatively combining semi-supervised learning and model soup to further improve the final model performance without adding significant computational *** validate the effectiveness of the scheme for a variety of link prediction models,especially on the dataset with dense *** terms of CompGCN,the model with the best overall performance among the tested models improves its Hits@1 metric by 14.7%on the FB15K-237 dataset and 7.8%on the WN18RR dataset after using the enhancement ***,it is observed that the semi-supervised learning strategy in the augmentation scheme has a significant improvement for multi-class link prediction models,and the performance improvement brought by the introduction of the model soup is related to the specific tested models,as the performances of some models are improved while others remain largely unaffected.