Privacy-Preserving Federated Learning for Thyroid Disease Detection: A Comparative Study on Feature Selection and Class Balancing
Detecting thyroid disorders early and correctly is very important for good medical care. Real clinical data from the UCI repository is used in this study to show a full federated learning system for identifying thyroid diseases. The technique includes a lot of data preprocessing, like handling categorical and numerical features, missing values, and normalisation. It also includes five cutting-edge class balancing methods (SMOTE, BorderlineSMOTE, ADASYN, RandomOverSampler, and SMOTEENN) and two advanced feature selection techniques (SelectKBest using ANOVA F-score and Recursive Feature Elimination with logistic regression). Each balanced and feature-selected datasetwas given to five federated clients so that they could work together to train a deep neural network using the Flower federated learning framework. This kept patient data private. The model’s performance was measured over 10 communication rounds using F1-score, accuracy, precision, recall, and the combination of SelectKBest and SMOTEENN. This combination achieved the highest F1-score (0.9123) and accuracy (0.9133) in finding cases of the minority class. These results showhowimportant it is for shared healthcare AI to include feature selection and class balance. SHAP was used to explain model predictions; T3, TSH, and FTI emerged as the strongest contributors, consistent with clinical knowledge. A permutation-importance analysis based on AUROC drop produced the same ordering, providing an independent, model-agnostic check. This will allow for a scalable and private way to accurately screen for thyroid problems in clinical settings.
Authors
- R. Kannadasan
- Divya Kesavulu (ORCID: https://orcid.org/0000-0001-8530-9012)
Institutions
- Vellore Institute of Technology University (IN)
Publication Details
- Journal
- IEEE Access
- Published
- 2025-01-01
- DOI
- https://doi.org/10.1109/access.2025.3627948
- Primary Topic
- Artificial Intelligence in Healthcare
- Type
- article
- Field-Weighted Citation Impact
- 0.00