A two-stage forecasting model using random forest subset-based feature selection and BiGRU with attention mechanism: Application to stock indices

The heteroscedastic and volatile characteristics of stock price data have attracted the interest of researchers from various disciplines, particularly in the realm of price forecasting. The stock market's non-stationary and volatile nature, driven by complex interrelationships among financial assets, economic developments, and market participants, poses significant challenges for accurate forecasting. This research aims to develop a robust forecasting model to improve the accuracy and reliability of stock price predictions using machine learning. A two-stage forecasting model is introduced. First, a random forest subset-based (RFS) feature selection with repeated [Formula: see text]-fold cross-validation selects the best subset of features from eight predictors: highest price, lowest price, closing price, volume, change, price change ratio, and amplitude. These features are then used as input in a bidirectional gated recurrent unit with an attention mechanism (BiGRU-AM) model to forecast daily opening prices of ten stock indices. The proposed model exhibits superior forecasting performance across ten stock indices when compared to twelve benchmarks, evaluated using root mean squared error (RMSE), mean absolute error (MAE), and the coefficient of determination, [Formula: see text]. The improved prediction accuracy enables financial professionals to make more reliable investment decisions, reducing risks and increasing profits.

Authors

Institutions

Publication Details

Journal
PLoS ONE
Published
2025-05-09
DOI
https://doi.org/10.1371/journal.pone.0323015
Citations
3
Primary Topic
Stock Market Forecasting Methods
Type
article
Field-Weighted Citation Impact
6.32
Controls
|||
ALL TIME
JAN
FEB
MAR
APR
MAY
JUN
JUL
AUG
SEP
OCT
article

A two-stage forecasting model using random forest subset-based feature selection and BiGRU with attention mechanism: Application to stock indices

Dharini Pathmanathan, Vimala Balakrishnan, Shafiqah Azman
3 citations
PLoS ONE
Stock Market Forecasting Methods
6.32
article

A two-stage forecasting model using random forest subset-based feature selection and BiGRU with attention mechanism: Application to stock indices

Dharini Pathmanathan, Vimala Balakrishnan, Shafiqah Azman
article en
3 citations

Abstract

The heteroscedastic and volatile characteristics of stock price data have attracted the interest of researchers from various disciplines, particularly in the realm of price forecasting. The stock market's non-stationary and volatile nature, driven by complex interrelationships among financial assets, economic developments, and market participants, poses significant challenges for accurate forecasting. This research aims to develop a robust forecasting model to improve the accuracy and reliability of stock price predictions using machine learning. A two-stage forecasting model is introduced. First, a random forest subset-based (RFS) feature selection with repeated [Formula: see text]-fold cross-validation selects the best subset of features from eight predictors: highest price, lowest price, closing price, volume, change, price change ratio, and amplitude. These features are then used as input in a bidirectional gated recurrent unit with an attention mechanism (BiGRU-AM) model to forecast daily opening prices of ten stock indices. The proposed model exhibits superior forecasting performance across ten stock indices when compared to twelve benchmarks, evaluated using root mean squared error (RMSE), mean absolute error (MAE), and the coefficient of determination, [Formula: see text]. The improved prediction accuracy enables financial professionals to make more reliable investment decisions, reducing risks and increasing profits.

PLoS ONEVol. 20(5)
Korea University (KR), University of Malaya (MY)
Openalex Percentile: Top 4%
Stock Market Forecasting Methods
6.32
AI Navigator

Ask Laika to Summarize, Analyze, and Connect papers live on the map.

Summarize Papers & Methodologies

Extract key findings, datasets, and comparative methods across publications.

Benchmark Rankings & Visual Analytics

Rank top research institutions, authors, funders, topics, and journals by Field-Weighted Citation Impact (FWCI) and paper volume with instant charts.

Connect Distant Disciplines

Bridge topological clusters on the map to find hidden collaborative intersections.