直接优化删除/插入指标,生成更精准的视觉解释图。
Learn to Rank: Visual Attribution by Learning Importance Ranking
- 用可微分排列学习替代硬排序,端到端训练解释模型。
- 在视觉变换器上实现更清晰、边界对齐的像素级解释。
- 适合需要高可信度解释的医疗、自动驾驶等安全场景。
解释复杂计算机视觉模型的决策对建立信任和问责制至关重要,尤其在安全关键领域。现有方法存在三重权衡:基于传播的方法高效但易受偏差影响且依赖架构;基于扰动的方法因果性更强,但成本高,对视觉变换器常产生粗粒度的块级解释;基于学习的解释器虽快速,但通常优化代理目标或从启发式教师中蒸馏。本文提出一种学习方案,直接优化删除和插入指标。由于这些指标依赖不可微的排序与排名,我们将其建模为排列学习问题,并使用Gumbel-Sinkhorn对硬排序进行可微松弛。这使得通过目标模型的归因引导扰动实现端到端训练成为可能。推理时,该方法可在单次前向传播中生成密集的像素级归因,支持少量梯度精修。实验表明,该方法在量化指标上持续提升,且解释结果更锐利、边界对齐,尤其适用于基于变换器的视觉模型。
原文摘要 · Abstract (English)
Interpreting the decisions of complex computer vision models is crucial to establish trust and accountability, especially in safety-critical domains. An established approach to interpretability is generating visual attribution maps that highlight regions of the input most relevant to the model's prediction. However, existing methods face a three-way trade-off. Propagation-based approaches are efficient, but they can be biased and architecture-specific. Meanwhile, perturbation-based methods are causally grounded, yet they are expensive and for vision transformers often yield coarse, patch-level explanations. Learning-based explainers are fast but usually optimize surrogate objectives or distill from heuristic teachers. We propose a learning scheme that instead optimizes deletion and insertion metrics directly. Since these metrics depend on non-differentiable sorting and ranking, we frame them as permutation learning and replace the hard sorting with a differentiable relaxation using Gumbel-Sinkhorn. This enables end-to-end training through attribution-guided perturbations of the target model. During inference, our method produces dense, pixel-level attributions in a single forward pass with optional, few-step gradient refinement. Our experiments demonstrate consistent quantitative improvements and sharper, boundary-aligned explanations, particularly for transformer-based vision models.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。