The Surprising Expressivity of a Single Neuron with Autoregressive Chain-of-Thought
Organizations: Weizmann Institute of Science · Toyota Technological Institute at Chicago
Abstract
The success of autoregressive language models trained with Chain-of-Thought~(CoT) supervision raises a fundamental question: How simple can a next-token predictor be while still enabling the efficient simulation of general computation through its autoregressively generated CoT? Surprisingly, we show that a single neuron suffices. For a broad family of activations, including threshold and ReLU, a single autoregressive neuron can simulate any feedforward circuit with the same activation. Specifically, a size- circuit on inputs can be simulated by a neuron of dimension using generation steps, with both bounds independent of circuit depth and tight under natural assumptions. For threshold activations, this improves upon the simulation of Joshi et al. (COLT 2025), whose bounds scale exponentially with circuit depth. Consequently, autoregressive threshold neurons can simulate polynomial-time Turing machines, while remaining efficiently learnable from CoT supervision. Our main expressivity result relies on a novel connection to Sidon sets from additive combinatorics. As an experimental illustration, we efficiently train a single autoregressive ReLU neuron to mimic a neural network teacher with nontrivial test accuracy on CIFAR-10.
Figures & tables
| Model | Supervision | Test accuracy | Teacher agreement |
|---|---|---|---|
| CNN teacher | 45,000 images | ||
| AR student | 4,000 traces |
Appendix figures & tables2 assets
Supplementary material from the paper’s appendix.
Appendix
| Stage | Channels | Kernel | Stride | Activation Shape | #Parameters |
|---|---|---|---|---|---|
| 1 | |||||
| 2 | |||||
| 3 | |||||
| 4 |
| Metric | Official test set |
|---|---|
| Student accuracy | |
| Teacher accuracy | |
| Teacher agreement | |
| Free-logit RMSE | |
| Teacher-forced output RMSE |
Neural Network Expressivity
Neural Network Expressivity studies which functions and computations neural architectures can represent, and how efficiently.
See all 117 papers