Add Trend column to KV cache retention sweep table
Browse files
README.md
CHANGED
|
@@ -107,26 +107,26 @@ rollout retains) *is* a load-time knob; the main table caps it to `Ctx` (1×).
|
|
| 107 |
Sweeping it to 0.5×, 1×, and 2× the window, with the same protocol (50 episodes,
|
| 108 |
seeds `0..49`, `max_steps=1000`):
|
| 109 |
|
| 110 |
-
| Env | `kv=16` (0.5×) | `kv=32` (1×) | `kv=64` (2×) |
|
| 111 |
-
|---|---:|---:|---:|
|
| 112 |
-
| `Ant-v5` | 5163.35±1608.02 | 5262.53±1400.17 | 4516.35±1773.93 |
|
| 113 |
-
| `HalfCheetah-v5` | 6793.17±2939.17 | 6816.48±3135.53 | 6468.21±3234.51 |
|
| 114 |
-
| `Hopper-v5` | 3361.71±103.69 | 2713.66±1075.57 | 992.92±445.63 |
|
| 115 |
-
| `Walker2d-v5` | 3950.09±459.01 | 3899.88±706.57 | 3842.19±718.60 |
|
| 116 |
-
| `Humanoid-v5` | 7431.52±2024.95 | 7892.65±1018.11 | 8040.41±38.02 |
|
| 117 |
|
| 118 |
The `kv=32` column matches the main table. At `kv=64` the rollout attends over
|
| 119 |
more history than the model's 32-token window — positions outside its native
|
| 120 |
range. This stays within the backbone's position capacity (Llama/RoPE,
|
| 121 |
`max_position_embeddings=256`), so it is an extrapolation regime, not a hard cap.
|
| 122 |
|
| 123 |
-
Best retention is **environment-dependent**
|
| 124 |
-
|
| 125 |
-
|
| 126 |
-
|
| 127 |
-
|
| 128 |
-
|
| 129 |
-
|
| 130 |
|
| 131 |
Evaluation runtime — every row above is measured on this one:
|
| 132 |
|
|
|
|
| 107 |
Sweeping it to 0.5×, 1×, and 2× the window, with the same protocol (50 episodes,
|
| 108 |
seeds `0..49`, `max_steps=1000`):
|
| 109 |
|
| 110 |
+
| Env | `kv=16` (0.5×) | `kv=32` (1×) | `kv=64` (2×) | Trend |
|
| 111 |
+
|---|---:|---:|---:|---|
|
| 112 |
+
| `Ant-v5` | 5163.35±1608.02 | 5262.53±1400.17 | 4516.35±1773.93 | ≈, -14% at 2× |
|
| 113 |
+
| `HalfCheetah-v5` | 6793.17±2939.17 | 6816.48±3135.53 | 6468.21±3234.51 | ≈ flat |
|
| 114 |
+
| `Hopper-v5` | 3361.71±103.69 | 2713.66±1075.57 | 992.92±445.63 | shorter ↑, 2× collapses |
|
| 115 |
+
| `Walker2d-v5` | 3950.09±459.01 | 3899.88±706.57 | 3842.19±718.60 | ≈ flat |
|
| 116 |
+
| `Humanoid-v5` | 7431.52±2024.95 | 7892.65±1018.11 | 8040.41±38.02 | longer ↑, steadiest |
|
| 117 |
|
| 118 |
The `kv=32` column matches the main table. At `kv=64` the rollout attends over
|
| 119 |
more history than the model's 32-token window — positions outside its native
|
| 120 |
range. This stays within the backbone's position capacity (Llama/RoPE,
|
| 121 |
`max_position_embeddings=256`), so it is an extrapolation regime, not a hard cap.
|
| 122 |
|
| 123 |
+
Best retention is **environment-dependent**, but for most envs the difference
|
| 124 |
+
across 0.5×/1×/2× is within run-to-run noise (`Trend` marks these `≈`). The real
|
| 125 |
+
exceptions: `Hopper-v5` clearly prefers a shorter window (0.5× is +24% with much
|
| 126 |
+
lower variance; 2× collapses to roughly a third of return, episodes ending early),
|
| 127 |
+
while `Humanoid-v5` is best at 2× (highest return and its steadiest — std 38
|
| 128 |
+
across all 50 episodes). The context window (1×) is a safe default; deviating from
|
| 129 |
+
it helps only in specific environments.
|
| 130 |
|
| 131 |
Evaluation runtime — every row above is measured on this one:
|
| 132 |
|