Memory or a Clock? Auditing a Memory Advantage in XLand-MiniGrid with Seeds, Maps and Evaluation Noise Apart
Organizations: Tecnológico de Monterrey
Abstract
On a partially observable XLand-MiniGrid task, an LSTM agent beats a memory-free MLP by 25 percentage points. The gap looks like a clear case of memory. We ask two questions about it. First, how much of a measured gap is noise? Separating training seed, map and evaluation randomness in 60 runs, each re-evaluated three times, we find that about ten paired seeds detect a 10-point gap in final success. The binary "task acquired" verdict, a common summary of learning curves, is 76% evaluation noise and would need about a hundred seeds for the same gap. Second, is the gap memory? On a sealed panel of new maps, opened once, a memory-free MLP of the LSTM's size comes within 3.3 points of the LSTM when it is given the time step, and gains no measurable benefit from its width without it. With the time step, an MLP a quarter of the LSTM's size, and the LSTM with its state erased at every step, also come within about 5 points. A linear probe reads the time step from the trained LSTM's state. On this task, most of the apparent memory advantage is an advantage of knowing the time, which matters because episodes are cut at a fixed limit and the reward shrinks with time. We recommend giving memory-free baselines the time step, and we release the protocol, the event-driven tooling and every checkpoint.
Figures & tables
| LSTM | MLP | LSTM MLP (paired) | |
|---|---|---|---|
| Trainable parameters | 11,367 | 2,919 | |
| Final test success (mean) | 84.0 [80.8, 86.9] | 58.9 [55.9, 61.7] | [20.8, 29.5] |
| Final test success (IQM) | 84.9 | 59.2 | |
| Final test success, re-evaluated | 83.9 [81.0, 86.7] | 58.7 [56.5, 61.0] | |
| Validation AUC | 44.6 [43.0, 46.1] | 38.8 [37.9, 39.7] | [4.0, 7.6] |
| Seeds acquiring B | 26 / 30 | 0 / 30 |
| Component | LSTM | MLP | LSTM MLP |
|---|---|---|---|
| 0.0059 [0.0030, 0.0087] | 0.0032 [0.0012, 0.0053] | 0.0096 [0.0043, 0.0151] | |
| 0.0346 [0.0275, 0.0469] | 0.0911 [0.0820, 0.1056] | 0.0368 [0.0332, 0.0486] | |
| 0.0174 [0.0104, 0.0210] | 0.0184 [0.0090, 0.0231] | 0.0362 [0.0191, 0.0461] | |
| 0.0778 [0.0661, 0.0906] | 0.1313 [0.1210, 0.1413] | 0.2111 [0.1918, 0.2332] | |
| Noise share of one evaluation | 16% | 37% | 25% |
| pp | pp | pp | pp interval | |||||
|---|---|---|---|---|---|---|---|---|
| Metric | ||||||||
| Final test success | 43 | 36 | 11 | 9 | 3 | 3 | 21 | 18 |
| Validation AUC | 9 | 8 | 3 | 2 | 2 | 2 | 5 | 4 |
| Acquired (yes/no) | 369 | 183 | 93 | 46 | 24 | 12 | 181 | 90 |
| Arm | Params | Clock | lr | Seeds | Sealed (%) | Acquired | AUC |
| LSTM | 11,367 | no | 20 | 95.0 [93.7, 96.3] | 20/20 (175k) | 0.60 | |
| MLP-wide + clock | 11,399 | yes | 20 | 91.7 [89.7, 93.5] | 20/20 (180k) | 0.62 | |
| stack4 + clock | 11,381 | yes | 10 | 74.2 [69.3, 78.5] | 5/10 (270k) | 0.38 | |
| LSTM-reset | 11,367 | no | 20 | 67.0 [65.1, 68.8] | 2/20 (240k) | 0.45 | |
| MLP-wide, no clock † | 11,401 | no | 10 | 65.1 [62.6, 67.6] | 0/10 | 0.44 | |
| MLP (small) | 2,919 | no | 10 | 61.9 [59.1, 64.6] | 0/10 | 0.46 |
| Contrast | Mean | BCa 95% | Holm | Ahead | Verdict | ||
|---|---|---|---|---|---|---|---|
| P1 | LSTM MLP-wide + clock | [0.9, 6.0] | 0.021 | 0.041 | 12/20 | not supported ∗ | |
| P2 | LSTM LSTM-reset | [25.3, 29.9] | 20/20 | supported | |||
| H3 | LSTM MLP-wide 2 stack4 | [32.5, 47.9] | 0.002 | 0.010 | 10/10 | supported | |
| E1 | clock: MLP-wide clock | [21.5, 30.5] | 0.002 | 10/10 | |||
| E2 | width: MLP-wide, no clock MLP | [ , 5.8] | 0.092 | 9/10 | |||
| E3 | LSTM MLP (small) | [31.6, 38.2] | 0.002 | 10/10 |
| Contrast | Mean | BCa 95% | Ahead | ||
|---|---|---|---|---|---|
| M1 | LSTM-reset + clock LSTM-reset | [21.6, 27.0] | 0.002 | 10/10 | |
| M2 | LSTM LSTM-reset + clock | [2.5, 6.9] | 0.002 | 10/10 | |
| M3 | MLP + clock MLP | [24.0, 31.6] | 0.002 | 10/10 | |
| M4 | MLP-wide + clock MLP + clock | [ , 5.1] | 0.66 | 6/10 | |
| M5 | LSTM MLP + clock | [2.6, 8.6] | 0.012 | 8/10 |
| Features | Trained state | Initial-weights state | Observation | Shuffled labels |
|---|---|---|---|---|
| LSTM | 0.84 | 0.52 | ||
| LSTM-reset | 0.04 |
Reinforcement Learning
Reinforcement learning studies how agents learn sequential decisions from rewards and interaction, balancing exploration, performance, safety, and adaptation.
See all 1,727 papers