Share your thoughts, 1 month free Claude Pro on usSee more
WorkDL logo mark

Improving MLLM Training Efficiency via Stage-Aware Sparsity

About

Multimodal Large Language Models (MLLMs) have demonstrated outstanding performance across a variety of domains. However, training MLLMs is often inefficient, as much of the computation is redundant due to the long input sequences from multimodal data and underutilized inter-layer operations. Notably, such redundancy is not static but varies across different stages of training. Building on this observation, we shift the focus to the training process itself and propose a training-efficient framework based on sparse representations, termed the Sparse Training Scheme (STS). Instead of applying a uniform sparsity strategy, STS adopts a stage-aware design that adapts to different sources of redundancy during training. Specifically, the framework consists of two complementary components: the Visual Token Compressor, which reduces the information load by compressing visual tokens during modality alignment, and the Layer Dynamic Skipper, which mitigates computational overhead by dynamically skipping unnecessary layers during instruction tuning. Our approach is broadly applicable to diverse MLLM architectures and has been extensively evaluated on multiple benchmarks, demonstrating its effectiveness and efficiency.

Kean Shi, Liang Chen, Haozhe Zhao, Baobao Chang• 2025

Related benchmarks

TaskDatasetResultRank
Object Hallucination EvaluationPOPE
Accuracy87.6
2056
Visual Question AnsweringGQA
Accuracy59.7
218
Multimodal EvaluationMME
MME Score1.45e+3
179
Science Question AnsweringSQA
Accuracy (SQA)70.2
52
Showing 4 of 4 rows

Other info

Follow for update