Transformer model pre-trained on TSX improves stock prediction

Researchers have developed a transformer-based model for stock return prediction, utilizing pre-training on a market index to enhance performance. The model, pre-trained on the Toronto Stock Exchange Index (TSX) and then fine-tuned on individual stocks, showed improved loss metrics compared to baseline models. While achieving lower mean squared error in regression tasks, its ability to generate higher average daily returns was surpassed by ensemble and XGBoost models. A practical application was also created to provide real-time trading support. AI

IMPACT Demonstrates potential for advanced ML techniques to refine quantitative finance strategies and trading tools.

RANK_REASON Academic paper detailing a novel application of transformer models to financial forecasting. [lever_c_demoted from research: ic=1 ai=0.7]

Read on arXiv cs.LG →

• Toronto Stock Exchange Index (TSX)
• Transformer
• XGBoost

• paper
• product

AI-generated summary · Google Gemini · from 1 sources. How we write summaries →