From explanation to unsupervised segmentation: fusion of multiple explanation maps for vision transformers
摘要
Vision transformers (ViTs) achieve state-of-the-art accuracy in recognition and segmentation, yet their self-attention makes decisions difficult to interpret. We call, and will refer to our method as ViTMix. Unlike prior CNN oriented ensemble explainers, ViTMix is tailored to ViTs and transfers to medical data with only brief target-domain fine tuning. Single-method explainers, Gradient Saliency, Grad-CAM, Layer-wise Relevance Propagation (LRP), or Attention Rollout, capture only partial evidence and often yield noisy heat maps. We introduce a model-agnostic post hoc fusion framework that supports combining multiple attribution maps via element wise multiplication and geometric mean. The mixed map keeps pixels highlighted by multiple methods while suppressing isolated artifacts, a behavior we motivate analytically via the Pigeonhole Principle. ViTMix operates on any ViT classifier; its sole requirement is a model that outputs class logits and-aside from the light fine tuning used on PH