VL-InterpreT: An Interactive Visualization Tool for Interpreting Vision-Language Transformers
This addresses the problem of understanding black-box multimodal transformers for researchers and practitioners, though it is incremental as it builds on existing interpretability concepts for NLP.
The paper tackles the lack of interpretability tools for vision-language transformers by introducing VL-InterpreT, an interactive visualization tool that tracks attention and hidden representations, and demonstrates its functionalities on tasks like Visual Commonsense Reasoning and WebQA, revealing insights into model behaviors.
Breakthroughs in transformer-based models have revolutionized not only the NLP field, but also vision and multimodal systems. However, although visualization and interpretability tools have become available for NLP models, internal mechanisms of vision and multimodal transformers remain largely opaque. With the success of these transformers, it is increasingly critical to understand their inner workings, as unraveling these black-boxes will lead to more capable and trustworthy models. To contribute to this quest, we propose VL-InterpreT, which provides novel interactive visualizations for interpreting the attentions and hidden representations in multimodal transformers. VL-InterpreT is a task agnostic and integrated tool that (1) tracks a variety of statistics in attention heads throughout all layers for both vision and language components, (2) visualizes cross-modal and intra-modal attentions through easily readable heatmaps, and (3) plots the hidden representations of vision and language tokens as they pass through the transformer layers. In this paper, we demonstrate the functionalities of VL-InterpreT through the analysis of KD-VLP, an end-to-end pretraining vision-language multimodal transformer-based model, in the tasks of Visual Commonsense Reasoning (VCR) and WebQA, two visual question answering benchmarks. Furthermore, we also present a few interesting findings about multimodal transformer behaviors that were learned through our tool.