IEEEJournal Paper2026

MFIFL: Federated Learning With Attention-Based Aggregation for Microscopic Fungal Images

Abstract

An accurate classification of fungal morphology is essential for clinical diagnosis. Centralized deep learning approaches raise concerns about data privacy and struggle with heterogeneous data distributions. To address these challenges, this study proposes an attention-guided federated learning framework for multi-class fungal morphology classification using the DeFungi dataset. The proposed method integrates an improved FedAtt attention mechanism at the client level with an attention-aware federated aggregation strategy to improve global model consistency under non-IID data settings. Extensive experiments demonstrate that the proposed framework achieves stable convergence, balanced class-wise performance, and statistically reliable improvements. The proposed model achieved 93.73% test accuracy, and the FAA aggregation mechanism improved performance by 8.92% over FedAvg in a privacy-preserving federated learning setting. These results highlight the effectiveness of attention-aligned aggregation for robust and privacy-preserving medical image classification, making the proposed framework suitable for real-world clinical deployment.

Key Achievements

Achieved 93.73% test accuracy on multi-class fungal morphology classification
FAA aggregation improved performance by 8.92% over FedAvg
Uses an improved FedAtt client-level attention mechanism
Attention-aware federated aggregation improves consistency under non-IID data
Privacy-preserving framework trained with the DeFungi dataset

Topics

Federated LearningFungal ClassificationMedical ImagingAttention MechanismDeFungiPrivacy