arXiv:2504.00595cs.CL2025-04被引 14

2B模型仅用220小时训练,高效开源多模态大模型。

Open-Qwen2VL: Compute-Efficient Pre-Training of Fully-Open Multimodal LLMs on Academic Resources

  • 用动态分辨率和多模态序列打包提升训练效率
  • 2900万图文对训练,仅需220小时A100算力
  • 全开源代码、数据与过滤方法,真正开放

复现前沿多模态大模型预训练面临全流程障碍,包括高质量数据筛选、多模态数据混合策略、序列打包技术及训练框架。我们提出Open-Qwen2VL,一个完全开源的20亿参数多模态大模型,仅用220个A100-40G GPU小时,在2900万图像-文本对上高效预训练。方法采用低到高动态图像分辨率与多模态序列打包,显著提升效率。数据集通过基于MLLM(如MLM-Filter)与传统CLIP的双重筛选,大幅提升质量。预训练在加州大学圣塔芭芭拉分校的8×A100-40G集群上完成,处理50亿打包多模态标记,仅为Qwen2-VL总1.4万亿标记的0.36%。最终指令微调后的Open-Qwen2VL在MMBench、SEEDBench、MMstar和MathVista等多个多模态基准测试中超越部分开源的SOTA模型Qwen2-VL-2B,证明其卓越训练效率。我们开源全部内容:计算高效与数据高效的训练细节、数据筛选方法、序列打包脚本、WebDataset格式预训练数据、基于FSDP的训练代码库,以及基础与指令微调模型检查点。我们将‘完全开源’重新定义为:1)训练代码库完整发布;2)详细数据筛选技术公开;3)所有预训练与监督微调数据可获取。

原文摘要 · Abstract (English)

The reproduction of state-of-the-art multimodal LLM pre-training faces barriers at every stage of the pipeline, including high-quality data filtering, multimodal data mixture strategies, sequence packing techniques, and training frameworks. We introduce Open-Qwen2VL, a fully open-source 2B-parameter Multimodal Large Language Model pre-trained efficiently on 29M image-text pairs using only 220 A100-40G GPU hours. Our approach employs low-to-high dynamic image resolution and multimodal sequence packing to significantly enhance pre-training efficiency. The training dataset was carefully curated using both MLLM-based filtering techniques (e.g., MLM-Filter) and conventional CLIP-based filtering methods, substantially improving data quality and training efficiency. The Open-Qwen2VL pre-training is conducted on academic level 8xA100-40G GPUs at UCSB on 5B packed multimodal tokens, which is 0.36% of 1.4T multimodal pre-training tokens of Qwen2-VL. The final instruction-tuned Open-Qwen2VL outperforms partially-open state-of-the-art MLLM Qwen2-VL-2B on various multimodal benchmarks of MMBench, SEEDBench, MMstar, and MathVista, indicating the remarkable training efficiency of Open-Qwen2VL. We open-source all aspects of our work, including compute-efficient and data-efficient training details, data filtering methods, sequence packing scripts, pre-training data in WebDataset format, FSDP-based training codebase, and both base and instruction-tuned model checkpoints. We redefine "fully open" for multimodal LLMs as the complete release of: 1) the training codebase, 2) detailed data filtering techniques, and 3) all pre-training and supervised fine-tuning data used to develop the model.

多模态开源高效训练大模型

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