基于深度学习的股票趋势预测算法

摘要/Abstract

摘要： 针对股票趋势预测难的问题，提出一种利用CNN和LSTM进行特征提取，并结合注意力机制和对抗训练的股票趋势预测算法——AACL（Adversarial Attentive CNN-LSTM）算法。该算法利用CNN提取股票的整体趋势信息，LSTM提取股票的短期波动信息，并通过注意力机制将多个股票联系起来，捕捉股票之间的涨跌关系。算法还引入了对抗训练，通过对数据进行干扰，提高算法的鲁棒性。为了验证算法的有效性，在KDD17、ACL18和China50这3个数据集上进行实验，并与现有的算法进行比较，实验结果表明本文提出的算法可以获得最优的预测效果。

关键词: 神经网络, 注意力机制, 对抗训练, 股票趋势预测

Abstract: To improve the accuracy of stock movement prediction， this paper proposes a stock movement prediction algorithm AACL（Adversarial Attentive CNN-LSTM）which utilizes CNN and LSTM for feature extraction and combines attention mechanism and adversarial training. The algorithm uses CNN to extract the overall trend information of the stock， LSTM to extract the short-term fluctuation information of the stock， and connects multiple stocks through the attention mechanism to capture the rising and falling relationship between stocks. The algorithm also introduces adversarial training to improve the robustness of the algorithm by interfering the data. To verify the effectiveness of the AACL algorithm， experiments are carried out on three data sets KDD17， ACL18， and China50， and compared with existing algorithms. Experiments results show that the algorithm proposed in this paper can obtain the best result.

Key words: neural network, attention mechanism, adversarial training, stock movement prediction

周润佳. 基于深度学习的股票趋势预测算法[J]. 计算机与现代化, 2023, 0(01): 69-73.

ZHOU Run-jia. Stock Movement Prediction Algorithm Based on Deep Learning[J]. Computer and Modernization, 2023, 0(01): 69-73.

参考文献［27］

［1］	YU P F， YAN X S. Stock price prediction based on deep neural networks［J］. Neural Computing and Applications， 2020，32(6):1609-1628.
［2］	NELSON D M Q， PEREIRA A C M， OLIVEIRA R A D. Stock market＇s price movement prediction with LSTM neural networks［C］// 2017 International Joint Conference on Neural Networks (IJCNN). 2017:1419-1426.
［3］	LI H， SHEN Y Y， ZHU Y M. Stock price prediction using attention-based multi-input LSTM［C］// Proceedings of the 10th Asian Conference on Machine Learning. 2018:454-469.
［4］	FENG F L， CHEN H M， HE X N， et al. Enhancing stock movement prediction with adversarial training［C］// 28th International Joint Conference on Artificial Intelligence. 2019:5843-5849.
［5］	HOSEINZADE E， HARATIZADEH S. CNNpred: CNN-based stock market prediction using a diverse set of variables［J］. Expert Systems with Applications， 2019，129(S):273-285.
［6］	DING Q G， WU S F， SUN H， et al. Hierarchical multi-scale Gaussian transformer for stock movement prediction［C］// Proceedings of the 29th International Joint Conference on Artificial Intelligence. 2021:4640-4646.
［7］	WANG J Y， ZHANG Y，TANG K， et al. Alphastock: A buying-winners-and-selling-losers investment strategy using interpretable deep reinforcement attention networks［C］// Proceedings of the 25th ACM SIGKDD International Conference on Knowledge Discovery & Data Mining. 2019:1900-1908.
［8］	DING D Z， ZHANG M， PAN X D， et al. Modeling extreme events in time series prediction［C］// Proceedings of the 25th ACM SIGKDD International Conference on Knowledge Discovery & Data Mining. 2019:1114-1122.
［9］	XU Y M， COHEN S B. Stock movement prediction from tweets and historical prices［C］// Proceedings of the 56th Annual Meeting of the Association for Computational Linguistics. 2018:1970-1979.
［10］	FENG F L， HE X N， WANG X， et al. Temporal relational ranking for stock prediction［J］. ACM Transactions on Information Systems (TOIS)， 2019，37(2):1-30.
［11］	KIM R， SO C H， JEONG M， et al. Hats: A hierarchical graph attention network for stock movement prediction［J］. arXiv preprint arXiv:1908.07999， 2019.
［12］	HSU Y L， TSAI Y C， LI C T. FinGAT: Financial graph attention networks for recommending top-k profitable stocks［J］. arXiv preprint arXiv:2106.10159， 2021.
［13］	SAWHNEY R， AGARWAL S， WADHWA A， et al. Spatiotemporal hypergraph convolution network for stock movement forecasting［C］// 2020 IEEE International Conference on Data Mining (ICDM). 2020:482-491.
［14］	CUI C， LI X， DU J， et al. Temporal-relational hypergraph tri-attention networks for stock trend prediction［J］. arXiv preprint arXiv:2107.14033， 2021.
［15］	CHENG R， LI Q. Modeling the momentum spillover effect for stock prediction via attribute-driven graph attention networks［J］. Proceedings of the AAAI Conference on Artificial Intelligence， 2021，35(1):55-62.
［16］	LI C， SONG D J， TAO D C. Multi-task recurrent neural networks and higher-order Markov random fields for stock price movement prediction: Multi-task RNN and higer-order MRFs for stock price classification［C］// Proceedings of the 25th ACM SIGKDD International Conference on Knowledge Discovery & Data Mining. 2019:1141-1151.
［17］	LI W， BAO R H， HARIMOTO K， et al. Modeling the stock relation with graph network for overnight stock movement prediction［C］// Proceedings of the 29th International Conference on International Joint Conferences on Artificial Intelligence. 2021:4541-4547.
［18］	CHEN Y M， WEI Z Y， HUANG X J. Incorporating corporation relationship via graph convolutional neural networks for stock price prediction［C］// Proceedings of the 27th ACM International Conference on Information and Knowledge Management. 2018:1655-1658.
［19］	WANG H Y， LI S， WANG T J， et al. Hierarchical adaptive temporal-relational modeling for stock trend prediction［C］// Proceedings of the 30th International Joint Conference on Artificial Intelligence. 2021:3691-3698.
［20］	YOO J， SOUN Y， PARK Y， et al. Accurate multivariate stock movement prediction via data-axis transformer with multi-level contexts［C］// Proceedings of the 27th ACM SIGKDD Conference on Knowledge Discovery & Data Mining. 2021:2037-2045.
［21］	LECUN Y， BENGIO Y. Convolutional networks for images， speech， and time series［M］// The Handbook of Brain Theory and Neural Networks. 1998:255-258.
［22］	HOCHREITER S， SCHMIDHUBER J. Long short-term memory［J］. Neural Computation， 1997，9(8):1735-1780.
［23］	ASWANI A， SHAZEER N， PARMAR N， et al. Attention is all you need［C］// Proceedings of the 31st International Conference on Neural Information Processing Systems. 2017:6000-6010.
［24］	GOODFELLOW I J， SHLENS J， SZEGEDY C. Explaining and harnessing adversarial examples［J］. arXiv preprint arXiv:1412.6572， 2014.
［25］	ZHANG L H， AGGARWAL C C， QI G J. Stock price prediction via discovering multi-frequency trading patterns［C］// Proceedings of the 23rd ACM SIGKDD International Conference on Knowledge Discovery and Data Mining. 2017:2141-2149.
［26］	ZHANG Z H， ZOHREN S， ROBERTS S. Deeplob: Deep convolutional neural networks for limit order books［J］. IEEE Transactions on Signal Processing， 2019，67(11): 3001-3012.
［27］	LIU G， MAO Y Z， SUN Q， et al. Multi-scale two-way deep neural network for stock trend prediction［C］// Proceedings of the 29th International Joint Conference on Artificial Intelligence. 2020:4555-4561.

[1]	王秋忆, 周浩, 郑婷婷. 改进RetinaNet的电力设备目标检测方法[J]. 计算机与现代化, 2024, 0(01): 47-52.
[2]	林启钊, 彭志平, 郭棉, 崔得龙. 基于双向多步预测的炉管温度场重构方法[J]. 计算机与现代化, 2024, 0(01): 53-58.
[3]	李亚平, 王军防, 余红梅, 窦一民, 肖媛, 田继林. Regformer：基于稀疏注意力的输油管道水力压降预测方法[J]. 计算机与现代化, 2024, 0(01): 59-66.
[4]	宋涛涛, 李艳萍, 李洪港, 韩春雪. 基于改进变结构趋近律的机械臂滑模控制系统[J]. 计算机与现代化, 2023, 0(12): 14-18.
[5]	周成诚, 曾庆军, 杨康, 胡家铭, 韩春伟. 基于高效通道注意力模块的运动想象脑电识别[J]. 计算机与现代化, 2023, 0(12): 19-23.
[6]	邱凯星, 冯广. 基于双重特征注意力的多标签图像分类模型[J]. 计算机与现代化, 2023, 0(12): 41-47.
[7]	张浩洋, 尹梓名, 乐珺怡, 沈达聪, 束翌俊, 杨自逸, 孔祥勇, 龚伟. 3D-SPRNet: 一种基于并行解码器和双注意力机制的胆囊癌分割模型[J]. 计算机与现代化, 2023, 0(12): 59-66.
[8]	张伯泉, 麦海鹏, 陈嘉敏, 逄锦聚. 基于高灰度值注意力机制的脑白质高信号分割[J]. 计算机与现代化, 2023, 0(12): 67-75.
[9]	宁娟, 周庆华, 曾小为. 改进YOLOv7算法在西林瓶轧盖缺陷检测中的应用[J]. 计算机与现代化, 2023, 0(12): 82-86.
[10]	王宇航, 董宝良, 公超, 尚真真, 姚康宁. 基于意图识别的空中群目标动态威胁评估[J]. 计算机与现代化, 2023, 0(12): 100-104.
[11]	罗明杰, 冯开平. 基于沙漏结构与注意力机制的轻量级人脸表情识别方法[J]. 计算机与现代化, 2023, 0(11): 89-94.
[12]	谷明轩, 范冰冰. 基于多模态特征融合的抑郁症识别[J]. 计算机与现代化, 2023, 0(10): 17-22.
[13]	陈俊义. 基于图节点动静态特征的健康事件预测模型[J]. 计算机与现代化, 2023, 0(10): 39-44.
[14]	张嘉琪, 徐啟蕾. 基于NAM-YOLO网络的苹果缺陷检测算法[J]. 计算机与现代化, 2023, 0(10): 53-58.
[15]	叶思佳, 魏延, 杜韩宇, 邓金枝. 结合注意力机制的HRNet图像语义分割算法[J]. 计算机与现代化, 2023, 0(10): 65-69.