arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2407.18519cs.LGcs.AIq-fin.STstat.ML
收藏

TCGPN:用于股票预测的时序相关性图预训练网络

TCGPN: Temporal-Correlation Graph Pre-trained Network for Stock Forecasting

School of Intelligence Science and Technology, Peking University(北京大学智能科学与技术学院);National Key Laboratory of General Artificial Intelligence(通用人工智能国家重点实验室)

Wenbo Yan, Ying Tan

AI总结:

针对STGNNs在缺乏周期性的股票预测中表现不佳且难以处理大量节点的问题,提出TCGPN,利用时序相关性融合编码器与预训练任务,结构独立于节点数量,通过微调MLP取得state-of-the-art结果。

摘要:

近期,结合时间特征和时间序列间的相关性已成为时间序列预测中的有效方法。时空图神经网络(STGNNs)在许多时序相关性预测问题上表现出良好性能。然而,当应用于缺乏周期性的任务(如股票数据预测)时,STGNNs的有效性和鲁棒性并不理想。且STGNNs受限于内存节省,无法处理具有大量节点的问题。本文提出一种名为时序相关性图预训练网络(TCGPN)的新方法来解决这些局限。TCGPN利用时序相关性融合编码器获取混合表示,并采用精心设计的时序与相关性预训练任务的预训练方法。整体结构独立于节点的数量和顺序,因此可通过各种数据增强获得更好结果。且训练时的内存消耗可通过多次采样显著降低。实验在表现出极小周期性的真实股票市场数据集CSI300和CSI500上进行。我们在下游任务中微调一个简单的MLP并取得state-of-the-art结果,验证了其捕获更鲁棒时序相关性模式的能力。

英文摘要:

Recently, the incorporation of both temporal features and the correlation across time series has become an effective approach in time series prediction. Spatio-Temporal Graph Neural Networks (STGNNs) demonstrate good performance on many Temporal-correlation Forecasting Problem. However, when applied to tasks lacking periodicity, such as stock data prediction, the effectiveness and robustness of STGNNs are found to be unsatisfactory. And STGNNs are limited by memory savings so that cannot handle problems with a large number of nodes. In this paper, we propose a novel approach called the Temporal-Correlation Graph Pre-trained Network (TCGPN) to address these limitations. TCGPN utilize Temporal-correlation fusion encoder to get a mixed representation and pre-training method with carefully designed temporal and correlation pre-training tasks. Entire structure is independent of the number and order of nodes, so better results can be obtained through various data enhancements. And memory consumption during training can be significantly reduced through multiple sampling. Experiments are conducted on real stock market data sets CSI300 and CSI500 that exhibit minimal periodicity. We fine-tune a simple MLP in downstream tasks and achieve state-of-the-art results, validating the capability to capture more robust temporal correlation patterns.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑