首页 | 本学科首页   官方微博 | 高级检索  
     

基于自监督学习的社交网络用户轨迹预测模型
引用本文:代雨柔,杨庆,张凤荔,周帆.基于自监督学习的社交网络用户轨迹预测模型[J].计算机应用,2021,41(9):2545-2551.
作者姓名:代雨柔  杨庆  张凤荔  周帆
作者单位:1. 电子科技大学 信息与软件工程学院, 成都 610054;2. 中国电子科技集团公司第十研究所, 成都 610036
基金项目:国家自然科学基金面上项目(62072077)。
摘    要:针对当前用户轨迹数据建模中存在的签到点稀疏性、长时间依赖性和移动模式复杂等问题,提出基于自监督学习的社交网络用户轨迹预测模型SeNext,对用户轨迹进行建模和训练来预测用户的下一个兴趣点(POI)。首先,使用数据增强的方式来丰富训练数据样本,以解决数据不足及个别用户足迹太少导致的模型泛化能力不足的问题;其次,将循环神经网络(RNN)、卷积神经网络(CNN)和注意力机制分别用于当前轨迹和历史轨迹的建模中,以此从高维稀疏的数据中提取有用的表示,用来匹配用户过去最相似的移动方式。最后,通过结合自监督学习并引入对比损失优化噪声对比估计(InfoNCE),SeNext在潜在空间学习隐含表示来预测用户的下一个POI。实验结果表明,在纽约数据集上,SeNext比最新的VANext(Variational Attention based Next)模型的预测准确度在Top@1上提高了11.10%左右。

关 键 词:轨迹预测  自监督学习  对比学习  注意力机制  深度学习  
收稿时间:2020-11-26
修稿时间:2021-01-26

Trajectory prediction model of social network users based on self-supervised learning
DAI Yurou,YANG Qing,ZHANG Fengli,ZHOU Fan.Trajectory prediction model of social network users based on self-supervised learning[J].journal of Computer Applications,2021,41(9):2545-2551.
Authors:DAI Yurou  YANG Qing  ZHANG Fengli  ZHOU Fan
Affiliation:1. School of Information and Software Engineering, University of Electronic Science and Technology of China, Chengdu Sichuan 610054, China;2. The 10 th Research Institute of China Electronics Technology Group Corporation, Chengdu Sichuan 610036, China
Abstract:Aiming at the existing problems in user trajectory data modeling such as the sparsity of check-in points, long-term dependencies and complex moving patterns, a social network user trajectory prediction model based on self-supervised learning, called SeNext, was proposed to model and train the user trajectory to predict the next Point Of Interest (POI) of the user. First, data augmentation was utilized to expand the training trajectory samples, which solved the problem of the deficiency of model generalization capability caused by insufficient data and too few footprints of some users. Second, Recurrent Neural Network (RNN), Convolutional Neural Network (CNN) and attention mechanism were adopted into the modeling of current and historical trajectories respectively, so as to extract effective representations from high-dimensional sparse data to match the most similar moving patterns of users in the past. Finally, SeNext learned the implicit representations in the latent space by combining self-supervised learning and introducing contrastive loss Noise Contrastive Estimation (InfoNCE) to predict the next POI of the user. Experimental results show that compared to the state-of-the-artVariational Attention based Next (VANext)model, SeNext improves the prediction accuracy about 11% on Top@1.
Keywords:trajectory prediction  self-supervised learning  contrastive learning  attention mechanism  deep learning  
本文献已被 万方数据 等数据库收录!
点击此处可从《计算机应用》浏览原始摘要信息
点击此处可从《计算机应用》下载全文
设为首页 | 免责声明 | 关于勤云 | 加入收藏

Copyright©北京勤云科技发展有限公司  京ICP备09084417号