arXiv:2409.11402cs.CLcs.AI2024-09被引 140

NVLM 1.0在多模态任务上媲美顶尖模型,还能提升纯文本性能。

NVLM: Open Frontier-Class Multimodal LLMs

  • 融合解码器与交叉注意力优势,设计新架构提升训练效率和推理能力。
  • 采用1D块标签技术处理高分辨率图像,显著提升OCR与多模态推理表现。
  • 强调数据质量与任务多样性,释放720亿参数模型权重供研究使用。

我们推出NVLM 1.0,一组前沿级多模态大语言模型,在视觉-语言任务中达到领先水平,性能可比肩主流闭源模型(如GPT-4o)及开源模型(如Llama 3-V 405B和InternVL 2)。令人意外的是,经过多模态训练后,其纯文本表现优于基础语言模型。在模型设计上,对比了仅解码器架构(如LLaVA)与交叉注意力架构(如Flamingo),结合两者优缺点提出新型结构,兼顾训练效率与多模态推理能力。引入1维块标签机制处理基于块的动态高分辨率图像,大幅增强多模态推理与文字识别任务表现。在数据方面,精心构建并公开预训练与监督微调数据集信息。研究表明,无论何种架构,数据质量与任务多样性的重要性超过规模。特别地,我们为NVLM-1.0构建生产级多模态能力,通过整合高质量纯文本数据与大量多模态数学推理数据,使模型在跨模态任务中表现出色,同时保持甚至提升原始语言模型的纯文本性能。模型权重已发布于HuggingFace:https://huggingface.co/nvidia/NVLM-D-72B,训练代码即将开源。

原文摘要 · Abstract (English)

We introduce NVLM 1.0, a family of frontier-class multimodal large language models (LLMs) that achieve state-of-the-art results on vision-language tasks, rivaling the leading proprietary models (e.g., GPT-4o) and open-access models (e.g., Llama 3-V 405B and InternVL 2). Remarkably, NVLM 1.0 shows improved text-only performance over its LLM backbone after multimodal training. In terms of model design, we perform a comprehensive comparison between decoder-only multimodal LLMs (e.g., LLaVA) and cross-attention-based models (e.g., Flamingo). Based on the strengths and weaknesses of both approaches, we propose a novel architecture that enhances both training efficiency and multimodal reasoning capabilities. Furthermore, we introduce a 1-D tile-tagging design for tile-based dynamic high-resolution images, which significantly boosts performance on multimodal reasoning and OCR-related tasks. Regarding training data, we meticulously curate and provide detailed information on our multimodal pretraining and supervised fine-tuning datasets. Our findings indicate that dataset quality and task diversity are more important than scale, even during the pretraining phase, across all architectures. Notably, we develop production-grade multimodality for the NVLM-1.0 models, enabling them to excel in vision-language tasks while maintaining and even improving text-only performance compared to their LLM backbones. To achieve this, we craft and integrate a high-quality text-only dataset into multimodal training, alongside a substantial amount of multimodal math and reasoning data, leading to enhanced math and coding capabilities across modalities. To advance research in the field, we release the model weights at https://huggingface.co/nvidia/NVLM-D-72B and will open-source the training code for the community soon.

多模态大模型视觉语言开源

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