Inverting Visual Representations with Detection Transformers

Jan Rathjens,Shirin Reyhanian,David Kappel,Laurenz Wiskott
2024-12-09
Abstract:Understanding the mechanisms underlying deep neural networks in computer vision remains a fundamental challenge. While many prior approaches have focused on visualizing intermediate representations within deep neural networks, particularly convolutional neural networks, these techniques have yet to be thoroughly explored in transformer-based vision models. In this study, we apply the approach of training inverse models to reconstruct input images from intermediate layers within a Detection Transformer, showing that this approach is efficient and feasible for transformer-based vision models. Through qualitative and quantitative evaluations of reconstructed images across model stages, we demonstrate critical properties of Detection Transformers, including contextual shape preservation, inter-layer correlation, and robustness to color perturbations, illustrating how these characteristics emerge within the model's architecture. Our findings contribute to a deeper understanding of transformer-based vision models. The code for reproducing our experiments will be made available at <a class="link-external link-http" href="http://github.com/wiskott-lab/inverse-detection-transformer" rel="external noopener nofollow">this http URL</a>.
Computer Vision and Pattern Recognition,Machine Learning
What problem does this paper attempt to address?