Our new X account is live! Follow @wizwand_team for updates
WorkDL logo mark

Parameter-Efficient Sparsity Crafting from Dense to Mixture-of-Experts for Instruction Tuning on General Tasks

About

Large language models (LLMs) have demonstrated considerable proficiency in general natural language processing (NLP) tasks. Instruction tuning, a successful paradigm, enhances the ability of LLMs to follow natural language instructions and exhibit robust generalization across general tasks. However, these models often encounter performance limitations across multiple tasks due to constrained model capacity. Expanding this capacity during the instruction tuning phase poses significant challenges. To address this issue, we introduce parameter-efficient sparsity crafting (PESC), which crafts dense models into sparse models using the mixture-of-experts (MoE) architecture. PESC integrates adapters into the MoE layers of sparse models, differentiating experts without altering the individual weights within these layers. This method significantly reduces computational costs and GPU memory requirements, facilitating model capacity expansion through a minimal parameter increase when guaranteeing the quality of approximation in function space compared to original sparse upcycling. Our empirical evaluation demonstrates the effectiveness of the PESC method. Using PESC during instruction tuning, our best sparse model outperforms other sparse and dense models and exhibits superior general capabilities compared to GPT-3.5. Our code is available at https://github.com/wuhy68/Parameter-Efficient-MoE.

Haoyuan Wu, Haisheng Zheng, Zhuolun He, Bei Yu• 2024

Related benchmarks

TaskDatasetResultRank
Commonsense ReasoningHellaSwag
Accuracy85.2
1460
Code GenerationHumanEval
Pass@148.8
850
Multi-task Language UnderstandingMMLU
Accuracy75.7
842
Mathematical ReasoningMATH
Accuracy24
162
Question AnsweringNatural Questions
EM31.2
52
General Language Understanding and ReasoningAcademic Benchmarks (test)
Average Score59.9
10
Showing 6 of 6 rows

Other info

Code

Follow for update