Share your thoughts, 1 month free Claude Pro on usSee more
WorkDL logo mark

ScAle: Attention Head Scaling as a Minimal Adapter for Spatial Reasoning in Vision Language Models

About

Spatial reasoning remains a persistent challenge for many vision language models (VLMs), and improving it typically requires fine-tuning with substantial additional parameters. Our preliminary analysis reveals that rescaling activations in selected transformer layers-without modifying pretrained weights-can significantly influence downstream performance. Motivated by this observation, we propose ScAle, an ultra-lightweight adaptation method that learns a small set of scalar coefficients to modulate last-token attention and MLP activations in a fully frozen backbone. We evaluate our method on the synthetic spatial reasoning benchmark SpatialEval and on real-world VQA datasets (COCOQA and VGQA) across multiple model families. Our method, ScAle, achieves up to 134.1% relative accuracy gains using only 1K trainable parameters without requiring millions of trainable parameters as in standard PEFT methods such as LoRA. Despite its extreme compactness, our approach recovers a substantial fraction of standard PEFT performance while preserving strong non-spatial VQA accuracy. These results demonstrate that bounded activation reweighting provides a simple, architecture-agnostic, and highly parameter-efficient alternative for adapting pretrained VLMs.

Rahul Chowdhury, Timothy A Rupprecht, Xuan Shen, Pu Zhao, Yanzhi Wang• 2026

Related benchmarks

TaskDatasetResultRank
Hallucination EvaluationPOPE
Accuracy90.4
281
Spatial ReasoningSpatialEval Spatial-Grid (20% train)
Accuracy83.8
20
Spatial ReasoningSpatialEval Maze-Nav 20% (train)
Accuracy70.3
20
Spatial ReasoningSpatialEval Spatial-Map (20% train)
Accuracy (Spatial-Map)76.8
20
Spatial ReasoningWhatsUp-VLM VGQA (test)
Accuracy0.935
6
Spatial ReasoningWhatsUp-VLM COCOQA (test)
Accuracy95
6
Showing 6 of 6 rows

Other info

Follow for update