cs.LGApr 28, 2026

Towards interpretable AI with quantum annealing feature selection

Authors: Francesco Aldo VenturelliEmanuele CostaSikha O KBruno Juliá-DíazMiguel A. González BallesterAlba Cervera-Lierta

Organizations: 1BCN Medtech, Universitat Pompeu Fabra, Barcelona, Spain · 2Barcelona Supercomputing Center (BSC) · 3Departament de F´ısica Qu`antica i Astrof´ısica, Facultat de F´ısica, Universitat de Barcelona, E-08028 Barcelona, Spain · 4Institut de Ci`encies del Cosmos, Universitat de Barcelona, ICCUB, Mart´ı i Franqu`es 1, E-08028 Barcelona, Spain. · 5ICREA, Barcelona, Spain.

Abstract

Deep learning models are used in critical applications, in which mistakes can have serious consequences. Therefore, it is crucial to understand how and why models generate predictions. This understanding provides useful information to check whether the model is learning the right patterns, detect biases in the data, improve model design, and build systems that can be trusted. This work proposes a new method for interpreting Convolutional Neural Networks in image classification tasks. The approach works by selecting the most representative feature maps that contribute to each prediction. To solve this combinatorial problem, we encode it into a quantum constrained optimization problem and propose to solve it using quantum annealing. We evaluate our method against the state-of-the-art explainable AI techniques, specifically GradCAM and GradCAM++, and observe an improved class disentanglement, i.e. the model's decision boundaries become more distinct and its reasoning more transparent. This demonstrates that our approach enhances the quality of explanations, making it easier to understand which features the model relies on for specific predictions. In addition, we study the computational behavior of the quantum annealing algorithm. Specifically, we analyze the minimum energy gap of the system during computation and the probability that the algorithm finds the correct solution. These analyses provide theoretical insight into why the method works effectively in practice.

Explore similar work

CardsList
  1. Learning Quantifiable Visual Explanations Without Ground-Truth

    May 18, 2026Amritpal Singh, Andrey Barsky, Mohamed Ali Souibgui +2XaiExplainable Artificial Intelligence