arXiv:2412.06534cs.CVcs.AI2024-12被引 1

用反演方法揭示视觉Transformer如何编码图像特征

Understanding Transformer-based Vision Models through Inversion

  • 设计模块化反演网络,高效重建中间层特征对应的图像
  • 在Detr和ViT上验证,能清晰还原形状与细节信息
  • 揭示模型对上下文、层级关联及色彩扰动的鲁棒性机制

理解深度神经网络的内在机制仍是机器学习与计算机视觉中的基础挑战。特征反演是一种有前景但尚未充分探索的方法,通过训练逆向神经网络从中间表示中重构图像。本文重新审视该方法,提出一种新型模块化变体,显著提升其在大规模视觉Transformer模型(如Detection Transformer和Vision Transformer)上的应用效率。我们系统地展示了该方法在两类模型上的应用,并通过重构图像的定性分析揭示了其有意义的语义表征。进一步的定量评估揭示了两种Transformer架构中图像特征的编码机制:包括对上下文形状与细节的编码方式、各层之间的相关性,以及对颜色扰动的鲁棒性。这些发现深化了对视觉Transformer内部表示的理解。实验代码已公开于github.com/wiskott-lab/inverse-tvm。

原文摘要 · Abstract (English)

Understanding the mechanisms underlying deep neural networks remains a fundamental challenge in machine learning and computer vision. One promising, yet only preliminarily explored approach, is feature inversion, which attempts to reconstruct images from intermediate representations using trained inverse neural networks. In this study, we revisit feature inversion, introducing a novel, modular variation that enables significantly more efficient application of the technique. We demonstrate how our method can be systematically applied to the large-scale transformer-based vision models, Detection Transformer and Vision Transformer, and how reconstructed images can be qualitatively interpreted in a meaningful way. We further quantitatively evaluate our method, thereby uncovering underlying mechanisms of representing image features that emerge in the two transformer architectures. Our analysis reveals key insights into how these models encode contextual shape and image details, how their layers correlate, and their robustness against color perturbations. These findings contribute to a deeper understanding of transformer-based vision models and their internal representations. The code for reproducing our experiments is available at github.com/wiskott-lab/inverse-tvm.

视觉Transformer特征反演可解释性

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