arXiv:2504.14145cs.DCcs.AI2025-04被引 6

动态调度多模态训练,提升大模型效率

DIP: Efficient Large Multimodal Model Training with Dynamic Interleaved Pipeline

  • 按模态拆分计算流,均衡各阶段负载
  • 细粒度切分数据批次,适配动态输入
  • 运行时异步调度,无需停训即调优

大型多模态模型(LMMs)在理解与生成任务中表现出色,但其训练效率受两大问题影响:由异构架构引发的流水线阶段失衡,以及多模态数据多样性带来的训练数据动态性。本文提出DIP,一种动态、模态感知的流水线调度框架。通过两项关键技术解决动态失衡:(1) 将不同模态的计算分离至专用流水线段,平衡连续阶段内的工作负载;(2) 动态将输入数据细分为模态特异的子微批次,实现跨段负载均衡。系统在训练期间利用空闲CPU异步生成流水线调度,无需阻塞训练即可针对每批输入动态调整执行策略。我们在五种不同规模的LMM上验证DIP,涵盖12B至94B参数量,包含视觉-语言与扩散模型。实验表明,相比现有最优系统,DIP最高提升97.3%吞吐量,展现出对波动性多模态训练负载的强大适应能力。

原文摘要 · Abstract (English)

Large multimodal models (LMMs) have demonstrated excellent capabilities in both understanding and generation tasks with various modalities. While these models can accept flexible combinations of input data, their training efficiency suffers from two major issues: pipeline stage imbalance caused by heterogeneous model architectures, and training data dynamicity stemming from the diversity of multimodal data. In this paper, we present DIP, a dynamic and modality-aware pipeline scheduling framework designed for LMM training. DIP tackles the challenge of dynamic imbalance via two key techniques: (1) separating computations of different modalities into dedicated pipeline segments to balance workloads within a continuous set of stages; (2) dynamically splitting input data into finer-grained, modality-specific sub-microbatches to balance workloads across these segments. By asynchronously generating pipeline schedules on idle CPU resources during training, DIP dynamically tailors stage executions to each input batch without stalling the training process. We validate DIP on a diverse set of five LMMs, ranging from 12B to 94B parameters and including vision-language and diffusion models. Experimental results show that our system achieves up to 97.3% higher throughput compared to state-of-the-art systems, demonstrating strong adaptability to fluctuating multimodal training workloads.

多模态训练加速流水线调度

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