MoEP+: Enhancing Modular Expert Paths for Data-Efficient Language Modeling

Noor Eidirhn Mustafa, Valdemar Melkersson, Pierre Nugues


Abstract
We present MoEP+, an enhanced version of the Modular Expert Paths (MoEP) decoder. MoEP+ retains MoEP’s top-k routing over parallel expert paths, simplifies its dimensional mapping layers, and modernizes the backbone with RoPE, RMS normalization, and SwiGLU. It also introduces a projection-specialized routing scheme in which different experts apply nonlinear residual transformations to distinct attention projections. We evaluate MoEP+ on the BabyLM 2026 strict-small track, which is limited to 10 million words, and present it as a combined architectural recipe rather than attributing the observed gains to any single component.

Under a matched training budget, MoEP+ outperforms the unmodified MoEP baseline on all eight zero-shot benchmarks. On five of these benchmarks, the advantage persists even when the baseline is trained for five times as many epochs. Most notably, MoEP+ is substantially more data-efficient: after only two epochs, it surpasses the unmodified MoEP’s ten-epoch BLiMP score. Despite its smaller size, it also remains competitive with the official BabyLM 2026 baselines, with its advantages concentrated in the reading-based measures.
Anthology ID:
2026.babylm-main.39
Volume:
Proceedings of the Second BabyLM Workshop
Month:
October
Year:
2026
Address:
Budapest, Hungary
Editors:
Leshem Choshen, Ryan Cotterell, Mustafa Omer Gul, Jaap Jumelet, Tal Linzen, Aaron Mueller, Suchir Salhan, Raj Sanjay Shah, Alex Warstadt, Ethan Gotlieb Wilcox
Venues:
BabyLM | WS
SIG:
Publisher:
Association for Computational Linguistics
Note:
Pages:
504–513
Language:
URL:
https://aclanthology.org/2026.babylm-main.39/
DOI:
Bibkey:
Cite (ACL):
Noor Eidirhn Mustafa, Valdemar Melkersson, and Pierre Nugues. 2026. MoEP+: Enhancing Modular Expert Paths for Data-Efficient Language Modeling. In Proceedings of the Second BabyLM Workshop, pages 504–513, Budapest, Hungary. Association for Computational Linguistics.
Cite (Informal):
MoEP+: Enhancing Modular Expert Paths for Data-Efficient Language Modeling (Mustafa et al., BabyLM 2026)
Copy Citation:
PDF:
https://aclanthology.org/2026.babylm-main.39.pdf