Visual Saliency Steering Distillation for Multimodal Chain-of-Thought Reasoning

· Source: cs.CV updates on arXiv.org · Field: Technology & Digital — Artificial Intelligence & Machine Learning · Depth: Expert, long

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

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

Topics

Code references

Best for: Research Scientist, AI Scientist, Machine Learning Engineer

Related on AIssential

Open in AIssential →

Editorial summary, takeaway, and curation by AIssential. Original article published by cs.CV updates on arXiv.org.