atlas

Research map / Architectures and efficiency / Mixture of experts

MoDE: A Mixture-of-Experts Model with Mutual Distillation among the Experts

AAAI 2024 · Architectures and efficiency › Mixture of experts · approach: Expert routing and specialization · 14 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 MoDE, a mixture-of-experts training method where experts mutually distill (via L2 loss on expert outputs/embeddings) at moderate strength to counter "narrow vision," evaluated on tabular, translation, and image classification tasks.

Problem

Sparse or dense gated mixture-of-experts suffer from "narrow vision": each expert learns only from the limited samples routed to it, which limits generalization of the overall MoE.

Approach

Add a mutual distillation loss among experts (each expert distills from the average of the other experts' outputs or from peers via squared error on expert outputs) with a moderate strength so experts pick up features learned by others while keeping their specialization; also introduces "expert probing" to estimate per-expert performance on its dominating sub-task.

Key results

Measured effects reported in the paper

What was changedWhereMeasuredEffect
Add moderate mutual distillation among expertsDMoE and SMoE · Tabular benchmark datasetsaccuracy: MoE -> MoDE is higher on all seven datasets for both gate typesHelped
Add mutual distillation among expertsTransformer MoE · IWSLT14, IWSLT17, WMT14 translation datasetsBLEU: MoE -> MoDE: 34.88 -> 35.14; 28.66 -> 28.91; 29.81 -> 33.17; 14.46 -> 15.40; 29.10 -> 29.48Helped
Add mutual distillation among expertsCNN MoE · MNIST, Fashion-MNIST, CIFAR-10, CIFAR-100accuracy: MoE -> MoDE: 0.9886 -> 0.9918; 0.9057 -> 0.9083; 0.9444 -> 0.9519; 0.7545 -> 0.7824Helped
Increase expert count from 2 to 4 or 8 in MoDE (ablation)MoDE · Tabular benchmark datasetsaccuracy: MoDE accuracy is higher with 8 experts than 2 experts on all seven datasetsHelped
Increase distillation strength to α=100 (ablation)MoDE · Mfeat-karhunenoverall accuracy: 0.9625 -> 0.9450Hurt

Surprising findings

Limitations

Experimental setup

Datasets: OpenML tabular datasets (Isolet, Mfeat-karhunen, Mfeat-factors, Optdigits, Satimage, First-order-theorem-proving, Artifi IWSLT14 En-De IWSLT17 En-Ar WMT14 En-De MNIST Fashion-MNIST CIFAR-10 CIFAR-100

Benchmarks: OpenML tabular datasets IWSLT14 En-De IWSLT17 En-Ar WMT14 En-De MNIST Fashion-MNIST CIFAR-10 CIFAR-100

Metrics: top-1 accuracy BLEU

Baselines: Single model MoE (DMoE) MoE (SMoE)

Builds on: Mixture of Experts (Jacobs et al. 1991) Knowledge distillation (Hinton et al. 2015) Deep mutual learning Sparsely-gated MoE (Shazeer et al. 2017)

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