From Index to Equity: Pre-Training Transformers for Stock Return Prediction
Dit artikel toont aan dat het vooraf trainen van een transformer-model op de Toronto Stock Exchange-index de prestaties van het model bij het voorspellen van de richting en de waarde van individuele aandelenrendementen aanzienlijk verbetert ten opzichte van LSTM- en XGBoost-benchmarks, ondanks dat laatstgenoemde benchmarks in de praktijk een hoger gemiddeld dagelijks rendement behalen.
Oorspronkelijk artikel gelicentieerd onder CC BY 4.0 (http://creativecommons.org/licenses/by/4.0/). Dit is een AI-gegenereerde uitleg van het onderstaande artikel. Het is niet geschreven of goedgekeurd door de auteurs. Raadpleeg het oorspronkelijke artikel voor technische nauwkeurigheid. Lees de volledige disclaimer
Stel je voor dat je probeert het weer te voorspellen voor een specifieke, kleine wijk. Het is ongelooflijk chaotisch; één huis kan zonnig zijn terwijl de buren hagel krijgen. Stel je nu voor dat je eerst probeert het weer te voorspellen voor een heel land. Het grote plaatje is veel rustiger en makkelijker te begrijpen, omdat de wilde schommelingen van individuele wijken elkaar gemiddeld opheffen.
Dit artikel gaat over het gebruik van een superslim computerbrein (een Transformer) om aandelenkoersen te voorspellen, en de auteurs ontdekten dat het het beste is om dit brein eerst het "landelijke" weer te leren voordat je vraagt het "wijkniveau-weer" te voorspellen.
Hier is de uiteenzetting van hun reis:
1. Het Probleem: De Beurs is Chaotisch
Voorspellen of een enkel aandeel morgen stijgt of daalt, is als proberen het exacte pad te raden van een enkel blad dat in een orkaan waait. De markt is rommelig, verandert voortdurend en zit vol ruis. Traditionele computermodellen (zoals LSTM's en XGBoost) hebben geprobeerd dit op te lossen, maar ze hebben vaak moeite omdat ze elke keer opnieuw vanaf nul moeten leren wanneer ze naar een nieuw aandeel kijken.
2. De Oplossing: De "Leerling"-Strategie (Pre-training)
De auteurs besloten een nieuwe leermethode te proberen die Transfer Learning heet. Denk aan het trainen van een meesterkok:
- Stap 1 (Pre-training): In plaats van de kok meteen te leren hoe hij een specifiek, raar gerecht moet bereiden, laat je hem eerst oefenen op een grote, rustige pan soep (de TSX-marktindex). De soep vertegenwoordigt de hele aandelenmarkt. Het is minder chaotisch dan een enkel aandeel. De kok leert de algemene kookregels (hoe smaken interageren, hoe hitte werkt) zonder afgeleid te worden door één raar ingrediënt.
- Stap 2 (Fine-tuning): Zodra de kok de basis van de grote soep begrijpt, geef je hem een specifiek, lastig recept (een enkel aandeel). Omdat hij al de fundamenten kent, leert hij het specifieke recept veel sneller en beter dan als hij vanaf nul was begonnen.
3. De Hulpmiddelen: Transformers versus de Oude Garde
Het team gebruikte een moderne AI-architectuur genaamd een Transformer (dezelfde technologie die tools zoals ChatGPT aandrijft). Ze vergeleken deze met oudere, betrouwbare tools:
- LSTM: Een oudere AI-soort die goed is in het onthouden van sequenties.
- XGBoost: Een krachtig statistisch hulpmiddel dat veel beslissingsbomen bouwt om patronen te vinden.
- Ensemble: Een "comité" dat de voorspellingen van al bovenstaande modellen middelt.
4. Wat Ze Vonden (De Resultaten)
Het experiment had twee hoofddoelen: de richting voorspellen (Omhoog of Omlaag) en de exacte waarde voorspellen (Hoeveel het stijgt).
Het "Richting"-Spel (Classificatie):
- Toen de Transformer probeerde een enkel aandeel vanaf nul te leren, was het verward en presteerde het slecht.
- Toen ze de "Leerling-Strategie" gebruikten (eerst pre-trainen op de grote marktindex), werd de Transformer veel beter in het raden of een aandeel zou stijgen of dalen.
- Echter, het ouderwetse XGBoost-model was nog steeds het beste in het simpelweg goed raden van de "Omhoog of Omlaag"-voorspelling.
Het "Exacte Waarde"-Spel (Regressie):
- Bij het proberen de exacte percentage-opbrengst te voorspellen, wonnen de Fine-tuned Transformer en de Fine-tuned LSTM van het XGBoost-model. Ze waren nauwkeuriger in het berekenen van het specifieke getal.
- Maar hier is de draai: Hoewel de Transformer nauwkeuriger was in het berekenen van het getal, maakte het Ensemble-model (het comité) in een gesimuleerde handelstest daadwerkelijk het meeste geld.
- Waarom? De Transformer was zeer voorzichtig en voorspelde getallen die zeer dicht bij nul lagen (veilige weddenschappen), terwijl de XGBoost/Ensemble-modellen bereid waren te wedden op aandelen met grotere schommelingen. Op de aandelenmarkt betekent "veilig" zijn soms dat je de grote winsten mist.
5. De Werkelijke Tool
De onderzoekers stopten niet alleen bij theorie. Ze bouwden een live trading-app (met behulp van een tool genaamd Streamlit).
- Je kunt een aandelen ticker invoeren (zoals "TD" of "RY").
- De app pakt de nieuwste data, voert deze door hun modellen en vertelt je welke 5 aandelen het meest lijken te stijgen en welke 5 lijken te crashen.
- Het is als een weer-app die je een dagelijkse voorspelling geeft voor je persoonlijke portefeuille.
De Conclusie
Het artikel concludeert dat hoewel geen enkel model perfect is (de aandelenmarkt is nog steeds zeer moeilijk te voorspellen), het eerst leren van het AI over het grote marktbeeld helpt het om de kleine plaatjes beter te begrijpen.
Het Transformer-model is een sterke kandidaat voor het begrijpen van de nuances van aandelenbewegingen, maar de "ouderwetse" methoden (zoals XGBoost) en "comité"-benaderingen houden nog steeds de kroon voor het maken van de meeste winst in deze specifieke test. De auteurs plannen om de Transformer in de toekomst nog groter en slimmer te maken, misschien door hem te leren wereldnieuws en economische rapporten te lezen, niet alleen historische koersgrafieken.
Verdrinkt u in papers in uw vakgebied?
Ontvang dagelijkse digests van de nieuwste papers die bij uw onderzoekswoorden passen — met technische samenvattingen, in uw taal.