FinGAT: Financial Graph Attention Networks for Recommending Top-$K$K Profitable Stocks

被引:49
作者
Hsu, Yi-Ling [1 ]
Tsai, Yu-Che [2 ]
Li, Cheng-Te [3 ]
机构
[1] Natl Taiwan Univ, Master Program Stat, Taipei 10617, Taiwan
[2] Natl Taiwan Univ, Dept Comp Sci & Informat Engn, Taipei 10617, Taiwan
[3] Natl Cheng Kung Univ, Inst Data Sci, Tainan 701, Taiwan
关键词
Profitable stock recommendation; graph attention networks; stock movement prediction; sector information;
D O I
10.1109/TKDE.2021.3079496
中图分类号
TP18 [人工智能理论];
学科分类号
081104 ; 0812 ; 0835 ; 1405 ;
摘要
Financial technology (FinTech) has drawn much attention among investors and companies. While conventional stock analysis in FinTech targets at predicting stock prices, less effort is made for profitable stock recommendation. Besides, in existing approaches on modeling time series of stock prices, the relationships among stocks and sectors (i.e., categories of stocks) are either neglected or pre-defined. Ignoring stock relationships will miss the information shared between stocks while using pre-defined relationships cannot depict the latent interactions or influence of stock prices between stocks. In this work, we aim at recommending the top-K profitable stocks in terms of return ratio using time series of stock prices and sector information. We propose a novel deep learning-based model, Financial Graph Attention Networks (FinGAT), to tackle the task under the setting that no pre-defined relationships between stocks are given. The idea of FinGAT is three-fold. First, we devise a hierarchical learning component to learn short-term and long-term sequential patterns from stock time series. Second, a fully-connected graph between stocks and a fully-connected graph between sectors are constructed, along with graph attention networks, to learn the latent interactions among stocks and sectors. Third, a multi-task objective is devised to jointly recommend the profitable stocks and predict the stock movement. Experiments conducted on Taiwan Stock, S & P 500, and NASDAQ datasets exhibit remarkable recommendation performance of our FinGAT, comparing to state-of-the-art methods.
引用
收藏
页码:469 / 481
页数:13
相关论文
共 31 条
  • [1] [Anonymous], 2015, PROC C EMPIRICAL MET, DOI DOI 10.18653/V1/D15-1166
  • [2] ModAugNet: A new forecasting framework for stock market index value with an overfitting prevention LSTM module and a prediction LSTM module
    Baek, Yujin
    Kim, Ha Young
    [J]. EXPERT SYSTEMS WITH APPLICATIONS, 2018, 113 : 457 - 480
  • [3] Box G E., 2015, TIME SERIES ANAL FOR
  • [4] Shocks and Stocks: A Bottom-up Assessment of the Relationship Between Oil Prices, Gasoline Prices and the Returns of Chinese Firms
    Broadstock, David C.
    Fan, Ying
    Ji, Qiang
    Zhang, Dayong
    [J]. ENERGY JOURNAL, 2016, 37 : 55 - 86
  • [5] Support vector machine with adaptive parameters in financial time series forecasting
    Cao, LJ
    Tay, FEH
    [J]. IEEE TRANSACTIONS ON NEURAL NETWORKS, 2003, 14 (06): : 1506 - 1518
  • [6] Chen K, 2015, PROCEEDINGS 2015 IEEE INTERNATIONAL CONFERENCE ON BIG DATA, P2823, DOI 10.1109/BigData.2015.7364089
  • [7] Incorporating Corporation Relationship via Graph Convolutional Neural Networks for Stock Price Prediction
    Chen, Yingmei
    Wei, Zhongyu
    [J]. CIKM'18: PROCEEDINGS OF THE 27TH ACM INTERNATIONAL CONFERENCE ON INFORMATION AND KNOWLEDGE MANAGEMENT, 2018, : 1655 - 1658
  • [8] Cho K., 2014, P 2014 C EMP METH NA, DOI 10.3115/v1/d14-1179
  • [9] Finding Matches in a Haystack: A Max-Pooling Strategy for Graph Matching in the Presence of Outliers
    Cho, Minsu
    Sun, Jian
    Duchenne, Olivier
    Ponce, Jean
    [J]. 2014 IEEE CONFERENCE ON COMPUTER VISION AND PATTERN RECOGNITION (CVPR), 2014, : 2091 - 2098
  • [10] Gated-Attention Readers for Text Comprehension
    Dhingra, Bhuwan
    Liu, Hanxiao
    Yang, Zhilin
    Cohen, William W.
    Salakhutdinov, Ruslan
    [J]. PROCEEDINGS OF THE 55TH ANNUAL MEETING OF THE ASSOCIATION FOR COMPUTATIONAL LINGUISTICS (ACL 2017), VOL 1, 2017, : 1832 - 1846