移動智能計算團隊的論文“Spatial-Temporal Interval Aware Individual Future Trajectory Prediction”被IEEE TKDE(CCF A)錄用。論文的第一作者為伟德国际BETVlCTOR移動智能計算團隊數據挖掘組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。