DeepClue: Visual Interpretation of Text-Based Deep Stock Prediction

被引:57
作者
Shi, Lei [1 ,2 ]
Teng, Zhiyang [3 ]
Wang, Le [1 ,2 ]
Zhang, Yue [3 ]
Binder, Alexander [3 ]
机构
[1] Chinese Acad Sci, Inst Software, SKLCS, Beijing 100864, Peoples R China
[2] UCAS, Beijing 100049, Peoples R China
[3] Singapore Univ Technol & Design, Singapore 487372, Singapore
关键词
Deep learning; visualization; model interpretation; stock prediction;
D O I
10.1109/TKDE.2018.2854193
中图分类号
TP18 [人工智能理论];
学科分类号
081104 ; 0812 ; 0835 ; 1405 ;
摘要
The recent advance of deep learning has enabled trading algorithms to predict stock price movements more accurately. Unfortunately, there is a significant gap in the real-world deployment of this breakthrough. For example, professional traders in their long-term careers have accumulated numerous trading rules, the myth of which they can understand quite well. On the other hand, deep learning models have been hardly interpretable. This paper presents DeepClue, a system built to bridge text-based deep learning models and end users through visually interpreting the key factors learned in the stock price prediction model. We make three contributions in DeepClue. First, by designing the deep neural network architecture for interpretation and applying an algorithm to extract relevant predictive factors, we provide a useful case on what can be interpreted out of the prediction model for end users. Second, by exploring hierarchies over the extracted factors and displaying these factors in an interactive, hierarchical visualization interface, we shed light on how to effectively communicate the interpreted model to end users. Specially, the interpretation separates the predictables from the unpredictables for stock prediction through the use of intercept model parameters and a risk visualization design. Third, we evaluate the integrated visualization system through two case studies in predicting the stock price with financial news and company-related tweets from social media. Quantitative experiments comparing the proposed neural network architecture with state-of-the-art models and the human baseline are conducted and reported. Feedbacks from an informal user study with domain experts are summarized and discussed in details. The study results demonstrate the effectiveness of DeepClue in helping to complete stock market investment and analysis tasks.
引用
收藏
页码:1094 / 1108
页数:15
相关论文
共 48 条
  • [1] Ammar W., 2016, TACL, V4, P431
  • [2] [Anonymous], 2017, GOOGLE DEEP DREAM
  • [3] [Anonymous], 1341 U MONTR
  • [4] [Anonymous], 2017, CROSS CORR
  • [5] Arras Leila, 2016, Rep4NLP @ACL, DOI 10.18653/v1/W16-1601
  • [6] On Pixel-Wise Explanations for Non-Linear Classifier Decisions by Layer-Wise Relevance Propagation
    Bach, Sebastian
    Binder, Alexander
    Montavon, Gregoire
    Klauschen, Frederick
    Mueller, Klaus-Robert
    Samek, Wojciech
    [J]. PLOS ONE, 2015, 10 (07):
  • [7] Bahdanau D, 2016, Arxiv, DOI arXiv:1409.0473
  • [8] Binder A., 2016, P INT C ART NEUR NET, V9887, P1
  • [9] Bordes Antoine, 2014, Question answering with subgraph embeddings
  • [10] Chen HC, 2013, INTELL SYST SER, P1, DOI [10.1155/2013/213234, 10.1007/978-3-642-38868-2_1, 10.1016/B978-0-12-404702-0.00001-X]