Affiliation:
1. School of Computer and Systems Sciences Jawaharlal Nehru University New Delhi India
Abstract
AbstractIn this study, we propose a novel Federated Learning Based Multi‐Head Attention (FBMA) framework for image classification problems considering the Independent and Identically Distributed (IID) and Non‐Independent and Identically Distributed (Non‐IID) medical data. The FBMA architecture integrates FL principles with the Multi‐Head Attention mechanism, optimizing the model performance and ensuring privacy. Using Multi‐Head Attention, the FBMA framework allows the model to selectively focus on important regions of the image for feature extraction, and using FL, FBMA leverages decentralized medical institutions to facilitate collaborative model training while maintaining data privacy. Through rigorous experimentation on medical image datasets: MedMNIST Dataset, MedicalMNIST Dataset, and LC25000 Dataset, each partitioned into Non‐IID data distribution, the proposed FBMA framework exhibits high‐performance metrics. The results highlight the efficacy of our proposed FBMA framework, indicating its potential for real‐world applications where image classification demands both high accuracy and data privacy.
Reference51 articles.
1. XuJ WangF.Federated learning for healthcare informatics. CoRR.2019; abs/1911.06270.
2. HIPAA Regulations — A New Era of Medical-Record Privacy?
3. de laTorreL.A guide to the california consumer privacy act of 2018. Available at SSRN 3275571.2018.
4. Regulation (EU) 2016/679 of the European Parliament and of the council of 27 April 2016 on the protection of natural persons with regard to the processing of personal data and on the free movement of such data (general data protection regulation);European Parliament and Council of the European Union;Off J Eur Union,2016
5. KonečnýJ McMahanHB YuFX RichtárikP SureshAT BaconD.Federated learning: strategies for improving communication efficiency. CoRR.2016; abs/1610.05492.