IC-019Visual token representations in LLaVA-1.5 evolve to align with interpretable text tokens

Clement Neo, Luke Ong, Philip Torr, Mor Geva, David Krueger, Fazl Barez

SourceTowards Interpreting Visual Information Processing in Vision-Language Models

Using the logit lens technique, the study analyzed how the representations of visual tokens evolve across layers in the language model of LLaVA-1.5. In mid-to-late layers, decoding the activation at a visual token position often yields a token that describes the object in the corresponding image patch. Quantitatively, in the best-performing layer, an average of 23.7% of object patch token positions correspond to the correct object class token. This refinement occurs despite the model not being pre-trained on next-token prediction for visual inputs. A similar pattern was observed for Qwen2VL-2B, where 6.5% of tokens corresponded correctly at the best layer.

Evidence
correlational
Key metric
23.7% of object patch token positions correspond to the correct object class token at the best-performing layer (layer 25.7). For Qwen2VL, 6.5% correspondence at layer 25.1.
Caveat
The logit lens is an inference technique and may not perfectly reflect the model's internal processing. The numbers reported are averages from a specific set of images from the COCO validation set.
Model
LLaVA-1.5 / LLaVA-v1.5, Qwen2-VL
Concepts
Linear representation, Depth-dependent structure
Datasets
MS COCO / COCO / COCO 2014 / COCO 2017 / COCO 20k / COCO-it / COCO-wl [eval]
Methods
Logit lens [primary]
Extraction
automatic-extraction