通过分阶段稀疏策略,提升多模态大模型训练效率
Improving MLLM Training Efficiency via Stage-Aware Sparsity
- 根据训练阶段动态调整稀疏策略,减少冗余计算
- 压缩视觉令牌与动态跳过层,降低计算开销
- 适用于多种架构,适合大规模多模态训练场景
多模态大语言模型(MLLM)在多个领域表现出色,但训练效率低下,主要源于多模态输入序列过长及层间操作利用不足。这种冗余并非静态,而是随训练阶段变化。针对此,我们提出基于稀疏表示的高效训练框架——稀疏训练方案(STS)。该方案不采用统一稀疏策略,而是设计分阶段自适应机制,包含两个互补组件:视觉令牌压缩器,在模态对齐阶段压缩视觉令牌以减轻信息负担;层动态跳过器,在指令微调阶段动态跳过冗余层以降低计算开销。本方法可广泛适配多种MLLM架构,并在多个基准上验证了其有效性与高效性。
原文摘要 · Abstract (English)
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.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。