DOT-MoE: converting dense models into sparse MoE via optimal transport

burkov · x · 2026-10-16

LLMs are expensive to run since every parameter fires for every token. Sparse MoE avoids this, but training from scratch is costly and unstable. Converting dense models into MoEs (MoEfication) is appealing, yet current methods rely on heuristics — random splits, weight similarity, or activation clustering — ignoring how neurons affect outputs. DOT-MoE reformulates neuron assignment as an optimal transport problem, shipping neurons to experts under strict capacity constraints (each expert gets exactly the same... [truncated]).

Original post →

More from Infra

Infra channel →