SYNTOLOGY HomeExplorerAtlasCodeMethodologyAboutDevelopersFeedPricing
Paper · 2312.15235 · 2023

MASTER: Market-Guided Stock Transformer for Stock Price Forecasting

arXiv · PDF · Open in the Atlas

Code that ran

We lifted 4 functions out of this paper's own repositories and ran 4 of them in a sandbox. "Ran" means the function executed on a synthesized input and returned a value. It is not a reproduction of the paper's results.

RepositoryRoleRan
SJTU-Quant/MASTER canonical 4 of 4
FunctionStatusWhere it lives
calc_ic Ran SJTU-Quant/MASTER/base_model.py
code served (permissive licence) · get_code("56769462a41ddeaa")
drop_extreme Ran SJTU-Quant/MASTER/qlib-update/pytorch_master_ts.py
code served (permissive licence) · get_code("c4b289d81ecb40ac")
drop_na Ran SJTU-Quant/MASTER/qlib-update/pytorch_master_ts.py
code served (permissive licence) · get_code("8560128f6975b469")
zscore Ran SJTU-Quant/MASTER/qlib-update/pytorch_master_ts.py
code served (permissive licence) · get_code("a10dc2b3fb7f59ea")

Repositories linked to this paper

Some links come from the archived Papers with Code dataset (CC BY-SA 4.0): attribution and licence.

Abstract

Stock price forecasting has remained an extremely challenging problem for many decades due to the high volatility of the stock market. Recent efforts have been devoted to modeling complex stock correlations toward joint stock price forecasting. Existing works share a common neural architecture that learns temporal patterns from individual stock series and then mixes up temporal representations to establish stock correlations. However, they only consider time-aligned stock correlations stemming from all the input stock features, which suffer from two limitations. First, stock correlations often occur momentarily and in a cross-time manner. Second, the feature effectiveness is dynamic with market variation, which affects both the stock sequential patterns and their correlations. To address the limitations, this paper introduces MASTER, a MArkert-Guided Stock TransformER, which models the momentary and cross-time stock correlation and leverages market information for automatic feature selection. MASTER elegantly tackles the complex stock correlation by alternatively engaging in intra-stock and inter-stock information aggregation. Experiments show the superiority of MASTER compared with previous works and visualize the captured realistic stock correlation to provide valuable insights.

For agents

The same record, over MCP at https://syntology.ai/mcp:

get_harvested_code_for_paper("2312.15235")
get_code_for_paper("2312.15235")
have("2312.15235")

Connect an agent — have() is free.