Search papers, labs, and topics across Lattice.
This study investigates the application of a simplified reasoning technique, termed protoreasoning, in tiny transformer models with approximately 1 million parameters, enabling detailed analysis of step-by-step reasoning. By focusing on reasoning-friendly tasks involving Dyck languages, the authors reveal that protoreasoning significantly narrows the out-of-distribution generalization gap. The findings suggest that the content of reasoning traces, rather than just their presence, is crucial for enhancing model performance in reasoning tasks.
Tiny transformers can achieve impressive reasoning capabilities through protoreasoning, challenging assumptions about the necessity of larger models for effective step-by-step reasoning.
We show that tiny transformers can profitably employ a simple form of Chain of Thought, which we call protoreasoning, allowing us to study step-by-step reasoning on ~1M-parameter models and opening up opportunities for much more detailed experimentation and analysis than is feasible for larger models. Current Large Language Models exhibit impressive step-by-step reasoning, but we have yet to understand its generality, i.e., when and how LLMs learn genuinely general algorithms rather than"bags of heuristics."Such questions are hard to settle on compute-intensive frontier models trained on opaque data. To work at model scales far below the threshold for natural-language competence, we define reasoning-friendly tasks on Dyck languages (sentences of correctly nested brackets). We find that protoreasoning traces substantially close the out-of-distribution generalization gap, and ablations confirm that the trace's content, not merely its extra tokens, drives the gain.
