Adaptive federated learning model with unsupervised autoencoder-based reconstruction for imbalanced medical image classification

Zainal Abidin, Nadzurah and Ismail, Amelia Ritahani and UNSPECIFIED and UNSPECIFIED (2026) Adaptive federated learning model with unsupervised autoencoder-based reconstruction for imbalanced medical image classification. In: 2025 10th International Conference on Information and Communication Technology for the Muslim World (ICT4M), 26-27 November 2025, KUALA LUMPUR, Malaysia.

PDF - Published Version Restricted to Repository staff only Download (966kB) | Request a copy

Abstract

Medical image classification has significantly advanced due to deep learning techniques, however, the performance remains limited by class imbalance and the nature distribution of healthcare data. Federated Learning (FL) allows collaborative model training across multiple data sources, without requiring the sharing of raw data, but its performance degrades under imbalanced and non-independent and identically distributed (non-IID) settings. This paper proposes an enhanced federated learning framework by integrating two key strategies: unsupervised autoencoder-based reconstruction, which identify and prioritises rare cases, and adaptive tuning, which dynamically adjusts aggregation behavior to balance contributions under imbalanced data conditions. These strategies improving the model’s sensitivity for underrepresented classes while maintaining stability under non-IID scenarios. The enhanced model was experimentally evaluated on two benchmark medical datasets, RetinaMNIST and PneumoniaMNIST, under three imbalance ratios (1 to 10, 1 to 5, and 1 to 2) and compared against standard FL, FL with GAN augmentation, and FL with autoencoder-based reconstruction. Results demonstrate that on RetinaMNIST, the enhanced model improved minority-class recall by 59.6 percent, F1-score by 33.9 percent over standard FL. On PneumoniaMNIST, macro recall, F1-score and AUC-ROC increased by 4.6, 4.7, and 3.3 percent respectively over the best baseline (FL with autoencoder-based reconstruction). These consistent gains indicate the framework’s scalability and robustness for federated medical image classification under real- world imbalance conditions.

Item Type:	Proceeding Paper (Plenary Papers)
Uncontrolled Keywords:	federated learning, medical image classification, class imbalance, non-IID data, adaptive tuning, autoencoder
Subjects:	Q Science > QA Mathematics > QA75 Electronic computers. Computer science
Kulliyyahs/Centres/Divisions/Institutes (Can select more than one option. Press CONTROL button):	Kulliyyah of Information and Communication Technology > Department of Computer Science Kulliyyah of Information and Communication Technology > Department of Computer Science Kulliyyah of Information and Communication Technology Kulliyyah of Information and Communication Technology
Depositing User:	Amelia Ritahani Ismail
Date Deposited:	19 Feb 2026 12:02
Last Modified:	19 Feb 2026 12:02
Queue Number:	2026-02-Q2123
URI:	http://irep.iium.edu.my/id/eprint/127405

Actions (login required)

View Item