移动智能计算团队的论文“Spatial-Temporal Interval Aware Individual Future Trajectory Prediction”被IEEE TKDE(CCF A)录用。论文的第一作者为304永利集团官网入口移动智能计算团队数据挖掘组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。