用扩散模型蒸馏视觉语言知识,低成本训练出顶尖图文描述模型
Vision-Language-Vision Auto-Encoder: Scalable Knowledge Distillation from Diffusion Models
- 通过冻结扩散模型解码器构建信息瓶颈,实现跨模态知识压缩
- 仅用单模态图像训练,生成质量媲美GPT-4o和Gemini 2.0 Flash的描述
- 总训练成本低于1000美元,大幅降低数据与算力需求
构建具备强大图文描述能力的先进视觉语言模型(VLM)通常需要在数十亿高质量图像-文本对上训练,耗时数百万GPU小时。本文提出视觉-语言-视觉(VLV)自编码框架,巧妙利用预训练组件:视觉编码器、文生图(T2I)扩散模型的解码器以及大型语言模型(LLM)。通过冻结T2I扩散模型解码器,在语言表征空间中建立信息瓶颈,实现知识蒸馏。该框架使用连续嵌入从条件扩散模型中提取知识,通过高质量重建展现全面语义理解。进一步微调预训练LLM以将中间语言表示解码为详细描述,构建出达到顶尖水平(SoTA)的图文描述器,性能可比肩GPT-4o和Gemini 2.0 Flash。方法具备极佳成本效益,主要使用单模态图像训练,充分利用现有预训练模型(图像编码器、T2I扩散模型、LLM),避免依赖大规模图像-文本配对数据集,总训练开销控制在1000美元以下。
原文摘要 · Abstract (English)
Building state-of-the-art Vision-Language Models (VLMs) with strong captioning capabilities typically necessitates training on billions of high-quality image-text pairs, requiring millions of GPU hours. This paper introduces the Vision-Language-Vision (VLV) auto-encoder framework, which strategically leverages key pretrained components: a vision encoder, the decoder of a Text-to-Image (T2I) diffusion model, and subsequently, a Large Language Model (LLM). Specifically, we establish an information bottleneck by regularizing the language representation space, achieved through freezing the pretrained T2I diffusion decoder. Our VLV pipeline effectively distills knowledge from the text-conditioned diffusion model using continuous embeddings, demonstrating comprehensive semantic understanding via high-quality reconstructions. Furthermore, by fine-tuning a pretrained LLM to decode the intermediate language representations into detailed descriptions, we construct a state-of-the-art (SoTA) captioner comparable to leading models like GPT-4o and Gemini 2.0 Flash. Our method demonstrates exceptional cost-efficiency and significantly reduces data requirements; by primarily utilizing single-modal images for training and maximizing the utility of existing pretrained models (image encoder, T2I diffusion model, and LLM), it circumvents the need for massive paired image-text datasets, keeping the total training expenditure under $1,000 USD.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。