AI Chat Paper
Note: Please note that the following content is generated by AMiner AI. SciOpen does not take any responsibility related to this content.
{{lang === 'zh_CN' ? '文章概述' : 'Summary'}}
{{lang === 'en_US' ? '中' : 'Eng'}}
Chat more with AI
PDF (480.6 KB)
Collect
Submit Manuscript AI Chat Paper
Show Outline
Outline
Show full outline
Hide outline
Outline
Show full outline
Hide outline
Research Article | Open Access

Federated and ensemble learning framework with optimized feature selection for heart disease detection

Olfa Hrizi1Karim Gasmi1Abdulrahman Alyami2( )Adel Alkhalil3Ibrahim Alrashdi1Ali Alqazzaz4Lassaad Ben Ammar5Manel Mrabet5Alameen E.M. Abdalrahman1Samia Yahyaoui6
Department of Computer Science, College of Computer and Information Sciences, Jouf University, Sakaka 72388, Saudi Arabia
Department of Information Systems, College of Computer and Information Sciences, Jouf University, Sakaka, Saudi Arabia
Department of Software Engineering, College of Computer Science and Engineering, University of Hail, Hail, 81481 Saudi Arabia
College of Computing and Information Technology, University of Bisha, Bisha 61922, Saudi Arabia
Prince Sattam bin Abdulaziz University, Al-Kharj, Saudi Arabia
Department of Physics, College of Science, Jouf University, Sakaka, Aljouf 72341, Saudi Arabia
Show Author Information

Abstract

Predictive models for early identification of heart disease must be precise and efficient because it is a major worldwide health concern. To improve classification performance while protecting data privacy, this study investigated a combined method that uses ensemble learning, feature selection, and federated learning (FL). The ensemble-based approaches proved the most predictive after testing several different machine learning (ML) models, including random forests, the light gradient boosting machine, support vector machines, k-nearest neighbors, convolutional neural networks, and long short-term memory. We used particle swarm optimization (PSO) for feature selection, which optimized the most relevant features in conjunction with voting and stacking approaches to further increase the model's performance. In addition, federated learning was implemented to allow decentralized training while preserving sensitive medical data. The results highlight the effectiveness of combining these techniques in the detection of heart disease, providing a scalable and privacy-preserving solution for real-world healthcare applications. Two benchmark datasets were used to validate the proposed approach, ensuring the reliability and generalizability of the findings. Furthermore, we used four performance metrics, namely accuracy, precision, recall, and F1score, to evaluate the selected models. Finally, federated learning was included to handle privacy issues and guarantee safe access to private medical data. This distributed method allows model training without centralizing patient data, so it is compatible with strict data privacy rules. With up to 95% precision, our method shows a notable increase in prediction accuracy according to the testing results. This work offers a strong, scalable, and safe solution for the early identification of cardiovascular diseases by combining ensemble learning, feature selection, and federated learning, opening the way for more general uses in medical diagnostics.

CLC number: 62H30, 68T05, 92C50, 68U35, 90C59

References

【1】
【1】
 
 
AIMS Mathematics
Pages 7290-7318

{{item.num}}

Comments on this article

Go to comment

< Back to all reports

Review Status: {{reviewData.commendedNum}} Commended , {{reviewData.revisionRequiredNum}} Revision Required , {{reviewData.notCommendedNum}} Not Commended Under Peer Review

Review Comment

Close
Close
Cite this article:
Hrizi O, Gasmi K, Alyami A, et al. Federated and ensemble learning framework with optimized feature selection for heart disease detection. AIMS Mathematics, 2025, 10(3): 7290-7318. https://doi.org/10.3934/math.2025334

69

Views

0

Downloads

3

Crossref

1

Web of Science

4

Scopus

Received: 08 February 2025
Revised: 17 March 2025
Accepted: 24 March 2025
Published: 15 March 2025
©2025 the Author(s), licensee AIMS Press.

This is an open access article distributed under the terms of the Creative Commons Attribution License (https://creativecommons.org/licenses/by/4.0)