Status | 已发表Published |
Title | Improving stock trend prediction with pretrain multi-granularity denoising contrastive learning |
Creator | |
Date Issued | 2024-04-01 |
Source Publication | Knowledge and Information Systems
![]() |
ISSN | 0219-1377 |
Volume | 66Issue:4Pages:2439-2466 |
Abstract | Stock trend prediction (STP) aims to predict price fluctuation, which is critical in financial trading. The existing STP approaches only use market data with the same granularity (e.g., as daily market data). However, in the actual financial investment, there are a large number of more detailed investment signals contained in finer-grained data (e.g., high-frequency data). This motivates us to research how to leverage multi-granularity market data to capture more useful information and improve the accuracy in the task of STP. However, the effective utilization of multi-granularity data presents a major challenge. Firstly, the iteration of multi-granularity data with time will lead to more complex noise, which is difficult to extract signals. Secondly, the difference in granularity may lead to opposite target trends in the same time interval. Thirdly, the target trends of stocks with similar features can be quite different, and different sizes of granularity will aggravate this gap. In order to address these challenges, we present a self-supervised framework of multi-granularity denoising contrastive learning (MDC). Specifically, we construct a dynamic dictionary of memory, which can obtain clear and unified representations by filtering noise and aligning multi-granularity data. Moreover, we design two contrast learning modules during the fine-tuning stage to solve the differences in trends by constructing additional self-supervised signals. Besides, in the pre-training stage, we design the granularity domain adaptation module (GDA) to address the issues of temporal inconsistency and data imbalance associated with different granularity in financial data, alongside the memory self-distillation module (MSD) to tackle the challenge posed by a low signal-to-noise ratio. The GDA alleviates these complications by replacing a portion of the coarse-grained data with the preceding time step’s fine-grained data, while the MSD seeks to filter out intrinsic noise by aligning the fine-grained representations with the coarse-grained representations’ distribution using a self-distillation mechanism. Extensive experiments on the CSI 300 and CSI 100 datasets show that our framework stands out from the existing top-level systems and has excellent profitability in real investing scenarios. |
Keyword | Contrastive learning Denoising Memory Multi-granularity data Pre-training Stock trend prediction |
DOI | 10.1007/s10115-023-02006-1 |
URL | View source |
Indexed By | SCIE |
Language | 英语English |
WOS Research Area | Computer Science |
WOS Subject | Computer Science, Artificial Intelligence ; Computer Science, Information Systems |
WOS ID | WOS:001131769900001 |
Scopus ID | 2-s2.0-85180656261 |
Citation statistics | |
Document Type | Journal article |
Identifier | http://repository.uic.edu.cn/handle/39GCC9TT/11471 |
Collection | Faculty of Science and Technology |
Corresponding Author | Guo, Jianxiong |
Affiliation | 1.Advanced Institute of Natural Sciences, Beijing Normal University, Zhuhai, 519087, China 2.Guangdong Key Lab of AI and Multi-Modal Data Processing, Department of Computer Science, BNU-HKBU United International College, Zhuhai, 519087, China 3.Department of Mathematical Science, BNU-HKBU United International College, Zhuhai, 519087, China |
First Author Affilication | Beijing Normal-Hong Kong Baptist University |
Corresponding Author Affilication | Beijing Normal-Hong Kong Baptist University |
Recommended Citation GB/T 7714 | Wang, Mingjie,Wang, Siyuan,Guo, Jianxionget al. Improving stock trend prediction with pretrain multi-granularity denoising contrastive learning[J]. Knowledge and Information Systems, 2024, 66(4): 2439-2466. |
APA | Wang, Mingjie, Wang, Siyuan, Guo, Jianxiong, & Jia, Weijia. (2024). Improving stock trend prediction with pretrain multi-granularity denoising contrastive learning. Knowledge and Information Systems, 66(4), 2439-2466. |
MLA | Wang, Mingjie,et al."Improving stock trend prediction with pretrain multi-granularity denoising contrastive learning". Knowledge and Information Systems 66.4(2024): 2439-2466. |
Files in This Item: | There are no files associated with this item. |
Items in the repository are protected by copyright, with all rights reserved, unless otherwise indicated.
Edit Comment