基于注意力机制的交通流量预测方法、装置、设备及介质

本发明涉及智慧交通领域和人工智能领域，公开了基于注意力机制的交通流量预测方法、装置、设备及介质，方法包括：通过交通流量预测模型中的时空编码器和多头自注意力机制，对预设交通流图和增强后的预设交通流图进行处理，得到第一正样本对、第一负样本对、第二正样本对、第二负样本对；将第一正样本对的特征向量、第二正样本对的特征向量进行拼接，得到第一拼接特征，基于第一拼接特征，确定第三损失值；将第一损失值、第二损失值、第三损失值相加，得到目标损失值；选取使用调整后的模型参数的交通流量预测模型作为目标模型，通过目标模型，生成目标交通流图在下一时刻的交通流量预测值。本发明有利于提高交通流量预测模型的预测性能。