Sector influence aware stock trend prediction using 3D convolutional neural network

被引:8
作者
Sinha, Siddhant [1 ]
Mishra, Shambhavi [2 ]
Mishra, Vipul [2 ]
Ahmed, Tanveer [2 ]
机构
[1] Manipal Inst Technol, Manipal, Karnataka, India
[2] Bennett Univ, Greater Noida 201310, Uttar Pradesh, India
关键词
Stock trend classification; Deep learning; Trading; Convolutional neural network; Technical indicators; Ensemble learning;
D O I
10.1016/j.jksuci.2022.02.008
中图分类号
TP [自动化技术、计算机技术];
学科分类号
0812 ;
摘要
Stock price prediction is a difficult task. This article takes on this challenge and proposes a 3D Convolutional Neural Network based approach to classify the directional trends in a stock's price. To do that, five companies from a sector are grouped together, and the overall trend in each is predicted simultaneously. This is done to analyze the influence of one company on another. For each company, multiple technical indicators are chosen, and the stock prices are converted into a 3D image of size 15 x 15 x 5. To find the best features, we experiment with hierarchical clustering. To complement the 3D Convolutional Neural Network, we also examine the idea of ensemble learning. The proposed method and several existing models are combined to improve the performance of the system. Experimentation is performed on forty-five different companies of the National Stock Exchange. Compared to other similar techniques in literature, our work has achieved up to 35% annual returns for some stocks, with the average being 9.19%. Lastly, we also try to show that grouping companies together and making the prediction on a sector could serve as a new benchmark for stock trend classification. (C) 2022 The Author(s). Published by Elsevier B.V. on behalf of King Saud University.
引用
收藏
页码:1511 / 1522
页数:12
相关论文
共 31 条
[1]   Uncovering nonlinear structure in real-time stock-market indexes: The S&P 500, the DAX, the Nikkei 225, and the FTSE-100 [J].
Abhyankar, A ;
Copeland, LS ;
Wong, W .
JOURNAL OF BUSINESS & ECONOMIC STATISTICS, 1997, 15 (01) :1-14
[2]  
Abraham Ajith, 2004, ARXIV PREPRINT CS 04
[3]   Evaluation of Tree-Based Ensemble Machine Learning Models in Predicting Stock Price Direction of Movement [J].
Ampomah, Ernest Kwame ;
Qin, Zhiguang ;
Nyame, Gabriel .
INFORMATION, 2020, 11 (06)
[4]  
Asiful Mohammed, 2018, IEEE S SERIES COMPUT
[5]   International diversification with frontier markets [J].
Berger, Dave ;
Pukthuanthong, Kuntara ;
Yang, J. Jimmy .
JOURNAL OF FINANCIAL ECONOMICS, 2011, 101 (01) :227-242
[6]  
Canziani A., 2016, arXiv preprint arXiv:1605.07678, DOI DOI 10.48550/ARXIV.1605.07678
[7]   Industry classifications and return comovement [J].
Chan, Louis K. C. ;
Lakonishok, Josef ;
Swaminathan, Bhaskaran .
FINANCIAL ANALYSTS JOURNAL, 2007, 63 (06) :56-70
[8]   A novel graph convolutional feature based convolutional neural network for stock trend prediction [J].
Chen, Wei ;
Jiang, Manrui ;
Zhang, Wei-Guo ;
Chen, Zhensong .
INFORMATION SCIENCES, 2021, 556 :67-94
[9]  
Ciucu StefanCristian., 2014, ARXIV PREPRINT ARXIV
[10]   Deep learning with long short-term memory networks for financial market predictions [J].
Fischer, Thomas ;
Krauss, Christopher .
EUROPEAN JOURNAL OF OPERATIONAL RESEARCH, 2018, 270 (02) :654-669