一种基站流量预测方法和装置

本发明提供一种基站流量预测方法和装置，包括：获取流量时空序列和基站流量预测模型，流量时空序列包括第一和第二流量序列；模型中的第一支路基于第一流量序列提取相邻基站流量间的空间相关性以及单个基站流量序列的时间相关性，并在时间维度上使用自注意力机制建模同一个基站流量在时间上的突发性变化，在空间维度上使用多头注意力机制建模同一时段不同基站间流量的突变性分布规律得到第一特征张量；第二支路提取第二流量序列的时间相关性得到第二特征张量；基于输出层对第一和第二特征张量进行特征融合，输出对各基站待预测时段的流量预测值。能够更全面地捕捉基站流量的变化规律，更有效地处理时空序列数据中的复杂关系，提高预测准确性。