Our new X account is live! Follow @wizwand_team for updates
WorkDL logo mark

Masking Teacher and Reinforcing Student for Distilling Vision-Language Models

About

Large-scale vision-language models (VLMs) have recently achieved remarkable multimodal understanding, but their massive size makes them impractical for deployment on mobile or edge devices. This raises the need for compact yet capable VLMs that can efficiently learn from powerful large teachers. However, distilling knowledge from a large teacher to a small student remains challenging due to their large size gap: the student often fails to reproduce the teacher's complex, high-dimensional representations, leading to unstable learning and degraded performance. To address this, we propose Masters (Masking Teacher and Reinforcing Student), a mask-progressive reinforcement learning (RL) distillation framework. Masters first masks non-dominant weights of the teacher to reduce unnecessary complexity, then progressively restores the teacher by gradually increasing its capacity during training. This strategy allows the student to learn richer representations from the teacher in a smooth and stable manner. To further refine knowledge transfer, Masters integrates an offline RL stage with two complementary rewards: an accuracy reward that measures the correctness of the generated responses, and a distillation reward that quantifies the ease of transferring responses from teacher to student. Unlike online think-answer RL paradigms that are computationally expensive and generate lengthy responses, our offline RL leverages pre-generated responses from masked teachers. These provide rich yet efficient guidance, enabling students to achieve strong performance without requiring the think-answer process.

Byung-Kwan Lee, Yu-Chiang Frank Wang, Ryo Hachiuma• 2025

Related benchmarks

TaskDatasetResultRank
Multimodal Capability EvaluationMM-Vet
Score85.6
282
Multi-discipline Multimodal UnderstandingMMMU
Accuracy74
266
Chart Question AnsweringChartQA
Accuracy95.9
229
Multimodal UnderstandingMMStar
Accuracy82
197
Diagram Question AnsweringAI2D
AI2D Accuracy88.9
196
Visual Mathematical ReasoningMathVista
Accuracy85
189
Multimodal UnderstandingSEED
Accuracy82.6
136
Multimodal EvaluationMM-Vet
Accuracy85.6
122
Multimodal UnderstandingSEED-2-Plus
Accuracy75.9
99
Real-world Visual Question AnsweringRealworldQA
Accuracy77.8
91
Showing 10 of 17 rows

Other info

Follow for update