轻量视觉语言模型对齐瓶颈新解,用记忆检索提升性能
TinyAlign: Boosting Lightweight Vision-Language Models by Mitigating Modal Alignment Bottlenecks
- 通过记忆检索增强多模态输入,缓解语言模型容量不足导致的对齐瓶颈
- 仅用40%微调数据即达基线性能,训练损失降低且收敛更快
- 适合资源受限场景,为轻量级多模态模型提供高效对齐新思路
轻量级视觉语言模型(VLMs)在资源受限场景中至关重要。当前对齐方法通常冻结视觉编码器和语言模型,仅训练小型连接模块,但其性能严重依赖语言模型的内在能力,而轻量模型因表征容量有限,难以实现高质量对齐。本文从互信息视角分析发现,语言模型容量受限会抑制多模态输入与输出间的有效互信息(EMI),从而影响对齐质量。为此,我们提出TinyAlign框架,受检索增强生成启发,从记忆库中动态检索相关上下文以丰富多模态输入,提升对齐效果。大量实验证明,TinyAlign显著降低训练损失,加快收敛速度,并提升任务性能。尤为突出的是,该方法使模型仅需40%的微调数据即可达到基线性能,展现出极强的数据效率。本工作不仅为构建更强大的轻量VLMs提供了实用路径,也引入了理解受限多模态系统中对齐瓶颈的新理论视角。
原文摘要 · Abstract (English)
Lightweight Vision-Language Models (VLMs) are indispensable for resource-constrained applications. The prevailing approach to aligning vision and language models involves freezing both the vision encoder and the language model while training small connector modules. However, this strategy heavily depends on the intrinsic capabilities of the language model, which can be suboptimal for lightweight models with limited representational capacity. In this work, we investigate this alignment bottleneck through the lens of mutual information, demonstrating that the constrained capacity of the language model inherently limits the Effective Mutual Information (EMI) between multimodal inputs and outputs, thereby compromising alignment quality. To address this challenge, we propose TinyAlign, a novel framework inspired by Retrieval-Augmented Generation, which strategically retrieves relevant context from a memory bank to enrich multimodal inputs and enhance their alignment. Extensive empirical evaluations reveal that TinyAlign significantly reduces training loss, accelerates convergence, and enhances task performance. Remarkably, it allows models to achieve baseline-level performance with only 40\% of the fine-tuning data, highlighting exceptional data efficiency. Our work thus offers a practical pathway for developing more capable lightweight VLMs while introducing a fresh theoretical lens to better understand and address alignment bottlenecks in constrained multimodal systems.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。