/Deep-Reinforcement-Learning-for-Automated-Stock-Trading-Strategy

Deep Reinforcement Learning for Automated Stock Trading Strategy

Primary LanguageJupyter NotebookGNU General Public License v3.0GPL-3.0

Deep-Reinforcement-Learning-for-Automated-Stock-Trading-Strategy

Deep Reinforcement Learning for Automated Stock Trading Strategy

Stock trading strategies play a critical role in investment. However, it is challenging to design a profitable strategy in a complex and dynamic stock market. In this paper, we propose a deep ensemble reinforcement learning scheme that automatically learns a stock trading strategy by maximizing investment return. We train a deep reinforcement learning agent and obtain an ensemble trading strategy using the three actor-critic based algorithms: Proximal Policy Optimization (PPO), Advantage Actor Critic (A2C), and Deep Deterministic Policy Gradient (DDPG). The ensemble strategy inherits and integrates the best features of the three algorithms, thereby robustly adjusting to different market conditions. In order to avoid the large memory consumption in training networks with continuous action space, we employ a load-on-demand approach for processing very large data. We test our algorithms on the 30 Dow Jones stocks which have adequate liquidity. The performance of the trading agent with different reinforcement learning algorithms is evaluated and compared with both the Dow Jones Industrial Average index and the traditional min-variance portfolio allocation strategy. The proposed deep ensemble scheme is shown to outperform the three individual algorithms and the two baselines in terms of the risk-adjusted return measured by the Sharpe ratio.

Ahoy 👋, I'm Ehsan Paydar

A passionate engineering master's student artificial intelligence originally from Iran

Connect with me:

EPSOFT

Languages and Tools:

aws d3js docker illustrator postgresql python