HAWAII: Hierarchical Visual Knowledge Transfer for Efficient Vision-Language Models

TL;DR

HAWAII introduces a hierarchical knowledge transfer framework that distills multiple visual experts into a single efficient vision encoder, improving vision-language task performance with minimal computational costs.

Contribution

The paper presents a novel hierarchical distillation method using teacher-specific LoRA adapters and a router to combine multiple visual experts into one efficient model.

Findings

Outperforms popular open-source VLMs on various tasks.

Reduces computational costs during training and inference.

Effectively integrates diverse visual knowledge into a single model.

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.