通过快慢双模式训练,让多模态大模型提速4倍且不丢性能。
Fast-Slow Efficient Training for Multimodal Large Language Models via Visual Token Pruning
- 快模式用视觉令牌裁剪加速训练,慢模式保留完整视觉输入
- 快慢模式协同:慢模式通过自蒸馏学习快模式知识
- 实测训练速度提升2.1到4.0倍,性能损失低于1%
多模态大语言模型(MLLMs)因模型规模庞大和视觉令牌数量多而面临严重的训练低效问题。现有方法主要聚焦于减小模型尺寸或可训练参数量。受视觉令牌裁剪(VTP)在提升推理效率方面成功启发,我们探索通过减少视觉令牌来实现高效训练的新方向。然而,直接在训练阶段应用VTP会导致训练-推理不一致:经裁剪训练的模型在未裁剪的完整视觉序列上表现不佳。为此,我们提出DualSpeed框架,包含快模式与慢模式。快模式作为主模式,集成现有VTP方法并引入模式隔离器以分离模型行为;慢模式作为辅助模式,在完整视觉序列上训练以保持训练-推理一致性,并通过自蒸馏从充分训练的快模式中学习。实验表明,DualSpeed使LLaVA-1.5训练速度提升2.1倍,LLaVA-NeXT提升4.0倍,性能保持在99%以上。代码已开源。
原文摘要 · Abstract (English)
Multimodal Large Language Models (MLLMs) suffer from severe training inefficiency issue, which is associated with their massive model sizes and visual token numbers. Existing efforts in efficient training focus on reducing model sizes or trainable parameters. Inspired by the success of Visual Token Pruning (VTP) in improving inference efficiency, we are exploring another substantial research direction for efficient training by reducing visual tokens. However, applying VTP at the training stage results in a training-inference mismatch: pruning-trained models perform poorly when inferring on non-pruned full visual token sequences. To close this gap, we propose DualSpeed, a fast-slow framework for efficient training of MLLMs. The fast-mode is the primary mode, which incorporates existing VTP methods as plugins to reduce visual tokens, along with a mode isolator to isolate the model's behaviors. The slow-mode is the auxiliary mode, where the model is trained on full visual sequences to retain training-inference consistency. To boost its training, it further leverages self-distillation to learn from the sufficiently trained fast-mode. Together, DualSpeed can achieve both training efficiency and non-degraded performance. Experiments show DualSpeed accelerates the training of LLaVA-1.5 by 2.1$\times$ and LLaVA-NeXT by 4.0$\times$, retaining over 99% performance. Code: https://github.com/dingkun-zhang/DualSpeed
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。