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 条
  • [51] Reinforced Negative Sampling over Knowledge Graph for Recommendation
    Wang, Xiang
    Xu, Yaokun
    He, Xiangnan
    Cao, Yixin
    Wang, Meng
    Chua, Tat-Seng
    [J]. WEB CONFERENCE 2020: PROCEEDINGS OF THE WORLD WIDE WEB CONFERENCE (WWW 2020), 2020, : 99 - 109
  • [52] KGAT: Knowledge Graph Attention Network for Recommendation
    Wang, Xiang
    He, Xiangnan
    Cao, Yixin
    Liu, Meng
    Chua, Tat-Seng
    [J]. KDD'19: PROCEEDINGS OF THE 25TH ACM SIGKDD INTERNATIONAL CONFERENCCE ON KNOWLEDGE DISCOVERY AND DATA MINING, 2019, : 950 - 958
  • [53] Neural Graph Collaborative Filtering
    Wang, Xiang
    He, Xiangnan
    Wang, Meng
    Feng, Fuli
    Chua, Tat-Seng
    [J]. PROCEEDINGS OF THE 42ND INTERNATIONAL ACM SIGIR CONFERENCE ON RESEARCH AND DEVELOPMENT IN INFORMATION RETRIEVAL (SIGIR '19), 2019, : 165 - 174
  • [54] Wang Z, 2014, AAAI CONF ARTIF INTE, P1112
  • [55] Yang Bishan, 2014, P 3 INT C LEARN REPR
  • [56] Graph Convolutional Neural Networks for Web-Scale Recommender Systems
    Ying, Rex
    He, Ruining
    Chen, Kaifeng
    Eksombatchai, Pong
    Hamilton, William L.
    Leskovec, Jure
    [J]. KDD'18: PROCEEDINGS OF THE 24TH ACM SIGKDD INTERNATIONAL CONFERENCE ON KNOWLEDGE DISCOVERY & DATA MINING, 2018, : 974 - 983
  • [57] Yu X., 2013, INT JOINT C ART INT, V27
  • [58] Personalized Entity Recommendation: A Heterogeneous Information Network Approach
    Yu, Xiao
    Ren, Xiang
    Sun, Yizhou
    Gu, Quanquan
    Sturt, Bradley
    Khandelwal, Urvashi
    Norick, Brandon
    Han, Jiawei
    [J]. WSDM'14: PROCEEDINGS OF THE 7TH ACM INTERNATIONAL CONFERENCE ON WEB SEARCH AND DATA MINING, 2014, : 283 - 292
  • [59] Collaborative Knowledge Base Embedding for Recommender Systems
    Zhang, Fuzheng
    Yuan, Nicholas Jing
    Lian, Defu
    Xie, Xing
    Ma, Wei-Ying
    [J]. KDD'16: PROCEEDINGS OF THE 22ND ACM SIGKDD INTERNATIONAL CONFERENCE ON KNOWLEDGE DISCOVERY AND DATA MINING, 2016, : 353 - 362
  • [60] Pairwise Link Prediction Model for Out of Vocabulary Knowledge Base Entities
    Zhang, Richong
    Mensah, Samuel
    Kong, Fanshuang
    Hu, Zhiyuan
    Mao, Yongyi
    Liu, Xudong
    [J]. ACM TRANSACTIONS ON INFORMATION SYSTEMS, 2020, 38 (04)