LkeRec: Toward Lightweight End-to-End Joint Representation Learning for Building Accurate and Effective Recommendation

被引:3
作者
Yan, Surong [1 ]
Lin, Kwei-Jay [2 ]
Zheng, Xiaolin [3 ]
Wang, Haosen [1 ]
机构
[1] Zhejiang Univ Finance & Econ, Sch Informat Management & Artificial Intelligence, Hangzhou 310018, Peoples R China
[2] Univ Calif Irvine, Dept Elect Engn & Comp Sci, Irvine, CA 92697 USA
[3] Zhejiang Univ, Coll Comp Sci & Technol, Hangzhou 310027, Peoples R China
关键词
End-to-end joint learning; knowledge graph embedding; self-attention; scalability; recommendation;
D O I
10.1145/3486673
中图分类号
TP [自动化技术、计算机技术];
学科分类号
0812 ;
摘要
Explicit and implicit knowledge about users and items have been used to describe complex and heterogeneous side information for recommender systems (RSs). Many existing methods use knowledge graph embedding (KGE) to learn the representation of a user-item knowledge graph (KG) in low-dimensional space. In this article, we propose a lightweight end-to-end joint learning framework for fusing the tasks of KGE and RSs at the model level. Our method proposes a lightweight KG embedding method by using bidirectional bijection relation-type modeling to enable scalability for large graphs while using self-adaptive negative sampling to optimize negative sample generating. Our method further generates the integrated views for users and items based on relation-types to explicitly model users' preferences and items' features, respectively. Finally, we add virtual "recommendation" relations between the integrated views of users and items to model the preferences of users on items, seamlessly integrating RS with user-item KG over a unified graph. Experimental results on multiple datasets and benchmarks show that our method can achieve a better accuracy of recommendation compared with existing state-of-the-art methods. Complexity and runtime analysis suggests that our method can gain a lower time and space complexity than most of existing methods and improve scalability.
引用
收藏
页数:28
相关论文
共 63 条
  • [1] Learning Heterogeneous Knowledge Base Embeddings for Explainable Recommendation
    Ai, Qingyao
    Azizi, Vahid
    Chen, Xu
    Zhang, Yongfeng
    [J]. ALGORITHMS, 2018, 11 (09)
  • [2] Bordes A., 2011, P AAAI C ARTIFICIAL, P301
  • [3] Bordes A., 2013, P 26 INT C NEUR INF, V2, P2787
  • [4] A semantic matching energy function for learning with multi-relational data Application to word-sense disambiguation
    Bordes, Antoine
    Glorot, Xavier
    Weston, Jason
    Bengio, Yoshua
    [J]. MACHINE LEARNING, 2014, 94 (02) : 233 - 259
  • [5] A Comprehensive Survey of Graph Embedding: Problems, Techniques, and Applications
    Cai, HongYun
    Zheng, Vincent W.
    Chang, Kevin Chen-Chuan
    [J]. IEEE TRANSACTIONS ON KNOWLEDGE AND DATA ENGINEERING, 2018, 30 (09) : 1616 - 1637
  • [6] Cai L., 2018, P 2018 C N AM CHAPT, P1470, DOI DOI 10.18653/V1/N18-1133
  • [7] Unifying Knowledge Graph Learning and Recommendation: Towards a Better Understanding of User Preferences
    Cao, Yixin
    Wang, Xiang
    He, Xiangnan
    Hu, Zikun
    Chua, Tat-Seng
    [J]. WEB CONFERENCE 2019: PROCEEDINGS OF THE WORLD WIDE WEB CONFERENCE (WWW 2019), 2019, : 151 - 161
  • [8] Representation Learning for Attributed Multiplex Heterogeneous Network
    Cen, Yukuo
    Zou, Xu
    Zhang, Jianwei
    Yang, Hongxia
    Zhou, Jingren
    Tang, Jie
    [J]. KDD'19: PROCEEDINGS OF THE 25TH ACM SIGKDD INTERNATIONAL CONFERENCCE ON KNOWLEDGE DISCOVERY AND DATA MINING, 2019, : 1358 - 1368
  • [9] Heterogeneous Network Embedding via Deep Architectures
    Chang, Shiyu
    Han, Wei
    Tang, Jiliang
    Qi, Guo-Jun
    Aggarwal, Charu C.
    Huang, Thomas S.
    [J]. KDD'15: PROCEEDINGS OF THE 21ST ACM SIGKDD INTERNATIONAL CONFERENCE ON KNOWLEDGE DISCOVERY AND DATA MINING, 2015, : 119 - 128
  • [10] Chen MH, 2017, PROCEEDINGS OF THE TWENTY-SIXTH INTERNATIONAL JOINT CONFERENCE ON ARTIFICIAL INTELLIGENCE, P1511