Predicting Stock Prices: Strategies of Ensemble Learning with Transformer, ARIMA, and Linear Regression Models

Leila Mozaffari,Jianhua Zhang
DOI: https://doi.org/10.1145/3696271.3696293
2024-01-01
Abstract:Stock market predictions play an important role in determining the future value of company stocks and other financial instruments traded on stock exchanges. Market prediction accuracy has been enhanced using machine learning techniques in recent years. To further improve stock market prediction accuracy, this paper proposes an ensemble learning approach. Based on data from Yahoo Finance, we compare the prediction performance of three different machine learning models: Transformer, Linear Regression, and ARIMA. From experimental results, we may conclude that the combination of Transformer and Linear Regression model using a stacking ensemble approach yields remarkable results for predicting the stock index of AAME. This hybrid model achieves MAPE (mean absolute percentage error) of 2.2437%, demonstrating its ability to account for stock market dynamics and to predict stock closing prices with a high degree of accuracy.
What problem does this paper attempt to address?