kissin42 commited on
Commit
7806154
·
verified ·
1 Parent(s): 3fe41a8

Add Trend column to KV cache retention sweep table

Browse files
Files changed (1) hide show
  1. README.md +14 -14
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**. `Hopper-v5` prefers a shorter window
124
- (0.5× is +24% with much lower variance; collapses to roughly a third of return,
125
- episodes ending early), while `Humanoid-v5` is best at (highest return and its
126
- steadiest std 38 across all 50 episodes). `Ant-v5` and `HalfCheetah-v5` peak at
127
- and lose ground by; `Walker2d-v5` is nearly flat. The context window (1×) is
128
- a safe default; retaining well past it can help or hurt depending on the
129
- environment.
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; collapses to roughly a third of return, episodes ending early),
127
+ while `Humanoid-v5` is best at(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