Prosiding Seminar Nasional Ilmu Komputer, Sosial Sains, Teknik Dan Multi-disiplin Ilmu
Vol. 2 (2026)

Implementation of Federated Learning for Disease Classification with Imbalanced Dataset

Syarifah Yusnaini Putri (Universitas Medan Area)
Nia Ramadani (Universitas Medan Area)
Andre Hasudungan Lubis (Universitas Medan Area)



Article Info

Publish Date
24 Jun 2026

Abstract

Imbalanced data is one of the main challenges in applying machine learning to stroke disease classification, where the number of stroke patient data is significantly smaller than non-stroke patient data. This condition causes classification models to become biased toward the majority class, resulting in poor detection of stroke cases. In addition, centralized learning approaches raise privacy and security concerns due to the sensitive nature of patient data. Therefore, this study proposes a Federated Learning Random Forest (FL-RF) approach to improve classification performance on imbalanced data while preserving data privacy. The study uses the Cerebral Stroke Prediction – Imbalanced Dataset obtained from Kaggle. The research stages include data preprocessing, semi non-IID data distribution across multiple clients, local training using Balanced Random Forest, and model aggregation within a federated learning environment. Model evaluation was conducted using accuracy, precision, recall, F1-score, confusion matrix, and ROC curve metrics. The results show that the conventional Random Forest achieved an accuracy of 0.9817 but failed to detect the minority class. Meanwhile, the proposed FL-RF model obtained an accuracy of 0.7976 with an improved recall of 0.56 and an F1-score of 0.0910 for the stroke class. These findings indicate that the FL-RF approach is more effective in improving sensitivity toward minority classes compared to conventional Random Forest.  

Copyrights © 2026






Journal Info

Abbrev

ikosstemi

Publisher

Subject

Religion Humanities Computer Science & IT Decision Sciences, Operations Research & Management Education Electrical & Electronics Engineering Engineering

Description

Prosiding Seminar Nasional Ilmu Komputer, Sosial Sains, Teknik dan Multi-Disiplin Ilmu (IKOSSTEMI) dengen No. ISSN: 3123-2574 (online) merupakan salah satu kegiatan yang diselenggarakan oleh Raskha Media Group Publisher. Seminar Nasional IKOSSTEMI (Ilmu Komputer, Sosial Sains, Teknik dan ...