Selective transfer learning with adversarial training for stock movement prediction

Yang Li*, Hong Ning Dai, Zibin Zheng

*Corresponding author for this work

Research output: Contribution to journalJournal articlepeer-review

16 Citations (Scopus)

Abstract

Stock movement prediction is a critical issue in the field of financial investment. It is very challenging since a stock usually shows highly stochastic property in price and has complex relationships with other stocks. Most existing approaches cannot jointly take the above two issues into account and thus cannot yield satisfactory prediction result. This paper contributes a new stock movement prediction model, Selective Transfer Learning with Adversarial Training (STLAT). Our STLAT method advances existing solutions in two major aspects: (i) tailoring the pre-trained and fine-tuned method for stock movement prediction and (ii) introducing the data selector module to select the more relevant training samples. More specifically, we pre-train the shared base model using three different tasks. The predictor task is constructed to measure the performance of the shared base model with source domain data and target domain data. The adversarial training task is constructed to improve the generalisation of the shared base model. The data selector task is introduced to select the most relevant and high-quality training samples from stocks in source domain. All three tasks are jointly trained with a loss function. As a result, the pre-trained shared base model can be fine-tuned with the stock data in target domain. To validate our method, we perform the back-testing on the historical data of two public datasets and a newly constructed dataset. Extensive experiments demonstrate the superiority of our STLAT method. It outperforms state-of-the-art stock prediction solutions on ACC evaluation of 3.76%, 4.12%, 4.89% on ACL18, KDD17 and CN50, respectively.

Original languageEnglish
Pages (from-to)492-510
Number of pages19
JournalConnection Science
Volume34
Issue number1
Early online date4 Jan 2022
DOIs
Publication statusPublished - 31 Dec 2022

Scopus Subject Areas

  • Software
  • Artificial Intelligence
  • Human-Computer Interaction

User-Defined Keywords

  • adversarial training
  • data selector
  • Stock movement prediction
  • transfer learning

Fingerprint

Dive into the research topics of 'Selective transfer learning with adversarial training for stock movement prediction'. Together they form a unique fingerprint.

Cite this