Journal of Frontiers of Computer Science and Technology ›› 2024, Vol. 18 ›› Issue (1): 231-243.DOI: 10.3778/j.issn.1673-9418.2208098

• Artificial Intelligence·Pattern Recognition • Previous Articles     Next Articles

Integrating Behavioral Dependencies into Multi-task Learning for Personalized Recommendations

GU Junhua, LI Ningning, WANG Xinxin, ZHANG Suqi   

  1. 1. School of Artificial Intelligence and Data Science, Hebei University of Technology, Tianjin 300401, China
    2. Hebei Province Key Laboratory of Big Data Computing (Hebei University of Technology), Tianjin 300401, China
    3. School of Science, Tianjin University of Commerce, Tianjin 300134, China
    4. School of Information Engineering, Tianjin University of Commerce, Tianjin 300134, China
  • Online:2024-01-01 Published:2024-01-01

将行为依赖融入多任务学习的个性化推荐模型

顾军华,李宁宁,王鑫鑫,张素琪   

  1. 1. 河北工业大学 人工智能与数据科学学院,天津 300401
    2. 河北省大数据计算重点实验室(河北工业大学),天津 300401
    3. 天津商业大学 理学院,天津 300134
    4. 天津商业大学 信息工程学院,天津 300134

Abstract: The introduction of multiple types of behavioral data alleviates the data sparsity and cold-start problems of collaborative filtering algorithms, which is widely studied and applied in the field of recommendations. Although great progress has been made in the current research on multi-behavior recommendation, the following problems still exist: failure to comprehensively capture the complex dependencies between behaviors; ignoring the relevance of behavior features to users and items, and the recommendation results are biased. This results in the learned feature vectors failing to accurately represent the user??s interest preferences. To solve the above problems, a person-alized recommendation model (BDMR) that integrates behavioral dependencies into multi-task learning is proposed, and in this paper, the complex dependencies between behaviors are divided into feature relevance and temporal relevance. Firstly, the user personalized behavior vector is set, and multiple interaction graphs are processed with graph neural networks which combine user, item and behavior features to aggregate higher-order neighborhood information, and attention mechanism is combined to learn feature relevance among behaviors. Secondly, the interaction sequence composed of behavior features and item features is input into a long and short-term memory network to capture the temporal relevance among behaviors. Finally, personalized behavior vectors are integrated into a multi-task learning framework to obtain more accurate user, behavior and item features. To verify the perf-ormance of this model, experiments are conducted on three real datasets. On the Yelp dataset, compared with the optimal baseline, HR and NDCG are improved by 1.5% and 2.9% respectively. On the ML20M dataset, HR and NDCG are increased by 2.0% and 0.5% respectively. On the Tmall dataset, HR and NDCG are improved by 25.6% and 30.2% respectively. Experimental results show that the model proposed in this paper is superior to baselines.

Key words: multi-behavior recommendation, graph neural networks, recurrent neural networks, multi-task learning framework

摘要: 多种类型行为数据的引入缓解了协同过滤算法存在的数据稀疏和冷启动问题,在推荐领域被广泛研究和应用。尽管当前对多行为推荐的研究已经取得很大进展,但仍然存在以下问题:未能全面捕获行为之间复杂的依赖关系;忽略了行为特征与用户和项目的相关性。这导致学习到的特征向量无法准确表达用户的兴趣偏好,使得推荐结果存在偏差。为了解决以上问题,提出了将行为依赖融入多任务学习的个性化推荐模型(BDMR),将行为之间复杂的依赖关系分为特征相关性和时序相关性。首先,设置用户个性化行为向量,利用图神经网络处理多个单行为交互图,联合用户、项目和行为特征聚合高阶邻域信息,结合注意力机制学习行为之间的特征相关性;其次,将行为特征和项目特征构成的交互序列输入长短期记忆网络,捕获行为之间的时序相关性;最后,将个性化行为向量融入多任务学习框架获取更加准确的用户、行为和项目特征。为了验证提出模型的性能,在三个真实数据集上进行对比实验,在Yelp数据集上,相较于最优基线,HR和NDCG分别提升了1.5%和2.9%;在ML20M数据集上,HR和NDCG分别提升了2.0%和0.5%;在Tmall数据集上,HR和NDCG分别提升了25.6%和30.2%。实验结果表明,该模型优于其他的基准模型。

关键词: 多行为推荐, 图神经网络, 循环神经网络, 多任务学习框架