[发明专利]基于注意力改进的交通预测方法及计算机介质

摘要

本发明属于智能交通技术领域，具体公开了一种基于注意力改进的交通预测方法及计算机介质，该方法利用全连接层计算注意力参数，随机采样注意力参数，计算每组采样的注意力参数中的分布与平均分布的KL散度值，根据每组注意力参数中的KL散度值，获取注意力参数的概率分布与均匀分布之间的差值Top‑u，根据差值Top‑u，获取差值最大的多个注意力参数，使用一个4维的张量Φ∈RB×T×N×N对获取的注意力参数同时进行空间和时间上的卷积，将卷积后的注意力参数输入全连接层，输出预测交通流量。采用本技术方案，在使用注意力获取动态的拉普拉斯矩阵，降低计算时间复杂度和空间复杂度，增加模型的感受野。