mindspore-ai / mindspore-ai/hyper-parallel
Training Efficiency Challenges in Multimodal Large Language Models
Nobody has claimed this yet.
- Dominant language
- Python
- Stars
- 53
- Forks
- 63
- Avg merge
- 23h 45m
- Merged PRs (30d)
- 63
Description
Multimodal Large Language Models (MLLMs) are becoming an important workload for large-scale AI training. However, training MLLMs efficiently is more challenging than training decoder-only LLMs. In practice, even with more powerful hardware and larger training clusters, MLLM training can still suffer from lower MFU and higher engineering cost.
A key reason is that MLLMs are not just larger language models. They usually contain heterogeneous model components, such as modality encoders and LLM backbones. These components have very different computation patterns and memory footprints. As a result, a training strategy that works well for homogeneous LLM layers may not directly fit the full MLLM structure.
Another challenge comes from the input data itself. Compared with text-only training, multimodal training needs to process images, videos, and other modalities. These inputs can be much heavier and more irregular than text. For example, a high-resolution image may generate a large number of visual tokens, while a video may contain many such frames. At the same time, images may have different resolutions, and videos may have different lengths. This makes the workload vary significantly across samples and microbatches.
These two properties introduce new difficulties for pipeline-based training. Pipeline parallelism relies on relatively balanced and predictable stage execution. However, in MLLM training, heterogeneous model components and dynamic multimodal inputs can easily break this assumption, leading to pipeline imbalance, idle time, and reduced training efficiency.
This issue aims to highlight the need for better framework-level support for MLLM training workloads. In particular, the framework should consider the following challenges:
- Heterogeneous model components with different compute and memory requirements.
- Dynamic multimodal inputs with varying resolutions, token counts, and sequence lengths.
- Pipeline imbalance caused by both model-side heterogeneity and data-side workload variation.
Improving support for these challenges is important for making large-scale MLLM training more efficient, predictable, and easier to optimize in production environments.
schema_version: 1
source: gitcode
gitcode_repo: mindspore/hyper-parallel
gitcode_issue: 209
source_url: https://gitcode.com/mindspore/hyper-parallel/issues/209
Contributor guide
No contributing guide indexed for this repository
First steps
- Read the whole issue, then the project's contributing guide.
- Comment on the issue to say you are picking it up — it saves two people doing the same work.
- Fork the repository and make your change on a branch.
- Open a pull request that references the issue number.
Research direction
No files, tests, or entry points are named. Start by surveying the repository’s existing pipeline-parallel training and multimodal workload handling, then define a focused change covering the stated heterogeneity and dynamic-input challenges. Done should include framework support and evidence that pipeline imbalance and training efficiency improve.
Written by the indexing model from the issue text.
Assessment
- Tech stack
- python
- Domain
- ai, distributed-systems, machine-learning, performance
- Issue type
- Feature
- Difficulty
- 5/5
- Estimated time
- Over a week
- Activity status
- Active
- Clarity
- Needs clarification
- Newbie friendliness
- 25/100