Modern Post-Training: A Deep Dive โ Will Brown, Prime Intellect
ํ ์ค ํต์ฌ
์ด ์์์ ํ๋์ post-training์ ํ๊ฒฝยทํ๋ค์คยท๋ฐํ์ยท์ค์ผ์คํธ๋ ์ดํฐ๋ก ๋ถํดํด, ํ๊ฐ๋ถํฐ RLยทSFTยทdistillationยท๋ฐฐํฌ๊น์ง ํ๋์ ์ฌ์ฌ์ฉ ๊ฐ๋ฅํ ํ์ดํ๋ผ์ธ์ผ๋ก ๋ฌถ๋ ๋ฐฉ๋ฒ์ ์ค๋ช ํ๋ค.
์๊ฐ๋ณ ์์ฝ
-
0:12
Prime Intellect์ post-training ์คํ๊ณผ ๋ชฉํ๊ฐ ์๊ฐ๋๋ค. ์คํ์์ค ๋ชจ๋ธ์ ์ค์ ์ ๋ฌด์ ์ ํ์ ๋ง๊ฒ ๋ ์ ์ฐ๋๋ก, ๋๊ท๋ชจ ์ฐ๊ตฌ ์ธํ๋ผ์ ํจ๊ป ํ์ตยทํ๊ฐยท๋ฐฐํฌ๋ฅผ ํ ํ๋ฆ์ผ๋ก ์ฐ๊ฒฐํ๋ ๋ฐฉํฅ์ด ํต์ฌ์ด๋ค.
-
6:00
ํ๊ฒฝ(environment)์ ๋จ์ํ RL์ฉ ์ฅ์น๊ฐ ์๋๋ผ, ๋ฐ์ดํฐยท์๋๋ฆฌ์คยท์ํธ์์ฉ ๊ท์นยท์ฑ์ ๊ธฐ์ค์ ๋ด๋ ๊ณตํต ์ธ์ด๋ก ์ ์๋๋ค. ํ๊ฐ(evals)์ post-training์ด ๊ฐ์ ๋จ์๋ฅผ ๊ณต์ ํ๋ฏ๋ก, ํ๊ฒฝ์ ์ ์ค๊ณํ๋ฉด ๋ชจ๋ธ ์ ํ, ํ๋กฌํํธ ์ต์ ํ, RL, SFT ๋ฐ์ดํฐ ์์ฑ๊น์ง ์์ฐ์ค๋ฝ๊ฒ ์ด์ด์ง๋ค.
-
10:00
Verifiers V1์ ํ๊ฒฝ์ task set, harness, runtime์ผ๋ก ๋ถํดํด ์ฌ์ค๊ณํ ๊ฒ์ด ํต์ฌ์ด๋ค. task set์ agent-agnosticํ ๋ฐ์ดํฐ์ ๊ท์น, harness๋ CLI agent๋ recursive LM ๊ฐ์ ์คํ ๋ฐฉ์, runtime์ Dockerยทsandboxยท๋ก์ปฌ ์คํ์ ๋ด๋นํ๋ฉฐ, ์ด ๋ถ๋ฆฌ๋ก ํ๊ฐ์ ํ์ต์ด ๊ฐ์ ๊ตฌ์กฐ๋ฅผ ๊ณต์ ํ๊ฒ ๋๋ค.
-
17:00
๋ณด์๊ณผ ๋ฉํธ๋ฆญ, group reward, ์ฌ์ฉ์ ์๋ฎฌ๋ ์ดํฐ, interception server, trace graph, renderers ๊ฐ์ ๊ตฌ์ฑ์์๊ฐ ํ๋์ agent ํ์ต์ ์ธ๋ถ ๋ฌธ์ ๋ฅผ ํด๊ฒฐํ๋ค. ํนํ ๊ธธ์ด ํจ๋ํฐยท๊ฐ๊ฒฐ์ฑ ๋ณด์์ฒ๋ผ ์ฌ๋ฌ ์ํ์ ๋น๊ตํด ์ต์ ํํ๋ฅผ ์ ๋ํ๋ ์ค๊ณ์, token space์ message space๋ฅผ ๋ถ๋ฆฌํด ๋ค๋ฃจ๋ ๊ตฌ์กฐ๊ฐ ์ค์ํ๊ฒ ๋ค๋ค์ง๋ค.
-
29:00
Primer RL์ async RL์ ์ ์ ๋ก ํ ์ค์ผ์คํธ๋ ์ด์ ๊ณ์ธต์ผ๋ก, inference์ trainer๋ฅผ ๋ถ๋ฆฌํด ๋๋ฆฐ rollout์ ์ ์ฒด ํ์ต ์๋๊ฐ ๋ฌถ์ด์ง ์๋๋ก ํ๋ค. DPPO, on-policy distillation, self-distillation, GRPO ๊ฐ์ ์๊ณ ๋ฆฌ์ฆ์ loss์ algorithm ๋ ๋ฒจ๋ก ์ถ์ํํด, ํ๊ฒฝ๊ณผ ์ธํ๋ผ๋ฅผ ๋ฐ๊พธ์ง ์๊ณ ๋ ๋ค์ํ post-training ๋ฐฉ์์ ์กฐํฉํ ์ ์๊ฒ ๋ง๋ ๋ค.
-
42:01
ํธ์คํฐ๋ ํ์ต ํ๋ซํผ์ multi-tenant Laura๋ฅผ ๋จผ์ ์ ๊ณตํ๊ณ , ๊ณง full fine-tuning๊น์ง ํ์ฅํ๋ค. ์ฌ์ฉ์๋ GPU ์ด์์ ์ง์ ์ ๊ฒฝ ์ฐ์ง ์๊ณ ๋ ํ๊ฒฝ ํจํค์ง, ๋ณด์ ํจ์, ์๊ณ ๋ฆฌ์ฆ, ํ์ต ์ค์ ์ ๋จ๊ณ์ ์ผ๋ก ์กฐ์ ํ๋ฉฐ ๋๊ท๋ชจ ์คํ์ ๋น ๋ฅด๊ฒ ๋ฐ๋ณตํ ์ ์๋ค.
์ค๋ฌด ์ธ์ฌ์ดํธ
- post-training์ ์์ํ ๋๋ ๋จผ์ ํ๊ฒฝ๊ณผ eval์ ์ค๊ณํ๋ฉด, ๊ทธ ๊ฒฐ๊ณผ๋ฅผ ๊ทธ๋๋ก RLยทSFTยทdistillation๋ก ์ฌ์ฌ์ฉํ ์ ์์ด ์ค๋ณต ๊ตฌํ์ ํฌ๊ฒ ์ค์ผ ์ ์๋ค.
- ๊ธด ์ถ๋ก ๊ณผ ๋ค์ํ rollout ์๊ฐ์ด ์์ด๋ agent ์์ ์์๋ async ๊ตฌ์กฐ๊ฐ ํ์ต ํจ์จ์ ์ข์ฐํ๋ฏ๋ก, inference์ trainer๋ฅผ ๋ถ๋ฆฌํ๊ณ off-policy ํ์ฉ ๋ฒ์๋ฅผ ๊ด๋ฆฌํ๋ ๊ฒ์ด ์ค์ํ๋ค.
- ์ค๋ฌด์์๋ correctness๋ง ๋ณด์ง ๋ง๊ณ group reward๋ก ๊ฐ๊ฒฐ์ฑ, ๋น์ฉ, ์ฌ์ฉ์ ์ํธ์์ฉ ํ์ง๊น์ง ํจ๊ป ์ต์ ํํด์ผ ์ค์ ์ ํ ์ฑ๋ฅ์ด ์์ ์ ์ผ๋ก ์ฌ๋ผ๊ฐ๋ค.