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

• 作者: 吴运杰 谷丽琼 逄金辉
• 来源期刊: 系统工程
• 学科: 经济
• 关键词: 股票预测 GRU 模型 Attention 机制 特征处理
• 年，卷（期）: 2020,（5）
• 所属期刊栏目: 金融系统工程
• 页码范围: 134-140
• 页数: 7页
• 分类号: F830
• 语种: 中文

摘要

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

作者信息

序号	姓名	单位	发文数	被引次数	H指数	G指数
1	逄金辉		17	61	5.0	7.0
2	谷丽琼		1	0	0.0	0.0
3	吴运杰		1	0	0.0	0.0