Training Language Models via Neural Cellular Automata
Dan Lee ⋅ Seungwook Han ⋅ Akarsh Kumar ⋅ Pulkit Agrawal
Abstract
Pre-training is crucial for large language models (LLMs), as it is when most representations and capabilities are acquired. However, natural language pre-training has problems: high-quality text is finite, contains human biases, and entangles knowledge with reasoning. This raises a fundamental question: is natural language the only path to intelligence? We propose using neural cellular automata (NCA) to generate synthetic, non-linguistic data for \textit{pre-pre-training} LLMs--training on synthetic-then-natural language. NCA data exhibits rich spatiotemporal structure and statistics resembling natural language while being controllable and cheap to generate at scale. We find that pre-pre-training on only 164M NCA tokens improves downstream language modeling by up to 6\% and accelerates convergence by up to 1.6$\times$ in 400M-3B models trained with Chinchilla optimal data budgets. \textit{Surprisingly, this even outperforms pre-pre-training on 1.6B tokens of high-quality natural language data with more compute.} Investigating what drives transfer, we find that attention layers are the most transferable, and that optimal NCA complexity varies by domain: code benefits from simpler dynamics, while math and web text favor more complex ones. These results enable systematic tuning of the synthetic distribution to target domains. More broadly, our work opens a path toward more efficient models with fully synthetic pre-training.
Chat is not available.
Successful Page Load