arXiv:2506.14435cs.CVcs.LG2025-06被引 2

用三值专家替代高精度专家,大幅降低多模态大模型内存占用。

MoTE: Mixture of Ternary Experts for Memory-efficient Large Multimodal Models

  • 训练更多三值(-1,0,1)专家,共享预训练前馈网络提升效率
  • 在3.4GB专家内存下,比全精度模型高4.3%平均准确率
  • 适合边缘设备部署,兼容后训练量化,内存越紧优势越明显

大型多模态混合专家(MoE)模型能通过扩大参数量提升性能,同时保持活跃参数固定。但以往方法在稀疏上采样时主要使用全精度专家,导致大量专家带来更高内存开销,难以在边缘设备部署。本文提出MoTE,一种可扩展的内存高效训练方法,从密集检查点训练三值专家混合模型。不同于训练少量高精度专家,我们采用更多低精度专家。具体地,使用预训练前馈网络(FFN)作为共享专家,并训练参数为{-1,0,1}的三值路由专家。大量实验表明,该方法在模型规模增大时仍具良好扩展性。MoTE在性能上媲美全精度基线模型MoE-LLaVA,同时内存更小。此外,该方法兼容后训练量化,在内存约束更严时优势更显著:当专家内存限制为3.4GB且结合后训练量化时,MoTE在下游任务上平均准确率比MoE-LLaVA高出4.3%,验证了其在内存受限设备上的有效性与潜力。

原文摘要 · Abstract (English)

Large multimodal Mixture-of-Experts (MoEs) effectively scale the model size to boost performance while maintaining fixed active parameters. However, previous works primarily utilized full-precision experts during sparse up-cycling. Despite they show superior performance on end tasks, the large amount of experts introduces higher memory footprint, which poses significant challenges for the deployment on edge devices. In this work, we propose MoTE, a scalable and memory-efficient approach to train Mixture-of-Ternary-Experts models from dense checkpoint. Instead of training fewer high-precision experts, we propose to train more low-precision experts during up-cycling. Specifically, we use the pre-trained FFN as a shared expert and train ternary routed experts with parameters in {-1, 0, 1}. Extensive experiments show that our approach has promising scaling trend along model size. MoTE achieves comparable performance to full-precision baseline MoE-LLaVA while offering lower memory footprint. Furthermore, our approach is compatible with post-training quantization methods and the advantage further amplifies when memory-constraint goes lower. Given the same amount of expert memory footprint of 3.4GB and combined with post-training quantization, MoTE outperforms MoE-LLaVA by a gain of 4.3% average accuracy on end tasks, demonstrating its effectiveness and potential for memory-constrained devices.

多模态专家模型量化内存优化

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。