时空记忆增强多级注意网络

通过时空记忆网络和多级注意机制,显著提升对长程和短程时空依赖的建模能力,实现更精准的交通预测。

Yan, Liu · 郭斌 · Jingxiang, Meng · 张大庆 · 於志文

IEEE Transactions on Knowledge and Data Engineering 2023

技术优势

同时捕捉长短程依赖

多级注意网络在不同空间(网格、区域)和时间(日、周)尺度上显式建模,避免顾此失彼。

融合外部因素影响

外部组件将外部因素与空间嵌入结合,高效生成位置感知的外部影响,提升预测准确性。

端到端优化提升性能

联合对比目标与监督目标进行端到端训练,进一步提升模型表现。

应用场景