Pourroostaei Ardakani, Saeid, Du, Nanjiang, Lin, Chenhong , Yang, Jiun‑Chi, Bi, Zhuoran and Chen, Lejun (2023) A federated learning-enabled predictive analysis to forecast stock market trends. Journal of Ambient Intelligence and Humanized Computing, 14 . pp. 4529-4535. ISSN 1868-5137
Full content URL: https://doi.org/10.1007/s12652-023-04570-4
This is the latest version of this item.
Documents |
|
|
PDF
s12652-023-04570-4.pdf - Whole Document Available under License Creative Commons Attribution 4.0 International. 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: | 53623 |
Deposited On: | 09 Mar 2023 09:51 |
Available Versions of this Item
-
A federated learning-enabled predictive analysis to forecast stock market trends. (deposited 23 Feb 2023 15:02)
- A federated learning-enabled predictive analysis to forecast stock market trends. (deposited 09 Mar 2023 09:51) [Currently Displayed]
Repository Staff Only: item control page