计算机应用

作者：代雨柔, 杨庆, 张凤荔, 周帆

单位：1. 电子科技大学 信息与软件工程学院, 成都 610054;2. 中国电子科技集团公司第十研究所, 成都 610036

关键词：轨迹预测,自监督学习,对比学习,注意力机制,深度学习

基金：国家自然科学基金面上项目（62072077）。

针对当前用户轨迹数据建模中存在的签到点稀疏性、长时间依赖性和移动模式复杂等问题，提出基于自监督学习的社交网络用户轨迹预测模型SeNext，对用户轨迹进行建模和训练来预测用户的下一个兴趣点（POI）。首先，使用数据增强的方式来丰富训练数据样本，以解决数据不足及个别用户足迹太少导致的模型泛化能力不足的问题；其次，将循环神经网络（RNN）、卷积神经网络（CNN）和注意力机制分别用于当前轨迹和历史轨迹的建模中，以此从高维稀疏的数据中提取有用的表示，用来匹配用户过去最相似的移动方式。最后，通过结合自监督学习并引入对比损失优化噪声对比估计（InfoNCE），SeNext在潜在空间学习隐含表示来预测用户的下一个POI。实验结果表明，在纽约数据集上，SeNext比最新的VANext（Variational Attention based Next）模型的预测准确度在Top@1上提高了11.10%左右。

来源：2021年第9期

《计算机应用》期刊编辑部

查看计算机应用杂志2021年第9期