基于注意力机制循环神经网络的时间序列预测方法及系统

基本信息

申请号 CN202010723438.8 申请日 -
公开(公告)号 CN111860785A 公开(公告)日 2020-10-30
申请公布号 CN111860785A 申请公布日 2020-10-30
分类号 G06N3/04(2006.01)I;G06N3/08(2006.01)I 分类 计算;推算;计数;
发明人 吴雪敏;潘炎;刘冶 申请(专利权)人 火烈鸟网络(广州)股份有限公司
代理机构 广州骏思知识产权代理有限公司 代理人 中山大学;火烈鸟网络(广州)股份有限公司
地址 510275广东省广州市海珠区新港西路135号
法律状态 -

摘要

摘要 本发明涉及一种基于注意力机制循环神经网络的时间序列预测方法及系统,包括:构建一数据集,从所述数据集中确认驱动序列和目标序列;构建基于注意力机制的循环神经网络模型,所述网络模型在驱动序列之间、驱动序列与目标序列之间、时间三个维度中运用注意力机制;将所述数据集划分出训练集、验证集和测试集,对模型参数进行定义,对所述网络模型进行训练,通过以均方误差为目标函数的反向传播算法来对所述参数进行学习;以及将所述测试集输入到训练后的所述网络模型中进行预测。本发明有效解决了由于输入序列过长而出现的信息丢失导致性能瓶颈的问题,同时有效捕捉较长的时间内序列之间的相关性,从而实现对时间序列的更精确的预测。