On an LSTM It Holds
Ongoing research. This is an experimental result from active work, not a settled conclusion. The numbers are what we measured and the method is described so you can judge it, but the programme is still running and later experiments may revise what it means.
A note on the language in these records. This is a working laboratory notebook for research into training AI models more cheaply and efficiently, so you will read that an approach did not work, that a result did not hold up, or that one method was worse than another. That is the research doing its job, not a verdict on the engineering we deliver to clients. Ruling an approach out is how the search narrows, and these are the pages that teach us the most: nearly every technique we now rely on came from understanding why something else fell short. Testing our own ideas at least as hard as anyone else's is the point of publishing them. More about this programme and why we run it.
Part of a bigger question: Is the task we are studying actually hard? – Often it is not. A rule from 1990 with no parameters beats the trained model on the task most of these results were measured on, and what an intervention costs is set by the task's own structure.
In plain English
What we asked. The easy-sum speed-up held on one kind of recurrent network and vanished on a transformer. Was it about recurrent networks in general, or just the one we used? We tried a second recurrent design, an LSTM.
What we found. It held, and more strongly: with the easy start, 15 of 16 runs learned the hard sum, typically in about 4,100 steps; ordinary training managed 5 of 16 in four times as long.
Why it matters. The benefit seems tied to how recurrent networks work: they have to build machinery to hold and combine past symbols, and easy examples help them build it. One caveat we are checking next: ordinary training's settings were not tuned for this model.
The rest of this page is the technical record: the design, every number, and the limits. It is written for a reviewer, and you do not need it to have understood the result above.
New here? How to read a research record
- Start at the verdict. Every record states, before the experiment was run, what result would have made us abandon the idea. That is the "kill test". Then it says whether the test fired. Nothing gets reinterpreted after the fact.
- Numbers in square brackets are uncertainty.
23.4 [18.1, 28.7]means our best estimate is 23.4 and the true value is probably somewhere in that range. If a range includes zero, we cannot claim an effect. - Results that rule an idea out are kept. Roughly half of what is published here says an approach did not work, including plenty of our own. Those pages are the output, not a shortfall: knowing which direction is a dead end is what lets the next experiment go somewhere better, and most of what we now rely on came out of understanding why something else fell short. Work that only publishes what worked is not measuring anything.
- Read the Limits section. Every record ends with what it does not show. It is the most honest part of any experiment and usually the shortest.
- Pro tip: the figures near the top are designed to carry the result on their own. If you read nothing else, read the caption under each one, which says what it shows and what to take from it.
EXPLORATORY. Not a preregistered study. Local CPU,32training runs of16000steps on sixteen fresh seeds. The design, a disclosed throwaway and the kill test were committed (a533e6f) before any run.
Program v2 Bucket A, item A58. Decisive computation: . Output: analysis/sum_mixture_lstm.py.analysis/sum_mixture_lstm.json
The question
The modular-sum speed-up holds on the GRU (A44, A53) and does not appear on a one-layer transformer (A55). Is it about recurrence, or about this GRU? The programme's other recurrent architecture is a tiny LSTM.
Design: the LSTM (width 48, one layer); the GRU's task and settings unchanged -- target (2, 7), rate 0.003 -- as a disclosed throwaway showed it inside the LSTM's reach (one seed solved at step 5250); 16000 steps, sixteen fresh seeds. Mixed (2000 steps of (2,3)/(2,4)/(2,5)) against plain with a 500-step learning-rate warm-up; steps to 0.9 (never = 16000). Kill test, fixed before execution: mixed minus plain includes zero or lies above it. Anchor, in code: with the GRU restored, the loop reproduces A40 -- held.
Results
| Arm (LSTM, sixteen fresh seeds) | Median steps to 0.9 | Solved by 16000 |
|---|---|---|
| easier sums mixed first | 4120 | 15 |
| plain, learning-rate warm-up | 16000 (budget) | 5 |
- Mixed minus plain:
-7259.4[-10905.0, -3613.7]steps; ratio of mean steps0.427.
The kill test does not fire. On the LSTM the mixture solves the task on fifteen of sixteen seeds at a median of 4120 steps, while the plain recipe solves five in the whole budget; the saving is larger than on the GRU and, with eleven plain seeds counted at the budget, understated. (The throwaway's plain seed solving at 5250 was one of the minority.)
What it says
The speed-up is about recurrence. It holds on both recurrent architectures on file (GRU at two widths, LSTM) and not on the transformer. That fits the reading from A49 and A52: a recurrent network must build machinery to hold and combine two past symbols, and the easy sums build it; a transformer can attend to both directly and does not need that help. The reading is still an interpretation; what is measured is the pattern across three architectures. The outside-venue draft's scope is updated to "recurrent networks".
What stands
- A58: kill test does not fire. LSTM: mixed minus plain
-7259.4[-10905.0, -3613.7]steps; ratio0.427; solved15against5. - Architecture pattern: GRU yes (two widths), LSTM yes (larger), one-layer transformer no.
Limits
- One LSTM configuration and rate (the GRU's); the plain LSTM was not separately tuned beyond the throwaway, so part of the gap may be the rate suiting the mixture better. Sixteen seeds.
Terms on this page
Every piece of vocabulary this record uses, in plain language. Generated from the text above, so it cannot drift out of step with it.
- architecture
- The blueprint of a model: what components it has and how they connect. Two models can be the same size and completely different architectures.
- GRU
- Gated Recurrent Unit. A compact design for processing sequences one item at a time, with internal switches controlling what it keeps in memory.
- kill test
- A condition written down before running the experiment that says what result would make us abandon the idea. Fixing it in advance is what stops a disappointing result being reinterpreted as an encouraging one.
- learning rate
- How big a step training takes each time it updates the model. Too small and nothing happens; too big and it never settles.
- LSTM
- Long Short-Term Memory. An older and larger relative of the GRU, also gated, also for sequences.
- recurrent
- A design that reads a sequence one item at a time, carrying memory forward. The main alternative is attention, which looks at everything at once.
- seed
- The number that fixes all the randomness in a training run. Same seed, same run. Running several seeds is how you tell a real effect from a lucky one.
- transformer
- The architecture behind most modern large language models. It uses attention to look at every part of the input at once, rather than reading in order.
- width
- How many internal numbers a model uses at each layer. The usual way we vary model size in these experiments.
Get new results as we publish them
Roughly monthly, one finding per email, in plain English first. Including the approaches that turned out not to work, which are usually the useful ones. No sales email.
Double opt-in: we send a confirmation link and add nobody who does not click it. One-click unsubscribe on every email.