MEGA Hub

Mixture of Training: Recombining Small-Scale Scaffolded Pretraining Runs into a Larger Language Model

Authors

Do you know Mohammed Sabry?You can claim authorship or link another user.Do you know Sean Augenstein?You can claim authorship or link another user.Do you know Keith Rush?You can claim authorship or link another user.Do you know Lucio Dery?You can claim authorship or link another user.

Abstract

We ask whether language-model pre-training can be decomposed into smaller, independently trainable jobs that can later be recomposed into a coherent larger model. We introduce Mixture of Training (MoT), a scaffolded modular pre-training procedure that partitions a target Transformer into contiguous layer blocks, trains each block inside a frozen pretrained aligner scaffold, and then recomposes the trained blocks with an optional short end-to-end adaptation pass. On a 1.3B-parameter Gemma-style model trained on C4, MoT provides a small-scale proof of mechanism: independently trained depth slices can be recomposed into a usable language model, and a quality-parity schedule reaches the same reported perplexity as the monolithic baseline. This parity setting processes more aggregate tokens and has a shorter idealized layer-equivalent critical path after aligner preparation; its effective compute advantage depends on reusing the aligner across runs. We therefore present MoT not as a general replacement for monolithic pre-training, but as a small-scale framework for studying whether scaffolded sub-runs can act as reusable training units.

Community

00

Publication notes

Author note
Accepted at the Workshop on Methods and Opportunities at Small Scale (MOSS), COLM 2026