基于粒子群优化LSTM的股票预测模型

doi:10.13700/j.bh.1001-5965.2019.0388

北京航空航天大学学报 ›› 2019, Vol. 45 ›› Issue (12): 2533-2542.doi: 10.13700/j.bh.1001-5965.2019.0388

基于粒子群优化LSTM的股票预测模型

宋刚^1,2, 张云峰^1,2, 包芳勋³, 秦超^1,2

1. 山东财经大学计算机科学与技术学院, 济南 250014;
2. 山东财经大学山东省数字媒体技术重点实验室, 济南 250014;
3. 山东大学数学学院, 济南 250100

收稿日期:2019-07-10 出版日期:2019-12-20 发布日期:2019-12-31
通讯作者: 张云峰 E-mail:yfzhang@sdufe.edu.cn
作者简介:宋刚男,硕士研究生。主要研究方向:机器学习、数据分析;张云峰男,博士,教授,博士生导师。主要研究方向:计算机视觉、数据分析。
基金资助:
国家自然科学基金（61672018，61772309）；国家自然科学基金-浙江两化融合联合基金（U1609218）；山东省重点研发计划（2016GSF120013，2017GGX10109，2018GGX101013）；山东省高等学校优势学科人才队伍培育计划；山东省自然科学杰出青年（ZR2018JL022）；山东省自然科学基金（ZR2019MF051）

Stock prediction model based on particle swarm optimization LSTM

SONG Gang^1,2, ZHANG Yunfeng^1,2, BAO Fangxun³, QIN Chao^1,2

1. School of Computer Science and Technology, Shandong University of Finance and Economics, Jinan 250014, China;
2. Shandong Key Laboratory of Digital Media Technology, Shandong University of Finance and Economics, Jinan 250014, China;
3. School of Mathematics, Shandong University, Jinan 250100, China

Received:2019-07-10 Online:2019-12-20 Published:2019-12-31
Supported by:
National Natural Science Foundation of China (61672018, 61772309); National Natural Science Foundation of China-Zhejiang Two Integration Joint Fund (U1609218); Key Research and Development Project of Shandong Province (2016GSF120013, 2017GGX10109, 2018GGX101013); Fostering Project of Dominant Discipline and Talent Team of Shandong Province Higher Education Institutions; Natural Science Foundation for Excellent Youth of Shandong Province (ZR2018JL022); Natural Science Foundation of Shandong Province (ZR2019MF051)

摘要/Abstract

摘要： 为了提高股票时间序列预测精度，增强预测模型结构参数可解释性，提出一种基于自适应粒子群优化（PSO）的长短期记忆（LSTM）股票价格预测模型（PSO-LSTM），该模型在LSTM模型的基础上进行改进和优化，因此擅长处理具有长期依赖关系的、复杂的非线性问题。通过自适应学习策略的PSO算法对LSTM模型的关键参数进行寻优，使股票数据特征与网络拓扑结构相匹配，提高股票价格预测精度。实验分别以沪市、深市、港股股票数据构建了PSO-LSTM模型，并对该模型的预测结果与其他预测模型进行比较分析。结果表明，基于自适应PSO的LSTM股票价格预测模型不但提高了预测准确度，而且具有普遍适用性。

关键词: 粒子群优化(PSO), LSTM神经网络, 自适应, 股票价格预测, 预测精度

Abstract: This paper proposes a stock price prediction model based on particle swarm optimization long short-term memory (PSO-LSTM). This model improves and optimizes the LSTM model, which makes it more appropriate for analyzing relationships such as long-term dependency and for solving complex nonlinear problems. Through finding the key parameters in LSTM model by the PSO algorithm with adaptive learning strategy, the stock data feature matches the network topology structure, and the model's prediction accuracy of stock price is improved. In the experiment, PSO-LSTM models are constructed respectively based on the stock datasets from Shanghai, Shenzhen and Hong Kong, and then they are compared to other prediction models. The comparison results show that the PSO-LSTM stock price prediction model achieves higher prediction accuracy and has general applicability.

Key words: particle swarm optimization (PSO), LSTM neural network, adaptive, stock price forecast, prediction accuracy

中图分类号:

TP183

宋刚, 张云峰, 包芳勋, 秦超. 基于粒子群优化LSTM的股票预测模型[J]. 北京航空航天大学学报, 2019, 45(12): 2533-2542.

SONG Gang, ZHANG Yunfeng, BAO Fangxun, QIN Chao. Stock prediction model based on particle swarm optimization LSTM[J]. JOURNAL OF BEIJING UNIVERSITY OF AERONAUTICS AND A, 2019, 45(12): 2533-2542.

参考文献

[1] SI Y W,YIN J.OBST-based segmentation approach to financial time series[J].Engineering Applications of Artificial Intelligence,2013,26(10):2581-2596.
[2] COSTA-JUSSÀ,MARTA R,ALLAUZEN A,et al.Introduction to the special issue on deep learning approaches for machine translation[J].Computer Speech & Language,2017,46(11):367-373.
[3] FAYEK H M,LECH M,CAVEDON L.Evaluating deep learning architectures for speech emotion recognition[J].Neural Network,2017,92(2):60-68.
[4] XING J L,LI K,HU W M,et al.Diagnosing deep learning mod-els for high accuracy age estimation from a single image[J].Pattern Recognition 2017,66(1):106-116.
[5] HOCHREITER S,SCHMIDHUBER J.Long short-term memory[J].Neural Computation,1997,9(8):1735-1780.
[6] DEIHIMI A,ORANG O,SHOWKATI H.Short-term electric load and temperature forecasting using wavelet echo state networks with neural reconstruction[J].Energy,2013,57(3):382-401.
[7] CAI M,LIU J.Maxout neurons for deep convolutional and LSTM neural networks in speech recognition[J].Speech Communication,2016,77(3):53-64.
[8] ZHOU C,SUN C,LIU Z,et al.A C-LSTM neural network for text classification[J].Computer Science,2015,1(4):39-44.
[9] CHEN K,ZHOU Y,DAI F.A LSTM-based method for stock returns prediction:A case study of China stock market[C]//IEEE International Conference on Big Data.Piscataway,NJ:IEEE Press,2015:2823-2824.
[10] JIA H J.Investigation into the effectiveness of long short term memory networks for stock price prediction[EB/OL].(2016-03-25)[2019-07-10].https://arxu-org/abs/1603.07893v1.
[11] NELSON D M Q,PEREIRA A C M,DEOLIVEIRA R A,et al.Stock market's price movement prediction with LSTM neural networks[C]//International Joint Conference on Neural Networks.Piscataway,NJ:IEEE Press,2017:1419-1426.
[12] FISCHER T,KRAUSS C.Deep learning with long short-term memory networks for financial market predictions[J].European Journal of Operational Research,2018,270(2):654-669.
[13] RODRIGUEZ A, LAIO A.Clustering by fast search and find of density peaks[J].Science,2014,344(6191):1492-1496.
[14] 胡旺,李志蜀.一种更简化而高效的粒子群优化算法[J].软件学报,2007,18(4):861-868.HU W,LI Z S.A simpler and more effective particle swarm optimization algorithm[J].Journal of Software,2007,18(4):861-868(in Chinese).
[15] ADEBIYI A A,ADEWUMI A O,AYO C K.Stock price prediction using the ARIMA model[C]//UKSIM-AMSS International Conference on Computer Modelling & Simulation.Piscataway,NJ:IEEE Press,2015:106-112.
[16] LIN Y,GUO H,HU J.An SVM-based approach for stock market trend prediction[C]//International Joint Conference on Neural Networks.Piscataway,NJ:IEEE Press,2013:1-7.
[17] USMANI M,ADIL S H,RAZA K,et al.Stock market prediction using machine learning techniques[C]//International Conference on Computer & Information Sciences.Piscataway,NJ:IEEE Press,2016:322-327.
[18] RATHER A M,AGARWAL A,SASTRY V.Recurrent neural network and a hybrid model for prediction of stock returns[J].Expert Systems with Applications,2015,42(6):3234-3241.
[19] MCNALLY S,ROCHE J,CATON S.Predicting the price of Bitcoin using Machine Learning[C]//26th Euromicro International Conference on Parallel Distributed and Network-based Processing.Piscataway,NJ:IEEE Press,2018:339-343.

基于粒子群优化LSTM的股票预测模型

Stock prediction model based on particle swarm optimization LSTM

PDF (PC)

可视化

摘要/Abstract

引用本文

使用本文

参考文献

相关文章 15

编辑推荐

Metrics

本文评价

[1]	刘畅, 杨锁昌, 汪连栋, 张宽桥. 基于快速自适应超螺旋算法的制导律[J]. 北京航空航天大学学报, 2019, 45(7): 1388-1397.
[2]	赵国荣, 李晓宝, 刘帅, 韩旭. 自适应非奇异快速终端滑模固定时间收敛制导律[J]. 北京航空航天大学学报, 2019, 45(6): 1059-1070.
[3]	李琳, 肖佳栋, 张铁, 肖蒙. 基于自适应迭代的机器人曲面恒力跟踪[J]. 北京航空航天大学学报, 2019, 45(4): 641-649.
[4]	梁小辉, 王青, 董朝阳. 基于切换系统的变体飞行器鲁棒自适应控制[J]. 北京航空航天大学学报, 2019, 45(3): 538-545.
[5]	巩祥瑞, 吕震宙, 孙天宇, 张雷雷, 封雷. 一种新的矩独立重要性测度分析方法及高效算法[J]. 北京航空航天大学学报, 2019, 45(2): 283-290.
[6]	刘炜伦, 张衡阳, 郑博, 高维廷. 蜂群无人机自组网多优先级自适应退避算法[J]. 北京航空航天大学学报, 2019, 45(2): 325-332.
[7]	肖楚晗, 李炯, 雷虎民, 王华吉. 基于AVSIMM算法的高超声速再入滑翔目标跟踪[J]. 北京航空航天大学学报, 2019, 45(2): 413-421.
[8]	黄樱, 牛保宁, 关虎, 张树武. 基于图像纹理的自适应水印算法[J]. 北京航空航天大学学报, 2019, 45(12): 2403-2414.
[9]	翟宇轩, 刘怡桑, 徐艺文, 陈忠辉, 房颖, 赵铁松. 基于HTTP自适应流媒体传输的3D视频质量评价[J]. 北京航空航天大学学报, 2019, 45(12): 2456-2462.
[10]	王尔申, 杨迪, 王传云, 曲萍萍, 庞涛, 蓝晓宇. PSO选星算法参数分析与改进[J]. 北京航空航天大学学报, 2019, 45(11): 2133-2138.
[11]	刘勇智, 李杰, 鄯成龙. 基于最优角度自适应TSF的SRM直接瞬时转矩控制[J]. 北京航空航天大学学报, 2019, 45(11): 2152-2159.
[12]	宋鑫, 郑冠男, 杨国伟, 姜倩. 几何不确定性区间分析及鲁棒气动优化设计[J]. 北京航空航天大学学报, 2019, 45(11): 2217-2227.
[13]	任鹏飞, 王洪波, 周国峰. 基于自适应伪谱法的高超声速飞行器再入轨迹优化[J]. 北京航空航天大学学报, 2019, 45(11): 2257-2265.
[14]	呼延嘉玥, 徐立军, 李小路. 自研激光雷达三维点云配准技术[J]. 北京航空航天大学学报, 2019, 45(10): 2099-2107.
[15]	张建军, 刘卫东, 高立娥, 李乐, 李泽宇. 水下机械手不确定遥操作自适应双边控制[J]. 北京航空航天大学学报, 2018, 44(9): 1918-1925.