DFLOP: A Data-driven Framework for Multimodal LLM Training Pipeline Optimization
Multimodal Large Language Models (MLLMs) have achieved remarkable advances by integrating text, image, and audio understanding within a unified architecture. However, existing distributed training frameworks remain fundamentally data-blind: they parallelize computation without accounting for variations in input data characteristics. This data unawareness leads to severe computation skew across stages and microbatches, where heterogeneous multimodal inputs incur different processing costs. Consequently, GPU resources are unevenly utilized, synchronization delays accumulate, and overall training efficiency degrades. To address this limitation, we present DFLOP, a d ata-driven f ramework for multimodal L LM training pipeline op timization. DFLOP continuously profiles runtime behavior to capture data-induced computation variance and employs predictive scheduling to balance workloads across stages and microbatches. By coupling data characteristics with execution planning, DFLOP substantially improves GPU utilization and throughput. Extensive experiments on large-scale multimodal benchmarks show that DFLOP achieves up to 3.6× faster training compared to state-of-the-art distributed training frameworks.
Authors
- Kwanghyun Park (ORCID: https://orcid.org/0000-0003-0757-2725)
- Jinkyu Jeong (ORCID: https://orcid.org/0000-0002-4905-9244)
- Rathijit Sen (ORCID: https://orcid.org/0000-0003-4736-2837)
- Hyeonsoo Lee (ORCID: https://orcid.org/0009-0004-3697-8470)
- Youngsok Kim (ORCID: https://orcid.org/0000-0002-1015-9969)
- Hyunjoon Kim (ORCID: https://orcid.org/0009-0008-9997-4169)
- Chaerim Lim
- Takki Yu
- Hyeonjun An
- Sihyun Kim (ORCID: https://orcid.org/0009-0005-3853-618X)
- Sangmin Jung (ORCID: https://orcid.org/0009-0008-3642-3634)
- Dongwook Kim (ORCID: https://orcid.org/0009-0006-9307-822X)
Institutions
- Microsoft (United States) (US)
- Yonsei University (KR)
Publication Details
- Journal
- Proceedings of the ACM on Management of Data
- Published
- 2026-05-18
- DOI
- https://doi.org/10.1145/3802037
- Primary Topic
- Multimodal Machine Learning Applications
- Type
- article
- Field-Weighted Citation Impact
- 0.00
Funders
- Yonsei University
- National Research Foundation of Korea