Feature Selection with Quantum Annealing for Biomedical Machine Learning Applications

Wait 5 sec.

Feature selection is a commonly used method in biomedical artificial intelligence and machine learning to identify a subset of high-quality variables that can be used to train downstream predictive models. It has been suggested that quantum feature selection (QFS), which takes advantage of the properties of quantum computers, may better identify variables that are correlated with the outcome while simultaneously reducing redundancy between selected variables. However, there are a limited number of studies evaluating their performance, particularly in real-world data sets. Here, we assess the performance of two QFS methods compared to random forest (RF) feature selection based on feature stability and the performance of a downstream classification algorithm when used to predict urinary tract infections in the emergency department from 211 original features extracted from the electronic health record. We found that a quantum binary quadratic model (BQM) and constrained quadratic model (CQM) had similar performance to RF feature selection (median F1 score of 0.60, 0.61, and 0.61 respectively) when 10 features were selected for an XGBoost classification model. The BQM and RF also had similar feature stability (0.91 and 0.94, respectively) while the CQM had lower stability (0.72). These findings show that QFS can be used with large, clinical data sets to identify features with high stability and predictive performance. As the capacity and quality of quantum computers continue to increase, these methods may offer additional benefits to classical feature selection methods.