arXiv:2412.16158cs.CV2024-12CVPR被引 9

让大模型直接理解图像,性能逼近主流模型。

HoVLE: Unleashing the Power of Monolithic Vision-Language Models with Holistic Vision-Language Embedding

  • 设计统一嵌入模块,让图文在同空间处理。
  • 多阶段训练提升图文对齐,准确率接近领先模型。
  • 无需微调语言模型,适合想简化架构的研究者。

大型语言模型(LLM)的快速发展推动了视觉-语言模型(VLM)的发展。单体式VLM避免了模态专用编码器,但性能通常较差。现有单体VLM大多需微调预训练语言模型以获得视觉能力,可能损害其语言性能。为此,本文提出高性能单体VLM HoVLE。我们发现,当图像嵌入与文本嵌入对齐时,LLM可有效理解图像。当前单体VLM的瓶颈在于缺乏统一的视觉-语言嵌入模块。HoVLE引入一个整体嵌入模块,将视觉和文本输入转换到共享空间,使LLM能像处理文本一样处理图像。此外,设计多阶段训练策略:先蒸馏预训练视觉编码器的视觉特征和LLM的文本嵌入,实现无配对图像与文本令牌的大规模训练;再在多模态数据上进行下一轮预测以对齐嵌入;最后加入指令微调。实验表明,HoVLE在多个基准测试中表现接近领先的组合式模型,远超以往单体模型。模型已开源:https://huggingface.co/OpenGVLab/HoVLE。

原文摘要 · Abstract (English)

The rapid advance of Large Language Models (LLMs) has catalyzed the development of Vision-Language Models (VLMs). Monolithic VLMs, which avoid modality-specific encoders, offer a promising alternative to the compositional ones but face the challenge of inferior performance. Most existing monolithic VLMs require tuning pre-trained LLMs to acquire vision abilities, which may degrade their language capabilities. To address this dilemma, this paper presents a novel high-performance monolithic VLM named HoVLE. We note that LLMs have been shown capable of interpreting images, when image embeddings are aligned with text embeddings. The challenge for current monolithic VLMs actually lies in the lack of a holistic embedding module for both vision and language inputs. Therefore, HoVLE introduces a holistic embedding module that converts visual and textual inputs into a shared space, allowing LLMs to process images in the same way as texts. Furthermore, a multi-stage training strategy is carefully designed to empower the holistic embedding module. It is first trained to distill visual features from a pre-trained vision encoder and text embeddings from the LLM, enabling large-scale training with unpaired random images and text tokens. The whole model further undergoes next-token prediction on multi-modal data to align the embeddings. Finally, an instruction-tuning stage is incorporated. Our experiments show that HoVLE achieves performance close to leading compositional models on various benchmarks, outperforming previous monolithic models by a large margin. Model available at https://huggingface.co/OpenGVLab/HoVLE.

视觉语言模型单体架构嵌入对齐多模态

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。