一种基于时空注意力机制的交通缺失数据补全方法

摘要

一种基于时空注意力机制的交通缺失数据补全方法，首先通过注意力机制的方式，捕获路网中所有的路段在当前时刻对路网交通状态的影响程度大小，并在不同时刻重新捕获空间相关性信息，提高数据补全的精度。其次，考虑交通数据的时序性，不同时刻的交通数据对当前时刻的数据影响程度是不一样的，通过时间注意力机制捕获这种不一致的时间相关性信息，在对当前缺失数据补全时保留最有效的信息，提升模型的补全效果。最后，在利用时空注意力机制捕获交通数据的时空相关性同时，考虑数据之间的相关性受到空间距离和时间间隔的增大而衰减，加入时空衰减性矩阵提高补全精度。本发明不仅大幅提高数据缺失率低的情况下补全精度，还提升在数据缺失率高的情况下的补全精度。