File size: 25,666 Bytes
83926c5 | 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 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126 127 128 129 130 131 132 133 134 135 136 137 138 139 140 141 142 143 144 145 146 147 148 149 150 151 152 153 154 155 156 157 158 159 160 161 162 163 164 165 166 167 168 169 170 171 172 173 174 175 176 177 178 179 180 181 182 183 184 185 186 187 188 189 190 191 192 193 194 195 196 197 198 199 200 201 202 203 204 205 206 207 208 209 210 211 212 213 214 215 216 217 218 219 220 221 222 223 224 225 | # RWKV7-G1j-1.5B ๋ฐ๋ณต ๋ถ๊ดด ์คํ
์คํ์ผ: 2026-09-27
## ์์ฝ
- Greedy baseline์์ ์ฆ๋ช
์ค ๋์ผํ ์ ํ์ด 5ํ ๋ฐ๋ณต๋๋ค. ์์ ํ ํฐ์ 241, 299, 357, 415, 473์ผ๋ก 58ํ ํฐ ๊ฐ๊ฒฉ์ด๋ค. ๋ณ๋๋ก 48ํ ํฐ ๊ตฌ์ ์ด 229โ287์์ ๋ฐ๋ณต๋๊ณ , 57ํ ํฐ ๊ตฌ์ ์ 284โ342์์ ๋ฐ๋ณต๋๋ค.
- โ๋ชจ๋ฅด๋ฉด ๋ชจ๋ฅด๊ฒ ์ต๋๋ค๋ผ๊ณ ๋ตํ๊ณ ๋ฉ์ถ๋ผโ๋ system ๋ฉ์์ง๋ง์ผ๋ก๋ ์ค๋จ๋์ง ์์๋ค. 512ํ ํฐ์ ๋ชจ๋ ์์ฑํ๊ณ , ๊ฑฐ์ ๋ฌธ๊ตฌ์ EOS๊ฐ ์์๋ค. ๊ฐ์ ์ํ ํ์ด๋ฅผ ๋ํ์ดํ๋ ๋์ ์ํ ๋
ผ์ฆ์ด ๋์๋ค.
- system ๋ฉ์์ง์ `repetition_penalty=1.1`์ ์ถ๊ฐํ๋ฉด ๋ฐ๋ณต๋์ ์ค์๋ค. rolling 64ํ ํฐ ์ฐฝ์ ์ค๋ณต 8-gram ํ๊ท ์ 0.063์์ 0.028๋ก, peak๋ 0.316์์ 0.175๋ก ๋ฎ์์ก๋ค.
- penalty๋ฅผ ์ ์ฉํด๋ 512ํ ํฐ๊น์ง ์์ฑํ๊ณ ์ํ ์ฆ๋ช
์ ์ฌ์ ํ ํ๋ ธ๋ค. ๋ฐ๋ณต ํจ๋ํฐ๋ ๋ฐ๋ณต์ ์ํํ์ง๋ง ์ข
๋ฃ๋ฅผ ๋ณด์ฅํ์ง ์์๋ค.
- ์ธ ์กฐ๊ฑด์์ recurrent state RMS๋ ๋น์ทํ ๋ฒ์์๋ค. ๋ฐ๋ณต์ state norm ํญ๋ฐ๊ณผ ํจ๊ป ๋ํ๋์ง ์์๋ค.
- ์์ด system ๋ฉ์์ง๋ฅผ ๋ฃ์ ์ถ๊ฐ ์คํ์์๋ ๋ ๋ชจ๋ธ ๋ชจ๋ 512ํ ํฐ๊น์ง ๊ฐ๊ณ ๊ฑฐ์ ํ์ง ์์๋ค. RWKV ๋ฐ๋ณต๋ฅ ์ ๋ฎ์์ก์ง๋ง OLMo๋ ํ๊ท ๋ฐ๋ณต๋ฅ ์ด ๊ฑฐ์ ๊ฐ๊ณ peak๋ ์ฝ๊ฐ ์ฌ๋๋ค.
- ์๋ต์ 2,048ํ ํฐ์ผ๋ก ๋๋ฆฌ๊ณ 256ํ ํฐ๋ง๋ค ์ฌ์ฉ๋ ์๋ฆผ์ ์ฃผ๋ harness๋ฅผ ์ ์ฉํ๋ค. RWKV๋ ๋์ผํ 256ํ ํฐ chunk๋ฅผ 8ํ ๋ฐ๋ณตํ๊ณ , OLMo๋ 184ํ ํฐ ๊ตฌ๊ฐ์ ๋ค์ชฝ์์ ๋ฐ๋ณตํ๋ค. ๋ ๋ค ๊ฑฐ์ ์ด๋ EOS ์์ด ๋๋ฌ๋ค.
## ๋ชจ๋ธ๊ณผ ๊ณตํต ์ค์
- ๋ชจ๋ธ: `RWKV/RWKV7-G1j-1.5B-20260831`
- ๊ณ์ธก์ ์ฌ์ฉํ ๋ก์ปฌ HF snapshot revision: `2c18b29ab7fbece25ff6112281eea0fa41fcb30f`
- ์คํ ํ๊ฒฝ: PyTorch 2.11.0+cu128, Transformers 5.17.0, RTX 3060, BF16
- ์
๋ ฅ์ ๋ชจ๋ธ chat template์ ์ฌ์ฉํ๊ณ `thinking=False`๋ก ๊ตฌ์ฑํ๋ค.
- ๊ณตํต ์์ฑ ์ค์ : greedy (`do_sample=False`), `max_new_tokens=512`, `eos_token_id=0`, stop string์ ๋ ๊ฐ์ ๊ฐํ ๋ค `User:`.
- system ์๋ prompt๋ 73ํ ํฐ, system ๋ฉ์์ง ํฌํจ prompt๋ 122ํ ํฐ์ด์๋ค.
- ๊ณ์ธก์ `model.generate()` ์์ฒด์ forward hook์ ์ฐ๊ฒฐํด ํ๋ค. ๊ฐ call์์ raw logits์ 24๊ฐ ์ธต์ WKV/attention-shift/FFN-shift state๋ฅผ ์ฝ์๋ค. baseline ์ฌ์์ฑ์ ์ด์ ํ ํฐ ๊ณ์ธก๊ณผ ํ ํฐ ID ์ ์ฒด๊ฐ ์ผ์นํ๋ค.
- ๋ณ๋ ์๋ token-by-token loop๋ ์ฒ์ 64ํ ํฐ ์ดํ ์์ฑ ๊ถค์ ์ด ๋ฌ๋ผ ์ต์ข
๊ฒฐ๊ณผ์์ ์ ์ธํ๋ค. ๊ฒฐ๊ณผ CSV๋ direct `generate()` ๊ฒฝ๋ก์์ ์ป์ ๊ฐ์ด๋ค.
### ํ๋กฌํํธ
์ฌ์ฉ์ ํ๋กฌํํธ:
> ์ ์ a,b๊ฐ ab+1 | a^2+b^2๋ฅผ ๋ง์กฑํ ๋, (a^2+b^2)/(ab+1)์ด ์์ ์ ๊ณฑ์์์ ์ฆ๋ช
ํด. ๋จ, a,b๋ ์์ ์ ์๋ค.
system ๋ฉ์์ง๊ฐ ์๋ ๋ ์กฐ๊ฑด:
> ๋ฌธ์ ๋ฅผ ํ ์ ์๊ฑฐ๋ ๋ต์ ๋ชจ๋ฅด๊ฒ ์ผ๋ฉด, ์ต์ง๋ก ๊ณ์ ์ถ์ธกํ์ง ๋ง๊ณ โ๋ชจ๋ฅด๊ฒ ์ต๋๋คโ๋ผ๊ณ ๋ตํ ๋ค ๋ฉ์ถฐ๋ผ.
### ์งํ ์ ์
- **Entropy / top-1 probability / margin**: ๋ค์ ํ ํฐ logits ๋ถํฌ์ ๋ถํ์ค๋, top-1 ํ๋ฅ , 1์์ 2์ logit ์ฐจ์ด. penalty run CSV์๋ ์์ `raw_*`์ repetition processor ์ ์ฉ ํ `penalty_*` ์ ์๊ฐ ๋ชจ๋ ์๋ค.
- **WKV state RMS**: ๊ฐ ์ธต์ `[heads, head_dim, head_dim]` recurrent matrix์์ `sqrt(mean(xยฒ))`๋ฅผ ๊ณ์ฐํ๋ค. shift state๋ ์ธต๋ณ hidden-size ๋ฒกํฐ์ RMS๋ค. `*_mean`๊ณผ `*_max`๋ 24๊ฐ ์ธต์ ๋ํ ์ง๊ณ๋ค.
- **State์ token ์ ๋ ฌ**: `predicts_token_step`์ ๋ค์ ํ ํฐ์ 1-based ๋ฒํธ๋ค. ํด๋น row์ recurrent state๋ ๊ทธ ํ ํฐ ์ง์ context, ์ฆ `context_generated_tokens`๊ฐ ์์ฑ ํ ํฐ๊น์ง ์ฒ๋ฆฌํ ์ํ๋ค.
- **rolling_dup8_rate_64**: ์ต๊ทผ ์ต๋ 64๊ฐ ์์ฑ ํ ํฐ์ 8-gram ์ค๋ณต ๋น์จ.
- **Cycle similarity**: baseline์์ ๋ฐ๋ณต ์์ ์ฒซ ํ ํฐ์ ์์ธกํ๋ step 241๊ณผ ์ดํ ๊ฐ์ ์์ ์์ step 299/357/415/473์ ๋น๊ตํ๋ค. CSV์ cosine์ ์ธต๋ณ state ๋ฐฉํฅ ์ ์ฌ๋ ํ๊ท /์ต์๊ฐ์ด๊ณ , `sym_kl_nats`๋ step241 logits ๋ถํฌ์ ๋น๊ตํ ๋์นญ KL์ด๋ค.
## ์ฌ์ prompt/sampler ํ์ธ
์๋๋ 512ํ ํฐ ๊ณ์ธก ์คํ ์ ์ ์งง์ screening์ด๋ค. ์งง์ ์ฅ๋ฌธ ์์ด ์ค๋ช
๊ณผ ํ๊ตญ์ด ์ด์ผ๊ธฐ๋ ๋ฐ๋ณต ๋ฃจํ ์ฌํ์ ์คํจํ๋ค. ์ด๋ ค์ด ์ ์๋ก prompt์ `thinking=True`๋ฅผ ์ฌ์ฉํ๋ฉด ๋ค decode ์ค์ ๋ชจ๋ ๋ฌธ์ ๋ฅผ ๋ํ์ดํ๊ฑฐ๋ ํ์ด๋ฅผ ๋ฏธ๋ฃจ๋ ํ์์ด ๋ณด์ฌ, ๋ณธ ๊ณ์ธก์์๋ `thinking=False`๋ก ๊ณ ์ ํ๋ค.
| Prompt | ์ค์ | ๊ธธ์ด | ๊ด์ฐฐ |
|---|---|---:|---|
| ๊ณต๊ณต ๋์๊ด์ ๊ณ์ ํฌ์ํด์ผ ํ๋ ์ด์ (์์ด) | greedy / T=1,p=.5 / T=.8,p=.95 / T=1,p=.95 | 153โ160 | ๋ค ์ถ๋ ฅ ๋ชจ๋ ๋๋ ทํ ๋ฐ๋ณต ๋ฃจํ ์์. duplicate 4-gram ๋น์จ 0โ.026. |
| ๋ก์ ์ฐ์ฐ ์ด์ผ๊ธฐ (ํ๊ตญ์ด) | ๊ฐ์ ๋ค ์ค์ | 160 | ๊ณ ์ ์ฃผ๊ธฐ ๋ฃจํ๋ ์์. T=1,p=.95๋ ๋ฌธ์ฅ์ด ํฌ๊ฒ ํํธ๋ฌ์ง. |
| ์ ์๋ก ์ฆ๋ช
(ํ๊ตญ์ด), `thinking=True` | ๊ฐ์ ๋ค ์ค์ | 220 | ๋ค ์ถ๋ ฅ ๋ชจ๋ ์ฆ๋ช
์ ์์ฑํ์ง ์๊ณ ๋ฌธ์ ๋ฅผ ๋ค์ ๋งํ๊ฑฐ๋ ์๊ธฐ ์ค๋ช
์ ๋ฐ๋ณตํจ. duplicate 4-gram ๋น์จ .088โ.198. |
์ด screening์ raw ์ถ๋ ฅ์ ๋ณ๋ ์ ์ฅํ์ง ์์๋ค. ์์ธํ ํ ํฐ๋ณ ์ง๋จ ๋ฐ ๋ณด๊ด๋ generation ํ
์คํธ๋ ์๋ ์ธ controlled run์ ๊ธฐ์ค์ผ๋ก ํ๋ค.
## ์กฐ๊ฑด๋ณ ๊ฒฐ๊ณผ
| ์กฐ๊ฑด | ์ถ๋ ฅ | ๋ฐ๋ณต ๊ด์ฐฐ | rolling 8-gram ํ๊ท / peak | ํ๊ท WKV / attention shift / FFN shift RMS |
|---|---:|---|---:|---:|
| Greedy baseline | 512ํ ํฐ, EOS ์์ | 39-token ์์ ํ 5ํ, 58ํ ํฐ ๊ฐ๊ฒฉ | 0.099 / 0.263 | 0.579 / 0.987 / 0.880 |
| System ๋ฉ์์ง | 512ํ ํฐ, EOS ์์, ๊ฑฐ์ ๋ฌธ๊ตฌ ์์ | 64-token ๊ตฌ์ ์ด 75โ354์์ ์ฌ๋ฑ์ฅ; ์ํ ๋
ผ์ฆ | 0.063 / 0.316 | 0.582 / 0.988 / 0.889 |
| System + repetition penalty 1.1 | 512ํ ํฐ, EOS ์์, ๊ฑฐ์ ๋ฌธ๊ตฌ ์์ | ๊ฐ์ฅ ๊ธด ๋์ผ n-gram 19ํ ํฐ (402โ449); ์ ํ ๊ณ ์ ๋ฐ๋ณต ์์ | 0.028 / 0.175 | 0.586 / 0.999 / 0.897 |
### Baseline์์ ๋ฐ๋ณต๋๋ ์์ logits์ state
| ํ ํฐ step | top-1 ํ๋ฅ | entropy (nats) | top-1โtop-2 margin | WKV RMS | attention shift RMS | FFN shift RMS |
|---:|---:|---:|---:|---:|---:|---:|
| 241 | 0.333 | 1.416 | 0.035 | 0.583 | 0.989 | 0.884 |
| 299 | 0.592 | 1.237 | 1.324 | 0.580 | 0.984 | 0.873 |
| 357 | 0.969 | 0.193 | 4.405 | 0.573 | 0.963 | 0.841 |
| 415 | 0.993 | 0.058 | 6.020 | 0.570 | 0.952 | 0.827 |
| 473 | 0.998 | 0.022 | 6.856 | 0.568 | 0.946 | 0.819 |
๊ฐ์ ์ ํ์ ์ฒซ token์์ ๋ง์ง๋ง ๋ฐ๋ณต๊น์ง:
- top-1 ํ๋ฅ : 0.333 โ 0.998, entropy: 1.416 โ 0.022 nats, margin: 0.035 โ 6.856.
- WKV RMS ํ๊ท : 0.583 โ 0.568; attention shift RMS: 0.989 โ 0.946; FFN shift RMS: 0.884 โ 0.819.
- WKV state cosine์ 24๊ฐ ์ธต ํ๊ท 0.972 (์ต์๊ฐ 0.931), ์๋ L2 ์ฐจ์ด๋ ํ๊ท 0.220์ด์๋ค. raw logit cosine์ 0.999์์ง๋ง ๋์นญ KL์ 2.361 nats๊น์ง ์ฆ๊ฐํ๋ค.
์ฆ ๋ฐ๋ณต ์์์ ํ ํฐ ์ ํ์ ์ ์ ๋ ํ์ ์ ์ผ๋ก ๋ณํ์ง๋ง, state norm์ ์ฝ๊ฐ ๊ฐ์ํ๋ค. State ๋ฐฉํฅ์ ์ ์ฌํ๊ฒ ์ ์ง๋๋ ๋งค ๋ฐ๋ณต์์ ์์ ํ ๊ฐ์ state๋ก ๋๋์์ค์ง๋ ์์๋ค. ์ด ๊ด์ธก์ norm ๋ฐ์ฐ๋ณด๋ค ์ถ๋ ฅ ๋ถํฌ์ ํ์ ๋ ์ฆ๊ฐ์ ๋ฐ๋ณต ์ ํ์ ๊ณ ์ฐฉ์ ๊ฐ๋ฆฌํจ๋ค.
### System prompt ์กฐ๊ฑด์ ์ค๋ฅ ์
๋ชจ๋ธ์ `k(ab+1) โก 0 (mod ab+1)`์์ `k โก 0 (mod ab+1)`๋ฅผ ๊ฒฐ๋ก ๋ด๋ ธ๋ค. ์ ์์ ๋ชจ๋ ์ ์ `k`์ ๋ํด ์ฑ๋ฆฝํ๋ฏ๋ก ๋ค ๊ฒฐ๋ก ์ ๋ฐ๋ผ์ค์ง ์๋๋ค. ๋ต์ ๋ชจ๋ฅด๊ฒ ์ผ๋ฉด ๋ฉ์ถ๋ผ๋ system ์ง์๋ ์คํ ์ข
๋ฃ๋ฅผ ์ ๋ฐํ์ง ์์๋ค.
### Repetition penalty ์กฐ๊ฑด์ ํด์
Penalty๋ ์์ฑ๋ ํ ํฐ์ ์ ์๋ฅผ ๋ฐ๊ฟ ๋์ผ ๊ตฌ์ ์ฌํ์ ์ค์์ง๋ง, EOS ํ ํฐ์ ๋ง๋ค๊ฑฐ๋ refusal์ ๊ฐ์ ํ์ง ์๋๋ค. ์ถ๋ ฅ์ ๋ค๋ฅธ ์๋ชป๋ ์ฆ๋ช
๊ฒฝ๋ก๋ก ์ด๋ํ๊ณ 512ํ ํฐ ํ๋๊น์ง ๊ณ์๋๋ค. ์ด ์กฐ๊ฑด์ CSV๋ raw logits์ penalty ์ ์ฉ ํ scores๋ฅผ ๋ณ๋๋ก ์ ์ฅํ๋ค.
## OLMo 2 1B ๋น๊ต
๋น๊ต ๋ชจ๋ธ์ [๊ณต์ OLMo 2 1B Instruct ์ฒดํฌํฌ์ธํธ](https://huggingface.co/allenai/OLMo-2-0425-1B-Instruct)์ธ `allenai/OLMo-2-0425-1B-Instruct`๋ค. revision `48d788eca847d4d7548f375ad03d3c9312f6139e`๋ฅผ ๊ณ ์ ํด ๋ค์ด๋ก๋ํ์ผ๋ฉฐ, `model.safetensors`๋ 2,969,854,224 bytes๋ค. ๋ชจ๋ธ ์นด๋์ ์ฃผ ์ธ์ด๊ฐ ์์ด๋ก ํ์๋ ๋ชจ๋ธ์ ํ๊ตญ์ด prompt๋ฅผ ๋ฃ์ ์คํ์ด๋ฏ๋ก, ์๋ ๋น๊ต๋ ์ด ํ๊ตญ์ด ๋ฌธ์ ์ ํ์ ํด์ ํด์ํด์ผ ํ๋ค.
๊ฐ์ ํ๊ตญ์ด ๋ฌธ์ ์ system ๋ฉ์์ง, greedy, `max_new_tokens=512`, BF16์ ์ฌ์ฉํ๋ค. OLMo ์
๋ ฅ์ ๋ชจ๋ธ์ chat template๋ก ๋ง๋ค์๊ณ , prompt ๊ธธ์ด๋ baseline 75ํ ํฐ, system ํฌํจ 138ํ ํฐ์ด์๋ค. OLMo EOS๋ tokenizer์ `<|endoftext|>` (`100257`)๋ก ์ค์ ํ๋ค. ๋ฐ๋ณต ํจ๋ํฐ ์กฐ๊ฑด์์๋ `repetition_penalty=1.1`์ ์ ์ฉํ๋ค.
| OLMo ์กฐ๊ฑด | ํ ํฐ | ์ต์ฅ ๋ฐ๋ณต n-gram (์์โ์ฌ๋ฑ์ฅ) | rolling 8-gram ํ๊ท / peak | raw logits ํ๊ท top-1 / entropy | EOS / ๊ฑฐ์ ๋ฌธ๊ตฌ |
|---|---:|---|---:|---:|---|
| Greedy baseline | 512 | 21 (33โ109) | 0.056 / 0.491 | 0.782 / 0.880 | ์์ / ์์ |
| System ๋ฉ์์ง | 512 | 40 (302โ344) | 0.102 / 0.386 | 0.850 / 0.666 | ์์ / ์์ |
| System + repetition penalty 1.1 | 512 | 23 (217โ310) | 0.024 / 0.246 | 0.806 / 0.791 | ์์ / ์์ |
์ธ ์ถ๋ ฅ ๋ชจ๋ 512ํ ํฐ ํ๋๊น์ง ์์ฑ๋๋ค. baseline์ ์ ์๋ผ๋ ์กฐ๊ฑด์ `a,b โ {โ1,1}`์ฒ๋ผ ์๋ชป ํด์ํ๊ณ ์์ ๋ฐ๋ณตํ๋ค. system ๋ฉ์์ง ์กฐ๊ฑด์ ์ฑ๋ฆฝํ์ง ์๋ ์ธ์๋ถํด๋ฅผ ์ ์ ๋ก ๊ฒฐ๋ก ์ ๋ฐ์ด๋ถ์๋ค. repetition penalty ์กฐ๊ฑด์ `๋ชจ๋ฅด๊ฒ ์ต๋๋ค`๋ก ๋ฉ์ถ์ง ์์๊ณ , ๊ฑฐ๋ญ์ ๊ณฑ ํ๊ธฐ๋ฅผ ๋ฐ๋ณตํ๋ ๋ ๋ค๋ฅธ ์๋ชป๋ ์ฆ๋ช
์ ๋ง๋ค์๋ค.
OLMo ์์์ system-only์ penalty ์กฐ๊ฑด์ ๋น๊ตํ๋ฉด ํ๊ท rolling ์ค๋ณต 8-gram ๋น์จ์ 0.102์์ 0.024๋ก ์ฝ 77% ๊ฐ์ํ๊ณ , peak๋ 0.386์์ 0.246์ผ๋ก ์ฝ 36% ๊ฐ์ํ๋ค. ์ต์ฅ ๋์ผ token span๋ 40์์ 23ํ ํฐ์ผ๋ก ์งง์์ก๋ค. ๋ฐ๋ณต์ ์ํ๋์ง๋ง ๋ต์ ์ ํ์ฑ๊ณผ ์ข
๋ฃ๋ ๊ฐ์ ๋์ง ์์๋ค. penalty ์กฐ๊ฑด์์ raw logits ํ๊ท top-1/entropy๋ 0.806/0.791์ด๊ณ , penalty ์ ์ฉ ํ๋ 0.774/1.048์ด์๋ค. ์ฆ penalty๊ฐ ํ ํฐ ๋ถํฌ๋ฅผ ๋ ํํํ๊ฒ ํ์ผ๋ ์์ฑ ์ข
๋ฃ๋ฅผ ์ ๋ํ์ง๋ ์์๋ค.
### ๋ ๋ชจ๋ธ์ ๊ด์ฐฐ
| ์กฐ๊ฑด | RWKV ํ๊ท / peak ์ค๋ณต๋ฅ | OLMo ํ๊ท / peak ์ค๋ณต๋ฅ | ์ต์ฅ ๋์ผ token span (RWKV / OLMo) |
|---|---:|---:|---:|
| Greedy baseline | 0.099 / 0.263 | 0.056 / 0.491 | 57 / 21 |
| System ๋ฉ์์ง | 0.063 / 0.316 | 0.102 / 0.386 | 64 / 40 |
| System + penalty 1.1 | 0.028 / 0.175 | 0.024 / 0.246 | 19 / 23 |
๊ฐ ๋ชจ๋ธ ๋ด์์ ๋ณด๋ฉด repetition penalty 1.1์ด ์ธ ์กฐ๊ฑด์ ํ๊ท ๋ฐ๋ณต๋ฅ ์ ๋ฎ์ท์ง๋ง, ๋ ๋ชจ๋ธ ๋ชจ๋ EOS ์์ด ํ๋๊น์ง ์ถ๋ ฅํ๋ค. baseline์ OLMo ํ๊ท ์ค๋ณต๋ฅ ์ RWKV๋ณด๋ค ๋ฎ์์ผ๋, ์งง์ ๊ตฌ๊ฐ peak๋ ๋ ๋์๋ค. ๋ฐ๋ผ์ ๋จ์ผ ๋ฐ๋ณต๋ฅ ์์น๋ง์ผ๋ก ์ด๋ ๋ชจ๋ธ์ด ๋ loop์ ์ทจ์ฝํ๋ค๊ณ ๊ฒฐ๋ก ๋ด๋ฆฌ๊ธฐ ์ด๋ ต๋ค.
์ฃผ์ํ ์ ์ ๋ ๋ชจ๋ธ์ tokenizer๊ฐ ๋ค๋ฅด๋ค๋ ๊ฒ์ด๋ค. `8-gram`๊ณผ ๋ฐ๋ณต span ๊ธธ์ด๋ ๋ชจ๋ธ token ๋จ์์ด๋ฏ๋ก ๋ชจ๋ธ ๊ฐ ์ ๋ ๊ธธ์ด๋ฅผ ์ง์ ๋น๊ตํ๋ฉด ์ ๋๋ค. ๋์ผ ๋ชจ๋ธ ๋ด ์ธ ์กฐ๊ฑด์ ๋ณํ๋ ๋น๊ตํ ์ ์์ง๋ง, ๋ชจ๋ธ ๊ฐ ๋น๊ต๋ ํ
์คํธ์ ์คยท๋ฌธ์ ๋จ์ ๋ถ์๊ณผ ํจ๊ป ๋ด์ผ ํ๋ค. ๋ํ OLMo 2๋ Transformer๋ผ RWKV์ WKV recurrent state๊ฐ ์๋ค. OLMo์์๋ raw/processed logits์ ์ถ๋ ฅ ๋ฐ๋ณต๋๋ฅผ ๊ธฐ๋กํ๊ณ , RWKV state RMS์ ๊ฐ์ ์์น๋ผ๊ณ ํด์ํ OLMo state norm์ ๋ง๋ค์ง ์์๋ค.
## ์์ด system ๋ฉ์์ง ์ถ๊ฐ ๋น๊ต
system ์ง์์ ์ธ์ด ํจ๊ณผ๋ฅผ ๋ณด๊ธฐ ์ํด ๊ฐ์ ์๋ฏธ์ ์์ด ๋ฌธ์ฅ์ ๋ ๋ชจ๋ธ์ ๋ฃ์ด ํ ๋ฒ์ฉ ์ถ๊ฐ ์คํํ๋ค. ํ๊ตญ์ด ์ํ prompt, greedy decoding, penalty ์์, 512ํ ํฐ ํ๋๋ ์ ์งํ๋ค.
> If you cannot solve the problem or do not know the answer, do not keep guessing. Say โI don't knowโ and stop.
| ๋ชจ๋ธ | system ์ธ์ด | ์
๋ ฅ prompt ํ ํฐ | ์ต์ฅ ๋ฐ๋ณต ๊ตฌ๊ฐ (step) | ์ค๋ณต 8-gram ํ๊ท / peak | raw top-1 ํ๊ท / entropy ํ๊ท | EOS / ๊ฑฐ์ ๋ฌธ๊ตฌ |
|---|---|---:|---|---:|---:|---|
| RWKV | ํ๊ตญ์ด | 122 | 64 (75โ354) | 0.063 / 0.316 | 0.881 / 0.419 | ์์ / ์์ |
| RWKV | ์์ด | 104 | 17 (383โ454) | 0.009 / 0.140 | 0.833 / 0.604 | ์์ / ์์ |
| OLMo 2 1B | ํ๊ตญ์ด | 138 | 40 (302โ344) | 0.102 / 0.386 | 0.850 / 0.666 | ์์ / ์์ |
| OLMo 2 1B | ์์ด | 108 | 34 (423โ462) | 0.098 / 0.404 | 0.898 / 0.401 | ์์ / ์์ |
RWKV์์๋ ์ด ํ ๋ฒ์ greedy ์คํ์์ ์์ด system ์กฐ๊ฑด์ ํ๊ท 8-gram ์ค๋ณต๋ฅ ์ด 0.063์์ 0.009๋ก, peak๊ฐ 0.316์์ 0.140์ผ๋ก ๋ฎ์์ก๋ค. ์ต์ฅ ๋ฐ๋ณต span๋ 64์์ 17ํ ํฐ์ผ๋ก ์ค์๋ค. ๊ทธ๋ฌ๋ ๋ชจ๋ธ์ `I don't know`๋ผ๊ณ ๋ตํ์ง ์์๊ณ , `b`๊ฐ 1์ ๋๋๋ค๊ณ ์๋ชป ๊ฒฐ๋ก ๋ธ ๋ค ์๋ชป๋ ์ฆ๋ช
์ ์ด์ด๊ฐ๋ค. raw logits์ top-1/entropy ํ๊ท ์ 0.881/0.419์์ 0.833/0.604๋ก ๋ฐ๋์๊ณ , ๋ง์ง๋ง top-1/entropy๋ 0.9999/0.0014์์ 0.9256/0.4040์ผ๋ก ๋ฎ์์ก๋ค. WKV RMS ํ๊ท ์ ํ๊ตญ์ด system์ 0.582์์ ์์ด system์ 0.585๋ก ๊ฑฐ์ ๊ฐ์๊ณ , ๋ง์ง๋ง RMS๋ 0.577์์ 0.590์ผ๋ก norm ํญ๋ฐ์ ์์๋ค.
OLMo์์๋ ํ๊ท ์ค๋ณต๋ฅ ์ด 0.102์์ 0.098๋ก ๊ฑฐ์ ๊ฐ์๊ณ , peak๋ 0.386์์ 0.404๋ก ์กฐ๊ธ ๋์์ก๋ค. ์ต์ฅ ๋ฐ๋ณต ๊ตฌ๊ฐ์ 40์์ 34ํ ํฐ์ผ๋ก ์งง์์ก์ง๋ง ๋ ๋ค์ธ step 423์์ ๋ค์ ๋ํ๋ฌ๋ค. ์์ด system ์กฐ๊ฑด์์๋ ํ๊ท raw entropy๊ฐ 0.666์์ 0.401๋ก ๋ฎ์์ก์ผ๋, ๋ง์ง๋ง ํ ํฐ ๋ถ๊ทผ์ top-1 0.638 / entropy 1.640์ผ๋ก ์คํ๋ ค ๋ถํ์คํ๋ค. ์ถ๋ ฅ์ ์๋ชป๋ ๋์ ์ ๊ฐ์ ๊ตฌ์ ๋ฐ๋ณต์ผ๋ก ๋๋ฌ๊ณ ๊ฑฐ์ ๋ฌธ๊ตฌ๋ ์์๋ค.
์ด๋ ๊ฐ ์ค์ ์ ํ ๋ฒ์ฉ greedy๋ก ์คํํ ๊ฒฐ๊ณผ๋ค. system์ ์๋ฏธ๋ ๋ง์ท์ง๋ง ์์ด ๋ฒ์ญ์ ํ ํฐ ๊ธธ์ด์ ๋ฌธ๊ตฌ๊ฐ ๋ฌ๋ผ์ง๋ฏ๋ก, ์ด ์ฐจ์ด๋ง์ผ๋ก system ์ธ์ด์ ์ผ๋ฐ์ ์ธ ํจ๊ณผ๋ผ๊ณ ๋จ์ ํ์ง ์๋๋ค. RWKV/OLMo์ tokenizer๋ ์๋ก ๋ค๋ฅด๋ค. ํ ํฐ ๋ฐ๋ณต ์์น๋ ๊ฐ ๋ชจ๋ธ ์์ ํ๊ตญ์ด/์์ด system ์กฐ๊ฑด ๋น๊ต์ ์ฃผ๋ก ์ฌ์ฉํ๋ค.
## 256ํ ํฐ๋ง๋ค ๋๋ ์ฌ์ฉ๋ ์๋ฆผ harness
์ด์ ์์ด system ๋ฉ์์ง์ ํ๊ตญ์ด ์ํ ๋ฌธ์ ๋ฅผ ๊ทธ๋๋ก ๋๊ณ , repetition penalty ์์ด ์๋ต ์์ฐ์ 2,048ํ ํฐ์ผ๋ก ๋๋ ธ๋ค. ๋งค 256๊ฐ์ ์์ฑ ์๋ต ํ ํฐ๋ง๋ค generation์ ์ค๋จํ๊ณ , ๋ค์ user turn์ ๋์ ํ ํฐ ์์ โ์ด์ ๋ด์ฉ์ ๋ฐ๋ณตํ์ง ๋ง๊ณ ์ด์ด์ ๋ตํ๋ผโ๋ ์๋ด๋ฅผ ๋ฃ์ด ์ฌ๊ฐํ๋ค. ๋ชจ๋ธ์ ํ ํฐ ์ฌ์ฉ๋ ์๋ฆผ์ 256, 512, โฆ, 1,792์์ 7ํ ๋ฐ์๋ค. controller ๋ก๊ทธ์๋ ๋ง์ง๋ง 2,048/2,048 checkpoint๋ ๊ธฐ๋กํ๋ค.
| ๋ชจ๋ธ | ์๋ต ์์ฑ๋ | chunk / ์๋ฆผ | ์๋ฆผ๋ฌธ ๋ด์ฉ ํ ํฐ | ๊ฐ chunk ์
๋ ฅ ๋ฌธ๋งฅ ํ ํฐ ์ | ๋์ ์ฌ์
๋ ฅ ๋ฌธ๋งฅ ํ ํฐ | EOS / ๊ฑฐ์ |
|---|---:|---:|---:|---|---:|---|
| RWKV | 2,048 | 8 / 7 | 525 | 104, 442, 780, 1,118, 1,456, 1,794, 2,132, 2,470 | 10,296 | ์์ / ์์ |
| OLMo 2 1B | 2,048 | 8 / 7 | 522 | 108, 450, 792, 1,134, 1,477, 1,819, 2,162, 2,505 | 10,447 | ์์ / ์์ |
์๋ต ์์ฑ๋์ ๋ชจ๋ธ์ด ๋ง๋ assistant token๋ง ์ผ๋ค. ์๋ฆผ๋ฌธ ๋ด์ฉ ํ ํฐ ์๋ harness ๋ฌธ์ฅ๋ง ํ ํฐํํ ๊ฐ์ด๊ณ chat-template์ role marker๋ ์ ์ธํ๋ค. ๊ฐ chunk ์
๋ ฅ ๋ฌธ๋งฅ ํ ํฐ ์์ ๊ทธ ํฉ๊ณ๋ system/user ๋ฉ์์ง, ์ด์ assistant ์๋ต, harness ๋ฉ์์ง๋ฅผ ๋งค ํธ์ถ๋ง๋ค ๋ค์ tokenizeํด ๋ชจ๋ธ์ ๋ฃ์ ๊ธธ์ด๋ค. ๋ฐ๋ผ์ ๋์ ์ฌ์
๋ ฅ ๋ฌธ๋งฅ ํ ํฐ์ ์๋ต ์์ฑ๋์ด๋ ์๋น์ค ์ฒญ๊ตฌ๋๊ณผ ๋ค๋ฅธ ๊ณ์ธก์น๋ค.
| ๋ชจ๋ธ | ์ต์ฅ ๋ฐ๋ณต span (step) | rolling 8-gram ํ๊ท / peak | raw top-1 ํ๊ท / entropy ํ๊ท | chunk ์ฒซ ํ ํฐ top-1 / entropy | chunk ๋ด๋ถ top-1 / entropy | state RMS |
|---|---|---:|---:|---:|---:|---|
| RWKV | 1,024 (1โ1,025) | 0.017 / 0.140 | 0.971 / 0.119 | 0.486 / 2.714 | 0.972 / 0.109 | WKV ํ๊ท 0.568, ๋ง์ง๋ง 0.558 |
| OLMo 2 1B | 184 (1,281โ1,793) | 0.037 / 0.404 | 0.854 / 0.514 | 0.391 / 2.763 | 0.856 / 0.505 | ํด๋น ์์ |
RWKV์ ์ฌ๋ assistant chunk๋ token ID ๋จ์๋ก ๋ชจ๋ ๊ฐ์๋ค. ์ฆ โ์ด์ ๋ฌธ์ฅ์ ๋ฐ๋ณตํ์ง ๋ง๋ผโ๋ ์๋ด ๋ค์๋ ๋งค๋ฒ ์ฒซ chunk๋ฅผ ๋ค์ ๋๊ณ , 1โ1,024 ํ ํฐ ๊ตฌ๊ฐ ์ ์ฒด๊ฐ 1,025โ2,048์์ ์ฌํ๋๋ค. ์ค๋จ ์งํ chunk ์ฒซ ํ ํฐ์์๋ top-1์ด ํ๊ท 0.486์ด๊ณ entropy๊ฐ 2.714์์ง๋ง, chunk ๋ด๋ถ์์๋ 0.972์ 0.109๋ก ๋ฐ๋์๋ค. WKV RMS ํ๊ท ์ 0.568, ๋ง์ง๋ง์ 0.558๋ก norm ๋ฐ์ฐ์ ๋ณด์ด์ง ์์๋ค.
OLMo๋ ๋งค chunk๋ง๋ค ๋ต์ ์กฐ๊ธ์ฉ ๋ฐ๊พธ์์ง๋ง, ์ํ์ ์ผ๋ก ํ๋ฆฐ ์ ๊ฐ๋ฅผ ์ด์ด๊ฐ๋ค. ๋ง์ง๋ง ๋ chunk ์์ ๋ถ๋ถ์์ โLet's revisit the equationโฆโ์ผ๋ก ์์ํ๋ 184-token ๊ตฌ๊ฐ์ ๋ฐ๋ณตํ๋ค. ๋ง์ง๋ง ํ ํฐ์ raw top-1 0.99998 / entropy 0.00023์ผ๋ก ๋งค์ฐ ํ์ ์ ์ด์์ผ๋ฉฐ, EOS๋ โI don't knowโ๋ ์ถ๋ ฅํ์ง ์์๋ค.
์ด harness๋ ์ด์ assistant chunk๋ฅผ ์๋ฃ๋ turn์ผ๋ก ๋ฃ๊ณ , usage update๋ฅผ ์๋ก์ด user turn์ผ๋ก ์ถ๊ฐํ ๋ค **์ ์ฒด ๋ํ ๋ฌธ๋งฅ์ ๋งค chunk๋ง๋ค ๋ค์ ์
๋ ฅ**ํ๋ค. ๋ฐ๋ผ์ cache/state๋ฅผ ๋ณด์กดํ ์ฑ ์ธ๋ถ์์ ์ ๊น ๋ฉ์ถ๋ passive streaming ์คํ์ด ์๋๋ผ, ํ ํฐ ์ฌ์ฉ๋ ์๋ด๋ฅผ ๋ฐ์ผ๋ฉฐ ๋ํ๋ฅผ ์ด์ด๊ฐ๋ ์๋๋ฆฌ์ค๋ค. ์๋ต ํ ํฐ ์์ ์ฌ์
๋ ฅ ๋ฌธ๋งฅ ์๋ฅผ ๋ถ๋ฆฌํด ๊ธฐ๋กํ๋ค.
## ์์ด ๊ฒฉ์ ๋ฌธ์ sampled pass@10
์ฌ์ฉ์๊ฐ ์ ์ํ ์์ด 2ร3 ๊ฒฉ์ ๋ฌธ์ ๋ฅผ ๋ค ๋ชจ๋ธ์ ๊ฐ๊ฐ 10๊ฐ์ ๋
๋ฆฝ ์ํ๋ก ์
๋ ฅํ๋ค. ๊ฐ ์
๋ ฅ์ ๋ชจ๋ธ๋ณ chat template๋ฅผ ๊ฑฐ์น ๋จ์ผ user ๋ฉ์์ง์ด๊ณ system ๋ฉ์์ง๋ ์ถ๊ฐํ์ง ์์๋ค. Qwen3.5๋ Base ์ฒดํฌํฌ์ธํธ์ด๋ฉฐ, ๋ํ ์
๋ ฅ ํ์์ ๋ชจ๋ธ tokenizer์ template๋ฅผ ์ฌ์ฉํ๋ค. ์ํ๋ง ์ค์ ์ do_sample=true, temperature=1.0, top_p=0.95, top_k ์์, repetition penalty 1.0, seed 260928โ260937, ์ต๋ ์ ํ ํฐ 4096์ด๋ค. ํ๋ณธ์ ์์ฐจ ์คํํ๋ค.
RWKV ์ฒดํฌํฌ์ธํธ๋ ๊ฐ๊ฐ RWKV 1.5B revision 2c18b29ab7fbece25ff6112281eea0fa41fcb30f์ RWKV 2.9B revision c6202c9e1b5de80640fa487b112da621067967ad๋ฅผ ์ฌ์ฉํ๋ค. OLMo 2 1B Instruct๋ revision 48d788eca847d4d7548f375ad03d3c9312f6139e, [Qwen/Qwen3.5-2B-Base](https://huggingface.co/Qwen/Qwen3.5-2B-Base)๋ revision b1485b2fa6dfa1287294f269f5fb618e03d52d7c๋ฅผ ์ฌ์ฉํ๋ค. ๊ฐ sample ๋๋ ํ ๋ฆฌ์ ์ค์ prompt, ๋ชจ๋ธ revision, seed, ์ค์ , ์์ฑ transcript์ ํ ํฐ๋ณ ์งํ๊ฐ ์๋ค.
๋ฌธ์ ์ ์ ๋ต์ **45**๋ค. ๊ฒฉ์๋ฅผ a b c / d e f๋ก ์ฐ๋ฉด ๊ฐ๋กํฉ์ 100(a+d)+10(b+e)+(c+f)=999๋ค. ๊ฐ ๊ดํธ์ ํฉ์ 0โ18์ด๋ฏ๋ก ๊ฐ ์๋ฆฌ์์ 9์ฌ์ผ ํ๋ค. ์ ํ์ ์ซ์ ํฉ์ S=a+b+c๋ผ ํ๋ฉด ์๋ ํ ํฉ์ 27โS์ด๊ณ , ์ธ๋กํฉ ์กฐ๊ฑด์ 10S+(27โS)=99์ด๋ฏ๋ก S=8์ด๋ค. ํฉ์ด 8์ธ ์ธ ์๋ฆฌ ์ซ์ ์กฐํฉ์ C(10,2)=45๊ฐ์ด๋ฉฐ, ๊ฐ ์ ํ์ ์๋ ํ์ ์นธ๋ณ๋ก 9์์ ๋บ ๊ฐ์ผ๋ก ์ ์ผํ๊ฒ ์ ํ๋ค. ์์์ 0์ฒ๋ผ ์ ํ 0๋ ํ์ฉํ๋ค.
| ๋ชจ๋ธ | ์ ๋ต ์ํ | ๊ฒฝํ์ pass@1 | ๊ด์ธก pass@10 | ํ๊ท ์์ฑ ํ ํฐ | ์ข
๋ฃ: EOS / loop / token cap |
|---|---:|---:|---:|---:|---:|
| RWKV 1.5B | 0/10 | 0.0 | 0 | 368.2 | 10 / 0 / 0 |
| RWKV 2.9B | 0/10 | 0.0 | 0 | 644.1 | 10 / 0 / 0 |
| OLMo 2 1B Instruct | 0/10 | 0.0 | 0 | 804.3 | 10 / 0 / 0 |
| Qwen3.5 2B Base | 0/10 | 0.0 | 0 | 4096.0 | 0 / 0 / 10 |
| ๋ชจ๋ธ | ํ๊ท raw entropy (nats) | ํ๊ท WKV RMS |
|---|---:|---:|
| RWKV 1.5B | 0.9860 | 0.5716 |
| RWKV 2.9B | 0.6245 | 0.3494 |
| OLMo 2 1B Instruct | 0.4801 | ํด๋น ์์ |
| Qwen3.5 2B Base | 0.6925 | ํด๋น ์์ |
Entropy๋ ๊ฐ ํ๋ณธ ์์์ ํ ํฐ๋ณ ํ๊ท ์ ๋ธ ๋ค ๋ค์ ํ๋ณธ ํ๊ท ์ ์ทจํ๋ค. Tokenizer์ ๋ชจ๋ธ ๊ตฌ์กฐ๊ฐ ๋ค๋ฅด๋ฏ๋ก ๋ชจ๋ธ ๊ฐ entropy ์ ๋๊ฐ์ ์ง์ ์ ์ธ ํ์ง ์์๊ฐ ์๋๋ค. WKV RMS๋ RWKV recurrent state์์๋ง ์ ์๋๋ค. ํ ๋ฌธ์ ์์ n=k=10์ด๋ฏ๋ก ๊ด์ธก pass@10์ ํ๋๋ผ๋ ๋ง์๋์ง๋ฅผ ๋ํ๋ด๋ 0/1์ด๋ฉฐ, ๋ค ๋ชจ๋ธ ๋ชจ๋ 0/10์ด๋ค.
RWKV 1.5B, RWKV 2.9B, OLMo 2์ ํ์ฌ 30๊ฐ transcript์ token_metrics.csv๋ ๋ณด๊ด๋ ์ฒซ ๋ฐฐ์น์ ๋์ ํ์ผ๊ณผ ๋ชจ๋ SHA-256์ด ์ผ์นํ๋ค. ์ฒ์์๋ ๋ณด๊ด ๋ฐฐ์น๋ฅผ ์ด์ค ์ด์ค์ผ์ดํ ํ๋กฌํํธ๋ก ์๋ชป ํ์ ํ์ผ๋, ์ค์ ์ ์ฅ๋ prompt์ ๊ฒฐ๊ณผ๊ฐ ๊ฐ๋ค๋ ํ์ธ ๋ค ๊ทธ ํ์ ์ ์ทจ์ํ๋ค. ๋ณด๊ด ํด๋ ์ด๋ฆ์ archived_first_pass_duplicate_outputs๋ก ์ ์ ํ์ผ๋ฉฐ, ์ค๋ณต ํ๋ณธ์ผ๋ก ์ง๊ณํ์ง ์์๋ค. ์ฒซ ๋ฐฐ์น๋ฅผ ์๋ชป๋ ์
๋ ฅ์ผ๋ก ์ ์ธํ๋ ์ผ์ ์๋ค.
Qwen์ 10๊ฐ ํ๋ณธ ๋ชจ๋ EOS๋ loop stop ๋์ 4096-token ์์ฑ ์ํ์ ๋๋ฌํ๋ค. ํ๊ท ์คํ ์๊ฐ์ ํ๋ณธ๋น ์ฝ 215์ด์๋ค. transcript๋ ๊ฒฉ์ ํด์์ ๋ํ์ดํด ๊ฒํ ํ์ง๋ง, ์ด๋ ํ๋ณธ๋ 45๋ฅผ ๋ต์ผ๋ก ์ ์ํ์ง ์์๋ค. ๊ณต์ ๋ชจ๋ธ ์ค์ ์ text context๋ 262,144 ํ ํฐ์ด๋ฏ๋ก ์ด ์คํ์์ ์๋ฆฐ ์์ธ์ context cap์ด ์๋๋ผ ์์ฒญํ ์ ํ ํฐ ์ํ์ด๋ค. ์ ํ์ causal-conv1d ๋ฐ flash-linear-attention ์ปค๋์ด ์ค์น๋์ง ์์ ์ฐธ์กฐ PyTorch ๊ฒฝ๋ก๋ฅผ ์ฌ์ฉํ๋ค. ์คํ์ ์ ์ ์๋ฃ๋์ผ๋ฉฐ, ์ด ๊ฒฝ๋ก๊ฐ Qwen ์ํ ์์ฑ ์๊ฐ์ ๋๋ ธ๋ค.
๋ฐ๋ณต ๊ฐ์ง๊ธฐ๋ ๋งค 4ํ ํฐ๋ง๋ค ํ์ธํ๋ค. prompt์ ์ด๋ฏธ ์๋ ๊ตฌ์ ์ ์ ์ธํ๊ณ 96/64-token span์ด ํ ๋ฒ ๋ ๋์ค๊ฑฐ๋, 48/32-token span์ด ์ธ ๋ฒ ๋ํ๋๊ฑฐ๋, ๊ธธ์ด 8โ47์ ๋์ผ ์ฃผ๊ธฐ๊ฐ ์ธ ๋ฒ ์ฐ์ ๋ํ๋๊ฑฐ๋, ์ต๊ทผ 64ํ ํฐ์ ์ค๋ณต 8-gram ๋น์จ์ด 0.55 ์ด์์ธ ์ฒดํฌํฌ์ธํธ๊ฐ ์ฐ์ ๋ ๋ฒ์ด๋ฉด ํด๋น ํ๋ณธ์ ๋ฉ์ถ๋ค. ์ด ๊ธฐ์ค์์ **40๊ฐ ํ๋ณธ ์ค loop ์๋ ์ข
๋ฃ๋ 0๊ฑด**์ด๋ค. RWKV/OLMo 30๊ฐ๋ EOS, Qwen 10๊ฐ๋ token cap์ผ๋ก ๋๋ฌ๋ค. ์ด ๊ฒฐ๊ณผ๋ ์ ํ ์๊ณ๊ฐ์ ํ์ง ๊ฒฐ๊ณผ์ด๋ฉฐ, ๋ฐ๋ณต์ ์ธ ๋ฌธ์ฅ ์ ๊ฐ๊ฐ ์์๋ค๋ ๋ป์ ์๋๋ค.
์ด๊ธฐ looser detector๋ RWKV 2.9B sample 2์ ์์ ๊ฒฉ์ ๋ฌธ๊ตฌ์ OLMo sample 4์ ์ฝ๋ ๋ฌธ๊ตฌ๋ฅผ loop๋ก ์ค์ธํ๋ค. ๋ span์ ๊ฐ๊ฐ 1608โ2065, 456โ837์์ ๋ค์ ๋์์ผ๋ ๊ทธ ์ฌ์ด rolling duplicate 8-gram ๋น์จ์ 0์ด์๋ค. ๊ฐ์ seed๋ฅผ ํ์ฌ ๊ฐ์ง ๊ธฐ์ค์ผ๋ก ์คํํ ๊ฒฐ๊ณผ RWKV๋ 2180ํ ํฐ, OLMo๋ 1321ํ ํฐ์์ EOS๋ก ์ข
๋ฃํ๋ค. ์ด ๋ ์ด๊ธฐ threshold ์คํ์ pass@10 ์ง๊ณ์์ ์ ์ธํ๊ณ initial_loose_detector_false_positives์ ๋ณด๊ดํ๋ค.
๋ชจ๋ ๋ค ๋ชจ๋ธ์ 40๊ฐ transcript๋ฅผ ๋ต ๊ธฐ์ค์ผ๋ก ๊ฒํ ํ๋ค. ์ ๋ต 45 ํ๋ณธ์ ์์ผ๋ฉฐ, ์์ธ ํ๋ณธ๋ณ ํ์ ์ runs/passk_10_sampled_autostop/manual_answer_audit.md์ ๊ธฐ๋กํ๋ค. Qwen ํ๋ณธ์ 4096ํ ํฐ์์ ๋ต๋ณ ์ค๊ฐ์ ์๋ฆด ์ ์์ด, ๊ฐ ์ต์ข
550์์ 45 ์ธ๊ธ ์ฌ๋ถ๋ฅผ ํ์ธํ๋ค. ์๋ grader๋ ๋ต ํํ ๋ฐฉ์์ด ๋ฌ๋ผ์ง ์ ์์ผ๋ฏ๋ก manual review์ ํจ๊ป ํด์ํ๋ค.

๊ฐ ํ๋ณธ์ token_metrics.csv์๋ ์์ logits, ์ํ๋ง ์ฒ๋ฆฌ logits, ํ ํฐ๋ณ ๋ฐ๋ณต๋ฅ ์ด ์๋ค. RWKV์๋ ๊ฐ ์ธต์ WKV/attention-shift/FFN-shift RMS์ ๋ณํ๋๋ ๊ธฐ๋กํ๋ค. ์ด๋ฒ ๋ค ๋ชจ๋ธ 40๊ฐ run์์ ์ต์ข
๊ฐ์ง ๊ท์น์ผ๋ก ์กํ loop span์ ์๋ค. ๋์จํ ๊ธฐ์ค์ผ๋ก ์ธํ ๋ ์คํ์ transcript์ ๋น๊ต ๊ทธ๋ํ๋ ๋ณ๋ ํด๋์ ๋จ๊ฒจ ๋์๋ค.
## ํ์ผ ๊ตฌ์ฑ
- run_manifest.json: ๋ชจ๋ธ revision, ํ๊ฒฝ, ๊ณตํต prompt์ ์์ฑ ์ค์
- ๊ฐ run ๋๋ ํ ๋ฆฌ์ run_config.json: sample๋ณ ์ค์ ์
๋ ฅ, generation ์ค์ , ๊ฐ์ง ์ ์ฑ
- runs/00_exploratory_sampler_screening.md: ์ฌ์ ์งง์ prompt/sampler ํ์ธ
- comparison.csv ๋ฐ comparison_all_models.csv: ์ด์ controlled runs ๋น๊ต
- english_system_language_comparison.csv ๋ฐ english_system_comparison.json: ์์ด system ์ถ๊ฐ ์คํ ๋น๊ต
- periodic_harness_comparison.csv, periodic_harness_events.csv, periodic_harness_comparison.json: ์ฃผ๊ธฐ์ 2,048-token harness ๋น๊ต
- models/OLMo-2-0425-1B-Instruct/: ๊ณ ์ revision OLMo model๊ณผ tokenizer
- models/RWKV7-G1j-2.9B-20260831/: RWKV ๊ณต์ 2.9B ์ฒดํฌํฌ์ธํธ, ๊ณ ์ revision
- models/Qwen3.5-2B-Base/: Qwen ๊ณต์ Base ์ฒดํฌํฌ์ธํธ, ๊ณ ์ revision b1485b2fa6dfa1287294f269f5fb618e03d52d7c
- runs/11_rwkv_1_5b_pass10_sampled_autostop/: RWKV 1.5B sample 10๊ฐ, transcript์ logits/state
- runs/12_rwkv_2_9b_pass10_sampled_autostop/: RWKV 2.9B sample 10๊ฐ, transcript์ logits/state
- runs/13_olmo2_1b_pass10_sampled_autostop/: OLMo 2 sample 10๊ฐ์ logits
- runs/14_qwen3_5_2b_base_pass10_sampled_autostop/: Qwen3.5 Base sample 10๊ฐ์ logits
- runs/passk_10_sampled_autostop/passk_summary.csv ๋ฐ passk_summary.json: ๋ค ๋ชจ๋ธ pass@10 ํต๊ณ
- runs/passk_10_sampled_autostop/sample_results.csv: 40๊ฐ ํ๋ณธ๋ณ ์ ๋ต flag, ์ข
๋ฃ, ํ ํฐ ์์ metrics ์์ฝ
- runs/passk_10_sampled_autostop/manual_answer_audit.md: transcript ์๋ ์ ๋ต ํ์ธ
- runs/passk_10_sampled_autostop/experiment_notes.md: ์คํ ์กฐ๊ฑด, prompt ๊ฐ์ฌ, ๊ฐ์ง ๊ท์น, ๊ฒฐ๊ณผ ์์ธ
- runs/passk_10_sampled_autostop/sample_lengths_and_stops.png: ๋ค ๋ชจ๋ธ๋ณ ํ๋ณธ ๊ธธ์ด์ ์ข
๋ฃ ์ด์
- runs/passk_10_sampled_autostop/loop_span_comparisons.json: ๊ฐํ๋ ๊ธฐ์ค์ผ๋ก ๊ฒ์ถ๋ loop ๋น๊ต; ์ด๋ฒ์๋ ๋น์ด ์์
- runs/passk_10_sampled_autostop/archived_first_pass_duplicate_outputs/: ์ฒซ ๋ฐฐ์น์ RWKV/OLMo ๊ฒฐ๊ณผ ๋ณต์ฌ๋ณธ; ํ์ฌ ํ๋ณธ๊ณผ transcript ๋ฐ token metrics SHA-256 ์ผ์น, ๋ณ๋ ์ง๊ณํ์ง ์์
- runs/passk_10_sampled_autostop/initial_loose_detector_false_positives/: ์ฒซ detector ๊ธฐ์ค์ ์คํ ๋ sample๊ณผ ์ฌํ metrics
- scripts/run_passk_sampled_autostop.py: ๋ค ๋ชจ๋ธ sampled pass@k ์คํ, ๋ฐ๋ณต ์๋ ์ข
๋ฃ์ ํ ํฐ๋ณ ๊ณ์ธก
- scripts/analyze_passk_diagnostics.py: pass@10 ๊ทธ๋ํ์ ๋ฐ๋ณต span diagnostics ์์ฑ
์ด Vast ์ธ์คํด์ค์์ /workspace๋ host volume์ด ์๋๋ฏ๋ก instance recycle/destroy ์ ๋ชจ๋ธ๊ณผ ์คํ ๊ฒฐ๊ณผ๊ฐ ์ญ์ ๋๋ค.
|