Deep reinforcement learning stock market trading, utilizing a CNN with candlestick images

被引:6
|
作者
Brim, Andrew [1 ]
Flann, Nicholas S. [1 ]
机构
[1] Utah State Univ, Dept Comp Sci, Logan, UT 84322 USA
来源
PLOS ONE | 2022年 / 17卷 / 02期
关键词
COEFFICIENTS;
D O I
10.1371/journal.pone.0263181
中图分类号
O [数理科学和化学]; P [天文学、地球科学]; Q [生物科学]; N [自然科学总论];
学科分类号
07 ; 0710 ; 09 ;
摘要
Billions of dollars are traded automatically in the stock market every day, including algorithms that use neural networks, but there are still questions regarding how neural networks trade. The black box nature of a neural network gives pause to entrusting it with valuable trading funds. A more recent technique for the study of neural networks, feature map visualizations, yields insight into how a neural network generates an output. Utilizing a Convolutional Neural Network (CNN) with candlestick images as input and feature map visualizations gives a unique opportunity to determine what in the input images is causing the neural network to output a certain action. In this study, a CNN is utilized within a Double Deep Q-Network (DDQN) to outperform the S&P 500 Index returns, and also analyze how the system trades. The DDQN is trained and tested on the 30 largest stocks in the S&P 500. Following training the CNN is used to generate feature map visualizations to determine where the neural network is placing its attention on the candlestick images. Results show that the DDQN is able to yield higher returns than the S&P 500 Index between January 2, 2020 and June 30, 2020. Results also show that the CNN is able to switch its attention from all the candles in a candlestick image to the more recent candles in the image, based on an event such as the coronavirus stock market crash of 2020.
引用
收藏
页数:25
相关论文
共 50 条
  • [1] Quantitative Trading on Stock Market Based on Deep Reinforcement Learning
    Wu, Jia
    Wang, Chen
    Xiong, Lidong
    Sun, Hongyong
    2019 INTERNATIONAL JOINT CONFERENCE ON NEURAL NETWORKS (IJCNN), 2019,
  • [2] Deep Reinforcement Learning Approach for Trading Automation in the Stock Market
    Kabbani, Taylan
    Duman, Ekrem
    IEEE ACCESS, 2022, 10 : 93564 - 93574
  • [3] A Deep Multimodal Reinforcement Learning System Combined with CNN and LSTM for Stock Trading
    Shin, Hong-Gi
    Ra, Ilkyeun
    Choi, Yong-Hoon
    2019 10TH INTERNATIONAL CONFERENCE ON INFORMATION AND COMMUNICATION TECHNOLOGY CONVERGENCE (ICTC): ICT CONVERGENCE LEADING THE AUTONOMOUS FUTURE, 2019, : 7 - 11
  • [4] Beating the Stock Market with a Deep Reinforcement Learning Day Trading System
    Conegundes, Leonardo
    Machado Pereira, Adriano C.
    2020 INTERNATIONAL JOINT CONFERENCE ON NEURAL NETWORKS (IJCNN), 2020,
  • [5] Stock Market Trading Based on Market Sentiments and Reinforcement Learning
    Suhail, K. M. Ameen
    Sankar, Syam
    Kumar, Ashok S.
    Nestor, Tsafack
    Soliman, Naglaa F.
    Algarni, Abeer D.
    El-Shafai, Walid
    Abd El-Samie, Fathi E.
    CMC-COMPUTERS MATERIALS & CONTINUA, 2022, 70 (01): : 935 - 950
  • [6] DEEP LEARNING FOR STOCK MARKET TRADING: A SUPERIOR TRADING STRATEGY?
    Fister, D.
    Mun, J. C.
    Jagric, V
    Jagric, T.
    NEURAL NETWORK WORLD, 2019, 29 (03) : 151 - 171
  • [7] A Stock Trading Strategy Based on Deep Reinforcement Learning
    Khemlichi, Firdaous
    Chougrad, Hiba
    Khamlichi, Youness Idrissi
    El Boushaki, Abdessamad
    Ben Ali, Safae El Haj
    ADVANCED INTELLIGENT SYSTEMS FOR SUSTAINABLE DEVELOPMENT (AI2SD'2020), VOL 2, 2022, 1418 : 920 - 928
  • [8] Stock Trading Strategies Based on Deep Reinforcement Learning
    Li, Yawei
    Liu, Peipei
    Wang, Ze
    SCIENTIFIC PROGRAMMING, 2022, 2022
  • [9] Application of Deep Reinforcement Learning on Automated Stock Trading
    Chen, Lin
    Gao, Qiang
    PROCEEDINGS OF 2019 IEEE 10TH INTERNATIONAL CONFERENCE ON SOFTWARE ENGINEERING AND SERVICE SCIENCE (ICSESS 2019), 2019, : 29 - 33
  • [10] Improving exploration in deep reinforcement learning for stock trading
    Zemzem, Wiem
    Tagina, Moncef
    INTERNATIONAL JOURNAL OF COMPUTER APPLICATIONS IN TECHNOLOGY, 2023, 72 (04) : 288 - 295