A federated learning-enabled predictive analysis to forecast stock market trends

Pourroostaei Ardakani, Saeid (2023) A federated learning-enabled predictive analysis to forecast stock market trends. Journal of Ambient Intelligence and Humanized Computing . ISSN 1868-5137

Full content URL: https://doi.org/10.1007/s12652-023-04570-4

WarningThere is a more recent version of this item available.

Documents
A federated learning-enabled predictive analysis to forecast stock market trends
Published open access manuscript
[img]
[Download]
[img]
Preview
PDF
s12652-023-04570-4.pdf - Whole Document

759kB
Item Type:Article
Item Status:Live Archive

Abstract

This article proposes a federated learning framework to build Random Forest, Support Vector Machine, and Linear Regression models for stock market prediction. The performance of the federated learning is compared against centralised and decentralised learning frameworks to figure out the best fitting approach for stock market prediction. According to the results, federated learning outperforms both centralised and decentralised frameworks in terms of Mean Square Error if Random Forest (MSE = 0.021) and Support Vector Machine techniques (MSE = 37.596) are used, while centralised learning (MSE = 0.011) outperforms federated and decentralised frameworks if a linear regression model is used. Moreover, federated learning gives a better model training delay as compared to the benchmarks if Linear Regression (time = 9.7 s) and Random Forest models (time = 515 s) are used, whereas decentralised learning gives a minimised model training delay (time = 3847 s) for Support Vector Machine.

Keywords:Federated learning, Decentralised learning, centralised learning, stock market trend prediction
Subjects:G Mathematical and Computer Sciences > G760 Machine Learning
Divisions:College of Science > School of Computer Science
Related URLs:
ID Code:53502
Deposited On:23 Feb 2023 15:02

Available Versions of this Item

Repository Staff Only: item control page