arXiv:2503.23830cs.DCcs.AI2025-03被引 6

解决多模态大模型训练中的数据不平衡问题,提升训练效率。

OrchMLLM: Orchestrate Multimodal Data with Batch Post-Balancing to Accelerate Multimodal Large Language Model Training

  • 通过批处理后平衡调度,消除不同模态数据的分布差异。
  • 在2560块H100上训练840亿参数模型时,达到41.6%的模型浮点利用率。
  • 适合大规模多模态模型训练,显著提升吞吐量和可扩展性。

多模态大语言模型(MLLMs)如GPT-4o备受关注。在探索其训练过程中,我们发现模态组成不一致现象:不同样本中某一模态的比例波动剧烈,加剧了小批量不平衡问题,导致数据并行实例间GPU利用率不均,严重降低训练效率与可扩展性,影响训练速度并阻碍进一步研究。为此,我们提出OrchMLLM,一个全面的框架,以缓解由模态组成不一致引发的训练低效问题。首先,提出批量后平衡调度器,高效消除顺序数据中的小批量不平衡;其次,集成全局多模态协调器,统一调度多模态数据,应对模态组成不一致带来的挑战。我们在多种规模的MLLM上评估该框架,结果表明,在2560块H100 GPU上训练三模态840亿参数模型时,实现41.6%的模型浮点运算利用率(MFU),相比Megatron-LM最高提升3.1倍吞吐量。

原文摘要 · Abstract (English)

Multimodal large language models (MLLMs), such as GPT-4o, are garnering significant attention. During the exploration of MLLM training, we identified Modality Composition Incoherence, a phenomenon that the proportion of a certain modality varies dramatically across different examples. It exacerbates the challenges of addressing mini-batch imbalances, which lead to uneven GPU utilization between Data Parallel (DP) instances and severely degrades the efficiency and scalability of MLLM training, ultimately affecting training speed and hindering further research on MLLMs. To address these challenges, we introduce OrchMLLM, a comprehensive framework designed to mitigate the inefficiencies in MLLM training caused by Modality Composition Incoherence. First, we propose Batch Post-Balancing Dispatcher, a technique that efficiently eliminates mini-batch imbalances in sequential data. Additionally, we integrate MLLM Global Orchestrator into the training framework to orchestrate multimodal data and tackle the issues arising from Modality Composition Incoherence. We evaluate OrchMLLM across various MLLM sizes, demonstrating its efficiency and scalability. Experimental results reveal that OrchMLLM achieves a Model FLOPs Utilization (MFU) of $41.6\%$ when training an 84B MLLM with three modalities on $2560$ H100 GPUs, outperforming Megatron-LM by up to $3.1\times$ in throughput.

多模态训练加速大模型调度优化

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