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.
Related benchmarks
| Task | Dataset | Result | Rank | |
|---|---|---|---|---|
| Hallucination Evaluation | POPE | Accuracy90.4 | 281 | |
| Spatial Reasoning | SpatialEval Spatial-Grid (20% train) | Accuracy83.8 | 20 | |
| Spatial Reasoning | SpatialEval Maze-Nav 20% (train) | Accuracy70.3 | 20 | |
| Spatial Reasoning | SpatialEval Spatial-Map (20% train) | Accuracy (Spatial-Map)76.8 | 20 | |
| Spatial Reasoning | WhatsUp-VLM VGQA (test) | Accuracy0.935 | 6 | |
| Spatial Reasoning | WhatsUp-VLM COCOQA (test) | Accuracy95 | 6 |