Building Continuous Quantum-Classical Bayesian Neural Networks for a Classical Clinical Dataset

Abstract

In this work, we are introducing a Quantum-Classical Bayesian Neural Network (QCBNN) that is capable to perform uncertainty-aware classification of classical medical dataset. This model is a symbiosis of a classical Convolutional NN that performs ultra-sound image processing and a quantum circuit that generates its stochastic weights, within a Bayesian learning framework. To test the utility of this idea for the possible future deployment in the medical sector we track multiple behavioral metrics that capture both predictive performance as well as model's uncertainty. It is our ambition to create a hybrid model that is capable to classify samples in a more uncertainty aware fashion, which will advance the trustworthiness of these models and thus bring us step closer to utilizing them in the industry. We test multiple setups for quantum circuit for this task, and our best architectures display bigger uncertainty gap between correctly and incorrectly identified samples than its classical benchmark at an expense of a slight drop in predictive performance. The innovation of this paper is two-fold: (1) combining of different approaches that allow the stochastic weights from the quantum circuit to be continues thus allowing the model to classify application-driven dataset; (2) studying architectural features of quantum circuit that make-or-break these models, which pave the way into further investigation of more informed architectural designs.

References in corpus (13)

• MedMNIST v2 -- A large-scale lightweight benchmark for 2D and 3D biomedical image classification
• Hands-on Bayesian Neural Networks -- a Tutorial for Deep Learning Users
• The effect of data encoding on the expressive power of variational quantum machine learning models
• Power of data in quantum machine learning
• Data re-uploading for a universal quantum classifier
• Quantum Generative Adversarial Networks for Learning and Loading Random Distributions
• The Born Supremacy: Quantum Advantage and Training of an Ising Born Machine
• Bayesian Deep Learning on a Quantum Computer
• Classical surrogates for quantum learning models
• Variational inference with a quantum computer
• Variational quantum generators: Generative adversarial quantum machine learning for continuous distributions
• Do Quantum Circuit Born Machines Generalize?
• Quantum-Aided Meta-Learning for Bayesian Binary Neural Networks via Born Machines