[发明专利]基于Transformer的多任务交通流预测方法、装置、终端及存储介质

摘要

本发明公开了一种基于Transformer的多任务交通流预测方法、装置及终端，本发明方案通过设置一个编码器和两个解码器，使用两个编码器分别处理交通流量任务和交通速度任务，使用编码器提取历史交通数据的历史时空特征并输入至每个解码器中，以增强各个解码器中输入的实时交通数据的特征；而且每个解码器还从另一个解码器中层次性抽取空间特征和节点特征以与自身特征融合和互补，从而通过一个预测任务来增强另一个预测任务的特征，实现特征上相互促进，提高预测结果的精度。