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

Amandeep Singh Bhatia, Mandeep Kaur Saggi, Sabre Kais

Keywords: collaborative learning, quantum machine learning, federated learning, healthcare

Abstract:

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 data 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.

Track: 11. General Track

Registration Id: SKNJTHMW536

Submission Number: 332