提出MoEITS算法,用信息论简化专家模型,降低大模型计算开销。
MoEITS: A Green AI approach for simplifying MoE-LLMs
- 基于信息论框架,设计轻量化专家模型简化方法
- 在Mixtral等模型上实现更高精度与更强压缩率
- 适合关注绿色AI和模型轻量化的研究者与工程师
大型语言模型正深刻改变学术与产业界。为提升性能,混合专家(MoE)架构成为主流,但带来高昂的训练与推理计算成本。本文提出一种名为MoEITS的新算法,通过标准化信息论框架实现对MoE-LLMs的高效简化。该方法从理论与实践双重角度深入分析,评估其计算复杂度、简化后模型的准确率及压缩率。实验在Mixtral $8\times7$B、Qwen1.5-2.7B与DeepSeek-V2-Lite上展开,对比当前最优的MoE剪枝方法。结果表明,MoEITS在所有基准测试中均表现更优,生成的模型兼具高精度与强计算效率。代码将开源于https://github.com/luisbalru/MoEITS。
原文摘要 · Abstract (English)
Large language models are transforming all areas of academia and industry, attracting the attention of researchers, professionals, and the general public. In the trek for more powerful architectures, Mixture-of-Experts, inspired by ensemble models, have emerged as one of the most effective ways to follow. However, this implies a high computational burden for both training and inference. To reduce the impact on computing and memory footprint as well as the energy consumption, simplification methods has arisen as very effective procedures. In this paper, an original algorithm, MoEITS, for MoE-LLMs simplification is presented. The algorithm is characterized by a refined simplicity, underpinned by standardized Information Theoretic frameworks. MoEITS is analyzed in depth from theoretical and practical points of view. Its computational complexity is studied. Its performance on the accuracy of the simplified LLMs and the reduction rate achieved is assessed through a thoroughly designed experimentation. This empirical evaluation includes a comparison with state-of-the-art MoE-LLM pruning methods applied on Mixtral $8\times7$B, Qwen1.5-2.7B, and DeepSeek-V2-Lite. The extensive experimentation conducted demonstrates that MoEITS outperforms state-of-the-art techniques by generating models that are both effective across all benchmarks and computationally efficient. The code implementing the method will be available at https://github.com/luisbalru/MoEITS.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。