Heterogeneous Information Network Embedding for Mention Recommendation

Feng Yi, Bo Jiang, Jianjun Wu
2020 IEEE Access  
Mention recommendation is the task of recommending the right candidate users in a message. Many works have been conducted on the problem of whom to mention. However, due to the sparsity and heterogeneous of mention data, none of them well solve the problem. The recent advances in network embedding representation learning provide an effective approach to model the sparsity and heterogeneous simultaneously in heterogeneous information network. To this end, we propose a novel Network Embedding
more » ... ion (NEM) recommendation model to recommend the right users in a message. NEM constructs a heterogeneous mention network based on different relationships among different entities. Then NEM learns a unified low dimensional embedding vector using random walk for users and messages by considering network structure and vertex content information. Finally, whom to mention is ranked by calculating the relevance scores from heterogeneous user and message embeddings. To evaluate the proposed method, we construct a large dataset and their corresponding social networks from a real-world social media platform. Through extensive experiments on real-world mention collection, we demonstrate that our proposed model outperforms the previous state-of-the-art methods in term of recommendation task. INDEX TERMS Mention recommendation, network embedding, relation network, heterogeneous information. This work is licensed under a Creative Commons Attribution 4.0 License. For more information, see https://creativecommons.org/licenses/by/4.0/ VOLUME 8, 2020
doi:10.1109/access.2020.2994313 fatcat:kor7j3umkndrxojutvklm5asqm