summaryrefslogtreecommitdiff
path: root/docs/outreach/EPT_TECHNICAL_SUMMARY.tex
blob: 4ac89fc2bd8e7dbf17a4643800e90be5571a179a (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
\documentclass[10pt]{article}
\usepackage[letterpaper,margin=0.75in,top=0.65in,bottom=0.65in]{geometry}
\usepackage[T1]{fontenc}
\usepackage{newpxtext,newpxmath}
\usepackage{booktabs,array,enumitem,xcolor,titlesec}
\definecolor{accent}{RGB}{138,31,31}
\titleformat{\section}{\normalsize\bfseries\color{accent}}{}{0pt}{}
\titlespacing{\section}{0pt}{10pt}{4pt}
\setlist[enumerate]{leftmargin=1.2em,itemsep=2pt,topsep=2pt,parsep=0pt}
\setlist[itemize]{leftmargin=1.1em,itemsep=2pt,topsep=2pt,parsep=0pt}
\pagestyle{empty}
\setlength{\parindent}{0pt}
\setlength{\parskip}{3.5pt}
\begin{document}

{\large\bfseries Training language models without backpropagation}\\
{\small Technical summary \,\textperiodcentered\, Yuren Hao, University of Illinois Urbana-Champaign
\,\textperiodcentered\, August 2026 \,\textperiodcentered\, yurenh2@illinois.edu}

\section{What we do}
We train standard transformer language models from scratch with Equilibrium Propagation (EP), a
learning rule in which the training signal comes from the system relaxing twice and reading local
differences, with no backward pass anywhere. The trained model is an ordinary transformer and runs
ordinary forward inference. Every design choice is constrained to operations a physical substrate
can perform, because the point of the exercise is a learning rule that analog and other
physics based hardware can run natively.

\textbf{Why this question is open.} Backpropagation requires a global backward pass that a physical
system cannot perform on itself, so an analog accelerator must either digitize every intermediate
state or carry a second adjoint copy of the hardware. Both destroy the energy advantage that
motivates the substrate. Learning rules that a substrate can run natively do exist, and the
published evidence for them stops at small vision models. Whether they hold at language model scale
is the gap this project addresses.

\section{Results to date (all against matched backprop twins)}
Twin discipline: identical architecture, tokenizer, data order, optimizer, step budget, and
evaluation. The only difference is the training rule. Two seeds per arm at the smaller sizes, three
backprop seeds at 72M. Corpus is FineWeb-Edu with a 32k vocabulary.

\begin{center}
\begin{tabular}{@{}lccc@{}}
\toprule
\textbf{Model} & \textbf{EP (val CE)} & \textbf{Backprop twin} & \textbf{Perplexity difference}\\
\midrule
11M  & 4.194 & 4.197 & $-0.4\%$ (EP ahead)\\
18M  & 3.945 & 3.933 & $+1.2\%$\\
36M  & 3.649 & 3.648 & $+0.1\%$\\
72M  & rerun in progress & 3.288 & tracks the twins at matched progress\\
135M & rerun queued & 3.090 & \\
\bottomrule
\end{tabular}
\end{center}

Differences at the first three sizes are smaller than the spread between seeds, so EP and
backpropagation are statistically indistinguishable there. Wall clock cost of EP training in
simulation is about twice backpropagation. To our knowledge the 135M model is the largest language
model trained from scratch with no backpropagation at any layer.

\section{A result from this week, and why it matters for how we work}
Between 72M and 135M we had been measuring a quality loss that grew with model width, resisted every
algorithmic remedy we tried, and looked exactly like a scaling limit of the learning rule. A per
layer probe traced it instead to arithmetic in our own simulator. The training signal is a small
displacement added to a much larger activation; in single precision the components of that
displacement below machine epsilon are destroyed by the addition, and the readout, which recovered
the displacement by subtracting the activation back out, returned the damaged copy without any error
signal. Wider models put more layers below that threshold, which is why the loss looked like a
scaling wall.

Reading the training signal from the stored displacement directly, never as the difference of two
large numbers, removes the effect at no computational cost. Verification was pre registered: a
double precision control, a test that shifted the affected layers exactly as predicted when the
nudge amplitude was scaled by eight in each direction, a five width probe showing the damage appear
precisely where the displacement ratio crosses the precision threshold, complete closure of the
measured loss on a controlled instrument, and a null test confirming the fix changes nothing at the
sizes that were never below the threshold.

We report this here because it is representative of how the project is run and because it is likely
to matter to others. Any implementation of this family of learning rules that recovers the contrast
by differencing two large states will hit the same floor, and the symptoms mimic an algorithmic
limitation closely enough that it would be easy to publish the wrong conclusion.

\section{What compute would buy}
\begin{enumerate}
\item \textbf{Validation at 300M} on 6B tokens with a matched backprop control. Report within three
weeks of access. Roughly 250 H100 hours by measured throughput.
\item \textbf{A matched ladder from 150M to 600M}, two seeds per arm, with the hardware relevant
measurements (quantization, injected device noise, operating windows) run at every size rather than
as a separate track. Roughly 2,750 H100 hours.
\item \textbf{A 1B stage}, 2,000 to 3,000 H100 hours, conditioned on agreed loss and stability gates
from the first two.
\end{enumerate}

\section{What gets released}
Checkpoints at every size for both the EP models and their backprop twins, full training logs,
training and measurement code, the probe that produced the precision result above, and the negative
results. The intent is a reference artifact for the physical learning and neuromorphic communities:
the empirical answer to whether hardware compatible learning scales, in a form other groups can
build on without spending the compute again. Publication is open by default.

\section{People}
Yuren Hao (UIUC) leads the project, advised by ChengXiang Zhai on the language modeling side and by
Rainer Engelken on learning dynamics. Collaborators include Alexi Gladstone (UIUC, energy based
models), Xiang Wan (Stanford), and Zeyi Liu (UIUC). The hardware measurement program is being
designed in consultation with experimentalists in physical learning.

\end{document}