Visual Saliency Steering Distillation for Multimodal Chain-of-Thought Reasoning
Summary
Visual Saliency Steering Distillation (VSSD) is a novel method designed to enhance multimodal chain-of-thought (CoT) reasoning in small models by addressing the suppression of tiny cross-modal differences during modality-interaction fusion. This issue particularly affects scenarios where different images pair with identical text or vice-versa. VSSD utilizes attention maps from multimodal large language models (MLLMs) like LLaVA-1.5 to create perturbed images that capture task-sensitive feature directions. It then applies singular value decomposition (SVD) to extract dominant steering vectors, which are injected into the intermediate layers of the student model's decoder via distillation. Experiments on the ScienceQA and M3CoT datasets, using a T5-base model with a frozen DETR visual encoder, demonstrate that VSSD consistently improves rationale generation and answer inference accuracy compared to various baselines.
Key takeaway
For Machine Learning Engineers developing small multimodal CoT models, if you struggle with distinguishing subtle visual-textual differences, consider implementing Visual Saliency Steering Distillation (VSSD). This technique, by generating attention-guided perturbed images and distilling steering vectors into decoder layers, demonstrably improves rationale generation and answer accuracy on datasets like ScienceQA and M3CoT. You should focus on extracting steering vectors from the last two decoder layers for optimal performance.
Key insights
VSSD enhances small multimodal CoT models' sensitivity to fine-grained visual-semantic differences through saliency-guided distillation.
Principles
- Modality fusion can suppress fine-grained cross-modal cues.
- Perturbing salient visual regions highlights task-sensitive features.
- Steering vectors from feature differences guide distillation.
Method
VSSD generates perturbed images using MLLM attention maps, computes steering vectors from feature differences via SVD, and injects them into intermediate decoder layers for distillation.
In practice
- Use LLaVA-1.5 for attention map generation.
- Mask top-0.1% salient image pixels.
- Extract steering vectors from the last two decoder layers.
Topics
- Multimodal Chain-of-Thought
- Visual Saliency
- Knowledge Distillation
- Steering Vectors
- Large Language Models
- ScienceQA
- M3CoT
Code references
Best for: Research Scientist, AI Scientist, Machine Learning Engineer
Related on AIssential
See Counsel's argued verdicts on the open AI decisions leaders are weighing →
Editorial summary, takeaway, and curation by AIssential. Original article published by cs.CV updates on arXiv.org.