Communication-efficient Quantum Federated Learning Optimization for Multi-Center Healthcare Data

• Published in: IEEE-EMBS International Conference on Biomedical and Health Informatics (BHI ...) (Online) pp. 1 - 8
• Main Authors: Singh Bhatia, Amandeep, Saggi, Mandeep Kaur, Kais, Sabre
• Format: Conference Proceeding
• Language: English
• Published: IEEE 10.11.2024
• Subjects: Cancer; Computational modeling; Convergence; Data privacy; Distributed databases; Federated learning; healthcare; Medical services; Optimization; Quantum circuit; quantum machine learning; quantum optimization; Tuning; variational quantum circuit
• ISSN: 2641-3604
• DOI: 10.1109/BHI62660.2024.10913485

In the healthcare sector, the scarcity of data and privacy concerns present formidable challenges to the widespread adoption of machine learning. In the present-day scenario, Federated Learning (FL) emerges as a pivotal solution, fostering the rapid evolution of distributed machine learning paradigms while adeptly addressing the problem of data governance and privacy. It allows distributed clients to collaboratively train a global model by synchronizing their local updates without sharing private data. In recent years, federated learning and quantum computing have individually shown great promise to revolutionize various sectors, including healthcare, finance, and manufacturing, where privacy protection is paramount. In this article, we propose a communication-efficient Quantum Federated Learning (QFL) framework based on a variational circuit that enables clients to efficiently train and transmit quantum model parameters, thereby reducing communication rounds significantly and enhancing QFL performance using quantum natural gradient descent (QNGD) optimization. This paper demonstrates the feasibility of a QFL framework for predicting the presence of coronary heart disease, diagnosing whether a patient is suffering from diabetes or not, and differentiating malignant and benign cancer by distributing the UCI datasets unbalanced among healthcare institutions. The proposed framework has the potential to incorporate privacy, security, and the expedited processing of distributed data. QNGD outperformed classical GD by reducing communication rounds by a range of 5% to 60%. In addition to reducing the communication rounds by optimizing the QFL training algorithm and achieving quicker convergence, it also determines the important features regardless of the data imbalance among the clients.