系统工程2020,Vol.38Issue(5) :134-140.

基于Attention机制的GRU股票预测模型

An Attention-based GRU Model for Stock Predicting

谷丽琼 吴运杰 逄金辉
系统工程2020,Vol.38Issue(5) :134-140.

基于Attention机制的GRU股票预测模型

An Attention-based GRU Model for Stock Predicting

谷丽琼 1吴运杰 2逄金辉3
扫码查看

作者信息

  • 1. 天津市电子信息技师学院 电子信息工程系,天津 300350
  • 2. 天津大学 智能与计算学部,天津 300072
  • 3. 北京理工大学 计算机科学与技术学院,北京 100081
  • 折叠

摘要

股市行情随众多影响因子不断变化,现有基于时间序列预测的方法难以捕捉该非线性动力系统的复杂变化规律,预测效果并不理想.本文提出基于Attention机制的GRU预测模型,捕捉关键时间点特征信息以解决对时间特征不敏感导致预测精度不高的问题以提升预测精确度.首先使用LSTM和GRU构建基础预测模型;然后对输入特征进行统计处理和筛选,选取更能反映股价变动规律的特征;最后基于编码器-解码器框架,在GRU模型上加入Attention机制,使模型聚焦于重要时间点的股票特征信息.本文在科大讯飞股票数据上进行实验,实验结果表明基于Attention机制的GRU模型在MAPE,RMSE,R2 score三个评价指标上均优于其他模型,Attention机制能够捕捉重要时间点局部特征,对预测模型的优化是可行和有效的.

关键词

股票预测/GRU/模型/Attention/机制/特征处理

引用本文复制引用

出版年

2020
系统工程
湖南省系统工程与管理学会

系统工程

CSTPCD北大核心
影响因子:0.721
ISSN:1001-4098
被引量16
参考文献量10
段落导航相关论文