atlas

Research map / Architectures and efficiency / Mixture of experts

Pushing Mixture of Experts to the Limit: Extremely Parameter Efficient MoE for Instruction Tuning

ICLR 2024 · Architectures and efficiency › Mixture of experts · approach: MoE LLM compression · 11 citations (OpenAlex)

Read the paperCheck your idea against 108,967 papers

AI-extracted summary. Written by AI from the paper's full text. Numbers and claims can be wrong or missing context, so check the paper before you cite or build on it.

Proposes Mixture of Vectors (MoV) and Mixture of LoRA (MoLORA), extremely parameter-efficient MoEs that use (IA)^3 vectors or LoRA adapters as experts with soft merging, for instruction tuning T5 models on P3 and evaluating zero-shot on unseen tasks.

Problem

Conventional MoEs require storing all experts and are costly to fully fine-tune; the paper asks whether MoE can be applied within parameter-efficient fine-tuning for instruction tuning and generalization to unseen tasks.

Approach

Replace each MoE expert with a lightweight PEFT adapter ((IA)^3 vector or LoRA), train router and experts from scratch on a frozen pretrained model, and use soft merging (weighted average of experts before applying the linear adapter) so that only one merged adapter is applied per block.

Key results

Measured effects reported in the paper

What was changedWhereMeasuredEffect
Replace single (IA)^3 with a 30-expert MoVT5-3B instruction tuning · P3 unseen tasksmedian accuracy (Average): 14.57%Helped
Replace LORA with 15-expert MoLORAT5-3B instruction tuning · P3 unseen tasksmedian accuracy (Average): 5.70%Helped
Increase MoV experts from 10 to 30 (ablation)MoV on T5-3B · P3 unseen tasksmedian accuracy (Average): 59.93 to 60.61Helped
Increase MoV experts from 30 to 60 (ablation)MoV on T5-3B · P3 unseen tasksmedian accuracy (Average): 60.61 to 59.83No clear effect
Increase MoV experts from 10 to 60 (ablation)MoV on T5-770M · P3 unseen tasksmedian accuracy (Average): 52.47 to 53.63Helped
Increase MoV experts from 10 to 60 (ablation)MoV on T5-11B · P3 unseen tasksmedian accuracy (Average): 62.3 to 64.08Helped
Increase MoLORA experts from 5 to 15 (ablation)MoLORA on T5-3B · P3 unseen tasksmedian accuracy (Average): 58.6, 59.77, and 60.79Helped
Use soft merging instead of discrete top-1 routing (ablation)MoV-10 on T5-3B · P3 unseen tasksmedian accuracy (Average): 54.92 -> 59.93Helped
Use soft merging instead of discrete top-2 routing (ablation)MoV-10 on T5-3B · P3 unseen tasksmedian accuracy (Average): 57.45 -> 59.93Helped
Add load-balancing loss to top-2 routing (ablation)MoV on T5-3B · P3 unseen tasksmedian accuracy (Average): 1.5 average median scoreHurt
Route using sentence embeddings instead of token embeddings (ablation)MoV-10 on T5-3B · P3 unseen tasksmedian accuracy (Average): 8.86%Hurt
Use batch size 2048 instead of the small-batch training setup (ablation)Parameter-efficient MoE · P3 unseen tasksperformanceHurt

Surprising findings

Limitations

Experimental setup

Datasets: P3

Benchmarks: ANLI CB RTE WSC WIC Copa WinoGrande HellaSwag

Metrics: median accuracy mean accuracy

Baselines: T0 (full fine-tuning) (IA)^3 LoRA Polytropon

Builds on: (IA)^3 LoRA Mixture of Experts T0 Polytropon

Closest papers by meaning

Writing a paper in this area? scime draft is a free LaTeX editor with AI included, real-time co-authors and a check of your references and venue rules before you submit. It also imports OpenAI Prism projects.

Open scime draft

Get the next scime Dispatch and a monthly field update

Free. We send a confirmation link first; every email has a one-click unsubscribe. Privacy