HAWAII: Hierarchical Visual Knowledge Transfer for Efficient Vision-Language Models
Yimu Wang, Mozhgan Nasr Azadani, Sean Sedwards, Krzysztof Czarnecki
TL;DR
HAWAII addresses the inefficiency of multi-expert vision-language models by distilling knowledge from multiple pretrained visual experts into a single vision encoder. It introduces a mixture-of-LoRA-adapters (MoLA) with teacher-specific and general adapters, guided by a hierarchical knowledge distillation (HKD) framework that operates at coarse- and fine-grained levels, including token importance scoring. The approach yields consistent performance gains over baselines across a broad set of vision-language benchmarks while maintaining low computational overhead. This work enables scalable, high-quality visual perception in VLMs and provides a practical pathway to integrate diverse visual knowledge with modest resource costs.
Abstract
Improving the visual understanding ability of vision-language models (VLMs) is crucial for enhancing their performance across various tasks. While using multiple pretrained visual experts has shown great promise, it often incurs significant computational costs during training and inference. To address this challenge, we propose HAWAII, a novel framework that distills knowledge from multiple visual experts into a single vision encoder, enabling it to inherit the complementary strengths of several experts with minimal computational overhead. To mitigate conflicts among different teachers and switch between different teacher-specific knowledge, instead of using a fixed set of adapters for multiple teachers, we propose to use teacher-specific Low-Rank Adaptation (LoRA) adapters with a corresponding router. Each adapter is aligned with a specific teacher, avoiding noisy guidance during distillation. To enable efficient knowledge distillation, we propose fine-grained and coarse-grained distillation. At the fine-grained level, token importance scores are employed to emphasize the most informative tokens from each teacher adaptively. At the coarse-grained level, we summarize the knowledge from multiple teachers and transfer it to the student using a set of general-knowledge LoRA adapters with a router. Extensive experiments on various vision-language tasks demonstrate the superiority of HAWAII compared to popular open-source VLMs. The code is available at https://github.com/yimuwangcs/wise-hawaii.
