Multi-task Learning Model based on Multiple Characteristics and Multiple Interests for CTR prediction

被引:0
|
作者
Xie, Yufeng [1 ]
Li, Mingchu [1 ]
Lu, Kun [1 ]
Shah, Syed Bilal Hussain [2 ]
Zheng, Xiao [3 ]
机构
[1] Dalian Univ Technol, Dalian, Peoples R China
[2] Manchester Metropolitan Univ, Manchester, Lancs, England
[3] Shandong Univ Technol, Sch Comp Sci & Technol, Zibo, Peoples R China
基金
美国国家科学基金会;
关键词
Recommendation algorithm; CTR prediction; multi-interest; multi-task learning;
D O I
10.1109/DSC54232.2022.9888898
中图分类号
TP3 [计算技术、计算机技术];
学科分类号
0812 ;
摘要
In the era of big data, the acquisition and utilization of information becomes difficult with the skyrocketing amount of data. It is often difficult for ordinary users to find the information or items they need, and personalized recommendation systems can solve this problem well. Currently, recommendation systems increasingly adopt models based on deep learning. The most critical issue in using deep learning for recommendation systems is how to use neural networks to accurately learn user representation vectors and item representation vectors. Many deep learning models used a single vector to represent users, but users' interests were often diverse. Therefore, some researchers consider using multiple vectors to represent user interests, and each interest vector corresponds to a category of items. This method sounds more scientific. However, these models still have problems. Their interpretation of user interests stays at the item level, and does not go deep into the item feature level. In order to solve this problem, we consider user interests from the perspective of item characteristics, and propose 3M (Multi-task, Multi-interest, Multi-feature) model. The 3M model trains multiple interest vectors for each user and extracts multiple characteristic vectors for each item at the same time, then uses a multi-task learning model to connect the characteristic vectors with the interest vectors and train them to obtain multiple interest scores. According to the multiple interest scores, the user click probability can be obtained. Experiments show that our model performs significantly better than the classic CTR(Click-Through Rate) prediction model on the experimental dataset.
引用
收藏
页数:7
相关论文
共 50 条
  • [1] Joint Prediction of Multiple Vulnerability Characteristics Through Multi-Task Learning
    Gong, Xi
    Xing, Zhenchang
    Li, Xiaohong
    Feng, Zhiyong
    Han, Zhuobing
    2019 24TH INTERNATIONAL CONFERENCE ON ENGINEERING OF COMPLEX COMPUTER SYSTEMS (ICECCS 2019), 2019, : 31 - 40
  • [2] ChroNet: A multi-task learning based approach for prediction of multiple chronic diseases
    Ruiwei Feng
    Yan Cao
    Xuechen Liu
    Tingting Chen
    Jintai Chen
    Danny Z. Chen
    Honghao Gao
    Jian Wu
    Multimedia Tools and Applications, 2022, 81 : 41511 - 41525
  • [3] ChroNet: A multi-task learning based approach for prediction of multiple chronic diseases
    Feng, Ruiwei
    Cao, Yan
    Liu, Xuechen
    Chen, Tingting
    Chen, Jintai
    Chen, Danny Z.
    Gao, Honghao
    Wu, Jian
    MULTIMEDIA TOOLS AND APPLICATIONS, 2022, 81 (29) : 41511 - 41525
  • [4] Personalized prediction for multiple chronic diseases by developing the multi-task Cox learning model
    Zhang, Shuaijie
    Yang, Fan
    Wang, Lijie
    Si, Shucheng
    Zhang, Jianmei
    Xue, Fuzhong
    PLOS COMPUTATIONAL BIOLOGY, 2023, 19 (09)
  • [5] Multi-Task Learning for Multiple Language Translation
    Dong, Daxiang
    Wu, Hua
    He, Wei
    Yu, Dianhai
    Wang, Haifeng
    PROCEEDINGS OF THE 53RD ANNUAL MEETING OF THE ASSOCIATION FOR COMPUTATIONAL LINGUISTICS AND THE 7TH INTERNATIONAL JOINT CONFERENCE ON NATURAL LANGUAGE PROCESSING, VOL 1, 2015, : 1723 - 1732
  • [6] Single Model for Influenza Forecasting of Multiple Countries by Multi-task Learning
    Murayama, Taichi
    Wakamiya, Shoko
    Aramaki, Eiji
    MACHINE LEARNING AND KNOWLEDGE DISCOVERY IN DATABASES, ECML PKDD 2021: APPLIED DATA SCIENCE TRACK, PT IV, 2021, 12978 : 335 - 350
  • [7] Toxicity Prediction in Cancer Using Multiple Instance Learning in a Multi-task Framework
    Li, Cheng
    Gupta, Sunil
    Rana, Santu
    Luo, Wei
    Venkatesh, Svetha
    Ashely, David
    Dinh Phung
    ADVANCES IN KNOWLEDGE DISCOVERY AND DATA MINING, PAKDD 2016, PT I, 2016, 9651 : 152 - 164
  • [8] Multiple object tracking method based on multi-task joint learning
    Qu Y.
    Li W.-H.
    Jilin Daxue Xuebao (Gongxueban)/Journal of Jilin University (Engineering and Technology Edition), 2023, 53 (10): : 2932 - 2941
  • [9] Multiple object tracking based on multi-task learning with strip attention
    Song, Yaoye
    Zhang, Peng
    Huang, Wei
    Zha, Yufei
    You, Tao
    Zhang, Yanning
    IET IMAGE PROCESSING, 2021, 15 (14) : 3661 - 3673
  • [10] Multi-task Learning Based on Multiple Data Sources for Cancer Detection
    Hong, Siyi
    2021 3RD INTERNATIONAL CONFERENCE ON MACHINE LEARNING, BIG DATA AND BUSINESS INTELLIGENCE (MLBDBI 2021), 2021, : 486 - 491