一种基于简化注意力机制的多节点交通流量预测方法

本发明公开了一种基于简化注意力机制的多节点交通流量预测方法，属于交通流量预测领域，该方法包括：采用多节点交通流量预测模型，以上一个T时段的节点交通流量数据，预测未来T时段节点的交通流量数据；所述多节点交通流量预测模型包括数据嵌入层、简化注意力层和输出层；所述简化注意力层，包括异构注意力分离层、简化注意力模块和异构注意力聚合层；所述简化注意力模块包括TSA模块、SSA模块和GSA模块；本发明采用了简化注意力机制可以加快注意力的计算速度，其中SSA模块、GSA模块可以更好的提取节点之间地理上的相似性依赖，以及变化模式上的相似性依赖；从而有效地提高模型预测的精准度。