Track papers across labs, topics, and trends.
Search papers, labs, and topics across Lattice.
This paper introduces MoEITS, a novel algorithm for simplifying Mixture-of-Experts Large Language Models (MoE-LLMs) using Information Theoretic frameworks. The method aims to reduce the computational burden, memory footprint, and energy consumption associated with MoE models. Experiments on Mixtral 8x7B, Qwen1.5-2.7B, and DeepSeek-V2-Lite demonstrate that MoEITS outperforms state-of-the-art MoE-LLM pruning methods in terms of both accuracy and computational efficiency.
MoEITS slashes the computational cost of MoE-LLMs while boosting accuracy, outperforming existing pruning methods on models like Mixtral and DeepSeek.
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.