arXiv:2606.23087cs.LG2026-06

提出流式解耦训练框架,显著提升视觉语言模型工业级训练效率

FlowTrain: Flow-Based Decoupled Training for Industrial-Grade Vision-Language Models

论文配图:FlowTrain: Flow-Based Decoupled Training for Industrial-Grade Vision-Language Models
图 1 · 摘自论文原文
  • 将视觉语言模型训练重构为生产者-消费者数据流,通过统一内存池解耦执行
  • 实现超过50%的模型利用率和最高1.7倍吞吐提升,逼近纯语言模型训练效率
  • 适合大规模视觉语言模型训练团队,尤其关注分布式训练效能优化的研究者

工业级视觉语言模型(VLM)的分布式训练效率远低于单模态大语言模型。现有方案或采用整体化设计对异构模块分配统一并行策略,或采用解耦部署将模块分离并作为批处理同步流水线执行。本文指出上述方法仍不充分,提出进一步解耦的可能。为此,我们设计了FlowTrain——一种基于流的解耦训练框架,将VLM训练重构成通过统一内存池协调的生产者-消费者数据流。编码器与主干网络可在全局虚拟地址空间中独立推进。由于这种执行解耦从根本上改变了资源分配与调度的优化目标,FlowTrain引入异构并行分配器,通过求解吞吐匹配问题为各模块分配特定并行策略。动态打包调度器在运行时根据实际语言模型侧计算成本构建均衡微批次。真实工作负载的大量实验表明,FlowTrain实现了超过50%的模型利用率(MFU),最高达1.7倍吞吐提升,显著缩小了与仅语言模型训练的效率差距。

原文摘要 · Abstract (English)

Industrial-grade distributed training of vision-language models (VLMs) remains far less efficient than that of unimodal LLMs. Existing solutions either follow a monolithic design that assigns uniform parallelism to heterogeneous modules or adopt a disaggregated deployment that separates modules while executing them as a batch-synchronized pipeline. In this paper, we highlight that the above solutions are still not sufficient, and VLM training can be further decoupled. To this end, we present FlowTrain, a flow-based decoupled training framework that reformulates VLM training as a producer-consumer dataflow coordinated through a unified memory pool. The encoder and backbone can progress independently over a global virtual address space. Since this execution decoupling fundamentally changes the optimization objective of allocation and scheduling, FlowTrain further introduces a heterogeneous parallel allocator that assigns module-specific parallelism strategies by solving a throughput matching problem. The dynamic packing scheduler is used to construct balanced microbatches at runtime according to the actual LLM-side computation cost. Extensive experiments on real-world workloads show that FlowTrain achieves over 50% MFU and up to 1.7x throughput improvement, narrowing the efficiency gap to LLM-only training.

视觉语言模型分布式训练流式架构模型效率

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