Attributed Heterogeneous Information Network Embedding with Self-Attention Mechanism for Product Recommendation

被引:0
作者
Wang H. [1 ]
Yang D. [1 ]
Nie T. [2 ]
Kou Y. [2 ]
机构
[1] School of Computer Science and Software Engineering, University of Science and Technology Liaoning, Anshan
[2] School of Computer Science and Engineering, Northeastern University, Shenyang
来源
Jisuanji Yanjiu yu Fazhan/Computer Research and Development | 2022年 / 59卷 / 07期
基金
中国国家自然科学基金;
关键词
Attributed heterogeneous information network; Deep neural network; Network embedding; Product recommendation; Self-attention mechanism;
D O I
10.7544/issn1000-1239.20210016
中图分类号
学科分类号
摘要
Heterogeneous network embedding based recommendation technology has the capability to capture the structural information in the network effectively, thus improving the recommendation performance. However, the existing recommendation technology based on heterogeneous network embedding not only ignores the attribute information of nodes and various types of edge relations between nodes, but also ignores the diverse influences of different nodes' attribute information on recommendation results. To address the above issues, a product recommendation framework based on attributed heterogeneous information network embedding with self-attention mechanism (AHNER) is proposed. The framework utilizes attributed heterogeneous information network embedding to learn the unified low-dimensional embedding representations of users and products. When learning node embedding representation, considering that different attribute information has different effects on recommendation results and different edge relations between nodes reflect users' different preferences for products, self-attention mechanism is exploited to mine the latent information of node attribute information and different edge types and learn attribute embedding representation is learned. Meanwhile, in order to overcome the limitation of traditional dot product method as matching function, the framework also exploits deep neural network to learn more effective matching function to solve the recommendation problem. We conduct extensive experiments on three public datasets to evaluate the performance of AHNER. The experimental results reveal that AHNER is feasible and effective. © 2022, Science Press. All right reserved.
引用
收藏
页码:1509 / 1521
页数:12
相关论文
共 38 条
  • [1] Chen Xuening, Liu Hanwen, Yang Dan, Improved LSH for privacy-aware and robust recommender system with sparse data in edge environment[J/OL], EURASIP Journal on Wireless Communications and Networking, (2019)
  • [2] Ye Zhonglin, Zhao Haixing, Zhang Ke, Et al., Network representation learning using the optimizations of neighboring vertices and relation model, Journal of Computer Research and Development, 56, 12, pp. 2562-2577, (2019)
  • [3] Wu Yao, Shen Derong, Kou Yue, Et al., Heterogeneous information networks embedding based on multiple meta-graph fusion, Journal of Computer Research and Development, 57, 9, pp. 1928-1938, (2020)
  • [4] Chang Shiyu, Han Wei, Tang Jiliang, Et al., Heterogeneous network embedding via deep architectures, Proc of the 21st ACM SIGKDD Int Conf on Knowledge Discovery and Data Mining, pp. 119-128, (2015)
  • [5] Pan Shirui, Wu Jia, Zhu Xingquan, Et al., Tri-party deep network representation, Proc of the 25th Int Joint Conf on Artificial Intelligence, pp. 1895-1901, (2016)
  • [6] Zhao Yu, Liu Zhiyuan, Sun Maosong, Representation learning for measuring entity relatedness with rich information, Proc of the 24th Int Joint Conf on Artificial Intelligence, pp. 1412-1418, (2015)
  • [7] Perozzi B, Ai-rfou R, Skiena S., Deepwalk: Online learning of social representations, Proc of the 20th ACM SIGKDD Int Conf on Knowledge Discovery and Data Mining, pp. 701-710, (2014)
  • [8] Grover A, Leskovec J., node2vec: Scalable feature learning for networks, Proc of the 22nd ACM SIGKDD Int Conf on Knowledge Discovery and Data Mining, pp. 855-864, (2016)
  • [9] Dong Yuxiao, Chawla N V, Swami A., metapath2vec: Scalable representation learning for heterogeneous networks, Proc of the 23rd ACM SIGKDD Int Conf on Knowledge Discovery and Data Mining, pp. 135-144, (2017)
  • [10] Shi Chuan, Hu Binbin, Zhao Xin Wayne, Et al., Heterogeneous information network embedding for recommendation, IEEE Transactions on Knowledge and Data Engineering, 31, 2, pp. 357-370, (2019)