arXiv:2604.25649cs.LG2026-04

用量子退火选关键特征图,让卷积网络的决策更透明。

Towards interpretable AI with quantum annealing feature selection

论文配图:Towards interpretable AI with quantum annealing feature selection
图 1 · 摘自论文原文
  • 将特征选择转为量子约束优化问题求解
  • 相比GradCAM等方法,类别区分度更高
  • 适合需要高可信度解释的医疗、金融场景

深度学习模型被广泛应用于关键任务,其错误可能带来严重后果。因此理解模型如何以及为何生成预测至关重要。这有助于检查模型是否学习正确模式、发现数据偏差、改进模型设计并建立可信赖系统。本文提出一种新方法,用于解释图像分类中的卷积神经网络。该方法通过选取对每项预测最具代表性的特征图来实现。为解决这一组合优化问题,将问题编码为量子约束优化问题,并提议使用量子退火求解。在与当前最优可解释AI技术(如GradCAM和GradCAM++)的对比中,观察到更优的类别解耦效果,即模型决策边界更清晰,推理过程更透明。这表明该方法提升了解释质量,使用户更容易理解模型依赖哪些特征进行特定预测。此外,研究了量子退火算法的计算行为,具体分析了计算过程中系统的最小能隙及算法找到正确解的概率,为方法在实践中有效提供了理论依据。

原文摘要 · Abstract (English)

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.

可解释AI量子退火特征选择卷积网络

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。