A hybrid classical-quantum approach for medical image classification
TL;DR
The results show that there is a statistically significant improvement in classification accuracy and F1-score of the hybrid approach over the classical transfer learning baselines under a simulation-based environment, which sets the stage for further research on real-world physical quantum hardware.
Abstract
With millions of medical data being produced daily, it is important to leverage current technology to improve the healthcare system. As traditional deep learning models are unable to effectively handle high-dimensional data, a hybrid classical-quantum approach is used to enhance traditional deep learning algorithms. The hybrid classical-quantum models are built upon the classical deep learning framework with quantum techniques. The classical layers are utilized for feature extraction while the quantum layer is used for computational tasks. This addresses computational limitations and data imbalance challenges in medical image analysis. Transfer learning models, such as ResNet18 and InceptionV3, are used in conjunction with a quantum variational circuit to perform the task. The model is trained on a chest X-ray dataset, which consists of two sets of data: binary and multiclass. The binary dataset consists of Normal and COVID-19 images, while the multiclass dataset has COVID-19, Normal, and Tuberculosis images for classification. PennyLane, as a simulator, has been used to perform quantum computing. The dataset has also been used to train the traditional transfer learning models, pre-trained ResNet18, and their results have been compared. Out of all the models tested, the hybrid classical-quantum model, namely Hybrid ResNet18, performed the best. It achieved an accuracy of 93.22% for the binary classification task and 93.91% for the multiclass classification. The results show that there is a statistically significant improvement in classification accuracy and F1-score of the hybrid approach over the classical transfer learning baselines under a simulation-based environment. This sets the stage for further research on real-world physical quantum hardware. Class imbalance and overlapping predictions have been addressed using methods such as data augmentation and confusion matrix analysis. Thus, quantum machine learning could be used as a vital tool to analyze medical data for better diagnostic results.