Same Task, Different Circuits: Disentangling Modality-Specific Mechanisms in VLMs
This work addresses the multi-modal performance gap in VLMs, which is an incremental improvement for enhancing model accuracy in vision-language tasks.
The paper tackled the performance gap between visual and textual modalities in Vision-Language Models (VLMs) by analyzing task-specific circuits, finding that disjoint circuits process modality-specific data positions, and proposed patching visual representations from later to earlier layers, which closed one-third of the performance gap on average in experiments.
Vision-Language models (VLMs) show impressive abilities to answer questions on visual inputs (e.g., counting objects in an image), yet demonstrate higher accuracies when performing an analogous task on text (e.g., counting words in a text). We investigate this accuracy gap by identifying and comparing the \textit{circuits} - the task-specific computational sub-graphs - in different modalities. We show that while circuits are largely disjoint between modalities, they implement relatively similar functionalities: the differences lie primarily in processing modality-specific data positions (an image or a text sequence). Zooming in on the image data representations, we observe they become aligned with the higher-performing analogous textual representations only towards later layers, too late in processing to effectively influence subsequent positions. To overcome this, we patch the representations of visual data tokens from later layers back into earlier layers. In experiments with multiple tasks and models, this simple intervention closes a third of the performance gap between the modalities, on average. Our analysis sheds light on the multi-modal performance gap in VLMs and suggests a training-free approach for reducing it.