Federated Learning in Predicting Heart Disease

Mehmet Haklıdır,Rusen Halepmollasi,Havvanur Dervisoglu,Burak Ülver,Rabia Arkan Yurtoglu
DOI: https://doi.org/10.1109/SIU59756.2023.10223935
2023-07-05
Abstract:Heart diseases are a significant health problem worldwide, and early diagnosis and prediction are of great im-portance. Artificial intelligence and machine learning techniques have become effective tools in predicting heart diseases. However, when traditional centralized approaches are used, challenges arise in terms of data privacy and security. The federated learning approach is presented as an important solution to address these issues. In this study, our aim is to predict heart diseases while preserving patient privacy and confidentiality using the federated learning approach. Experiments were conducted using the open-source dataset "PTB Diagnostic ECG Database" to compare the results of centralized and federated learning. Through this comparison, it has been shown that federated learning can achieve results close to centralized learning in cases where sharing sensitive data such as health data is problematic. Additionally, in the experiments conducted using different feature selection and machine learning models, both logistic regression and support vector machines showed the best accuracy results with mRMR feature selection in both centralized (86%) and federated training (85%). When examining the F1-Score results, applying feature selection in both classification models allowed for the best results (federated learning: 91%, centralized learning: 92%).
Computer Science,Medicine
What problem does this paper attempt to address?