Abstract:Neural networks have proven to be very successful in automatically capturing the composition of language and different structures across a range of multi-modal tasks. Thus, an important question to investigate is how neural networks learn and organise such structures. Numerous studies have examined the knowledge captured by language models (LSTMs, transformers) and vision architectures (CNNs, vision transformers) for respective uni-modal tasks. However, very few have explored what structures are acquired by multi-modal transformers where linguistic and visual features are combined. It is critical to understand the representations learned by each modality, their respective interplay, and the task's effect on these representations in large-scale architectures. In this paper, we take a multi-modal transformer trained for image captioning and examine the structure of the self-attention patterns extracted from the visual stream. Our results indicate that the information about different relations between objects in the visual stream is hierarchical and varies from local to a global object-level understanding of the image. In particular, while visual representations in the first layers encode the knowledge of relations between semantically similar object detections, often constituting neighbouring objects, deeper layers expand their attention across more distant objects and learn global relations between them. We also show that globally attended objects in deeper layers can be linked with entities described in image descriptions, indicating a critical finding - the indirect effect of language on visual representations. In addition, we highlight how object-based input representations affect the structure of learned visual knowledge and guide the model towards more accurate image descriptions. A parallel question that we investigate is whether the insights from cognitive science echo the structure of representations that the current neural architecture learns. The proposed analysis of the inner workings of multi-modal transformers can be used to better understand and improve on such problems as pre-training of large-scale multi-modal architectures, multi-modal information fusion and probing of attention weights. In general, we contribute to the explainable multi-modal natural language processing and currently shallow understanding of how the input representations and the structure of the multi-modal transformer affect visual representations.

The Double Helix inside the NLP Transformer

The NLP Engine: A Universal Turing Machine for NLP

An Intrinsic Dimension Perspective of Transformers for Sequential Modeling

Unleashing the Power of Transformer for Graphs

Reach the Remote Neighbors: Dual-Encoding Transformer for Graphs

Analyzing Transformers in Embedding Space

An Augmented Transformer Architecture for Natural Language Generation Tasks

Design of a Modified Transformer Architecture Based on Relative Position Coding

Understanding and Improving Transformer from a Multi-Particle Dynamic System Point of View.

Are there identifiable structural parts in the sentence embedding whole?

Better Explain Transformers by Illuminating Important Information

Context-Sensitive Visualization of Deep Learning Natural Language Processing Models

Transformer with Layer Fusion and Interaction

Analyzing Transformer Dynamics as Movement through Embedding Space

What Does a Language-And-Vision Transformer See: The Impact of Semantic Information on Visual Representations

Investigating semantic subspaces of Transformer sentence embeddings through linear structural probing

Analyzing Encoded Concepts in Transformer Language Models

Theoretical Understanding of In-Context Learning in Shallow Transformers with Unstructured Data

A Unified Framework for Interpretable Transformers Using PDEs and Information Theory

Bidirectional Transformer with Absolute-Position Aware Relative Position Encoding for Encoding Sentences

The Bottom-up Evolution of Representations in the Transformer: A Study with Machine Translation and Language Modeling Objectives