移动智能计算团队的论文“Spatial-Temporal Interval Aware Individual Future Trajectory Prediction”被IEEE TKDE(CCF A)录用。论文的第一作者为必威西汉姆体育官网移动智能计算团队数据挖掘组2022级博士研究生姜毅恒,通讯作者为团队内徐原博副教授,指导教师为杨永健教授,作者还包括王恩教授。
本文提出了一种针对时空序列数据的时空间隔感知预测模型STiSAN+,可同时应用于兴趣点推荐和用户轨迹预测两大场景。STiSAN+模型主要包含旋转时间感知编码器(Rotary Time Aware Position Encoder, RoTAPE),多头间隔感知注意力机制(Multi-Head Interval Aware Attention Block, MHIAAB)以及时空相关性记忆单元(Spatial-Temporal Relation Memory, STR Memory)。其中,RoTAPE从时间角度同时分析用户行为的长期模式和周期性规律;MHIAAB将时空间隔作为先验知识引入注意力机制;STR Memory对未来序列中的时空间隔进行预测,以辅助轨迹预测任务。在6个真实世界基准数据集上进行了验证。大量实验结果表明,STiSAN+在兴趣点推荐、未来轨迹预测两个任务上,均优于目前最先进的模型。此外,文中还验证了RoTAPE、MHIAAB可作为两个“即插即用”的模块,为基于注意力机制的序列模型带来性能提升。
IEEE TKDE全称IEEE Transactions on Knowledge and Data Engineering,是中国计算机学会推荐的A类顶级国际期刊,在数据挖掘领域享有极高的学术声誉,影响因子指数8.9。