์ž์œจ ์ง€์†ํ•™์Šตํ˜• ๊ธˆ์œต ํŠธ๋ ˆ์ด๋”ฉ ์—์ด์ „ํŠธ

๊ฐœ๋ฐœ ์ค‘์ธ ์—ฐ๊ตฌ ํ”„๋กœ์ ํŠธ์ž…๋‹ˆ๋‹ค. ์ด ์ €์žฅ์†Œ์˜ ์ฒดํฌํฌ์ธํŠธ๋Š” ์™„์„ฑ๋˜๊ฑฐ๋‚˜ ์ˆ˜์ต์„ฑ์ด ๊ฒ€์ฆ๋œ ์ƒํ’ˆ์ด ์•„๋‹™๋‹ˆ๋‹ค. ํˆฌ์ž ์กฐ์–ธ์ด๋‚˜ ๋งค๋งค ์ถ”์ฒœ์œผ๋กœ ์‚ฌ์šฉํ•˜์ง€ ๋งˆ์„ธ์š”.

ํ”„๋กœ์ ํŠธ ๋ชฉํ‘œ

์ด ํ”„๋กœ์ ํŠธ์˜ ๋ชฉํ‘œ๋Š” ํŠน์ • ํŠธ๋ ˆ์ด๋”์˜ ๋งค๋งค๋ฒ•์ด๋‚˜ ๊ณ ์ •๋œ ์ „๋žต์„ ์˜๊ตฌ์ ์œผ๋กœ ๋ณต์‚ฌํ•˜๋Š” ๊ฒƒ์ด ์•„๋‹™๋‹ˆ๋‹ค. ์‹œ์žฅ์„ ๊ณ„์† ๊ด€์ฐฐํ•˜๊ณ , ๊ฐ€์ƒ ๋งค๋งค์˜ ๊ฒฐ๊ณผ์—์„œ ์ˆ˜์ˆ˜๋ฃŒยท์„ธ๊ธˆยท์Šคํ”„๋ ˆ๋“œยท์Šฌ๋ฆฌํ”ผ์ง€๋ฅผ ๋ฐ˜์˜ํ•œ ์ˆœ์†์ต์„ ๊ฒฝํ—˜์œผ๋กœ ์‚ผ์•„ ์ •์ฑ…์„ ๋ฐœ์ „์‹œํ‚ค๋Š” ์ž์œจ ํŠธ๋ ˆ์ด๋”ฉ ์—์ด์ „ํŠธ๋ฅผ ๋งŒ๋“œ๋Š” ๊ฒƒ์ž…๋‹ˆ๋‹ค.

๋ชจ๋ธ์€ ์‹œ์žฅ ๋ฐ์ดํ„ฐ์™€ ํฌํŠธํด๋ฆฌ์˜ค ์ƒํƒœ๋ฅผ ํ•จ๊ป˜ ๋ณด๊ณ  ์ข…๋ชฉ์„ ์„ ํƒํ•˜๊ณ , ๋งค์ˆ˜ยท๋ณด์œ ยท๋งค๋„ ํ–‰๋™๊ณผ ์ž๊ธˆ ๋ฐฐ๋ถ„์„ ์ •ํ•˜๋ฉฐ, ๋ณด์œ  ํฌ์ง€์…˜์„ ์œ ์ง€ยท๊ต์ฒดยท์ถ•์†Œยท์ฒญ์‚ฐํ•˜๋Š” ๋ฐฉํ–ฅ์œผ๋กœ ํ•™์Šตํ•ฉ๋‹ˆ๋‹ค. ํ˜„๊ธˆ, ๋ณด์œ ์ˆ˜๋Ÿ‰, ํ‰๋‹จ, ์‹คํ˜„์†์ต, ํ‰๊ฐ€์†์ต, ๊ฑฐ๋ž˜๋น„์šฉ, ์œ ๋™์„ฑ, ํˆฌ์ž ๊ฐ€๋Šฅ ์ž๋ณธ์€ ํ™˜๊ฒฝ์ด ๊ณ„์‚ฐํ•ด ๋ชจ๋ธ์— ์•Œ๋ ค์ฃผ๋Š” ๊ธˆ์œต ๊ทœ์น™์ž…๋‹ˆ๋‹ค. ์–ด๋–ค ์ƒํ™ฉ์—์„œ ์–ด๋–ค ์ž์‚ฐ์„ ์‚ฌ๊ณ  ์–ผ๋งˆ๋‚˜ ๋ฐฐ๋ถ„ํ• ์ง€๋Š” ๊ฒฝํ—˜์œผ๋กœ ์ฐพ๊ฒŒ ํ•˜๋Š” ๊ฒƒ์ด ๋ชฉํ‘œ์ž…๋‹ˆ๋‹ค.

๊ณต๊ฐœ ๋ชจ๋ธ, teacher ์‹ ํ˜ธ, ๊ณผ๊ฑฐ ํŠธ๋ ˆ์ด๋” ๊ธฐ๋ก์€ ์ดˆ๊ธฐ ๊ธˆ์œต ๋ฌธ๋ฒ•์„ ์ตํžˆ๋Š” ์ž๋ฃŒ๋กœ ์“ธ ์ˆ˜ ์žˆ์ง€๋งŒ, ์ตœ์ข… ์ •์ฑ…์˜ ์ •๋‹ต์œผ๋กœ ๊ณ ์ •ํ•˜์ง€ ์•Š์Šต๋‹ˆ๋‹ค. ์žฅ๊ธฐ ๋ชฉํ‘œ๋Š” ์‹œ์žฅ ๊ฒฝํ—˜๊ณผ ๋ณด์ƒ์„ ์ด์šฉํ•ด ๋…๋ฆฝ์ ์ธ ์ •์ฑ…์„ ๋ฐœ์ „์‹œํ‚ค๋Š” ๊ฒƒ์ž…๋‹ˆ๋‹ค.

ํ˜„์žฌ ๋ชจ๋ธ ์„ค์ •๊ฐ’

์ „์ฒด ๊ตฌ์กฐ

ํ˜„์žฌ ๊ธฐ์ค€ ๋ชจ๋ธ์€ PyTorch Transformer์ด๋ฉฐ ์ „์ฒด ํŒŒ๋ผ๋ฏธํ„ฐ๋Š” 512,723,465๊ฐœ์ž…๋‹ˆ๋‹ค. ์—ฌ๊ธฐ์„œ ํŒŒ๋ผ๋ฏธํ„ฐ๋Š” ํ•™์Šต์œผ๋กœ ์กฐ์ •๋  ์ˆ˜ ์žˆ๋Š” ๋ชจ๋ธ ๊ฐ€์ค‘์น˜์˜ ์ˆซ์ž์ž…๋‹ˆ๋‹ค. ์•ฝ 5.13์–ต ๊ฐœ์˜ ํŒŒ๋ผ๋ฏธํ„ฐ๋ฅผ ๊ฐ€์ง„ ํ•˜๋‚˜์˜ ๋ชจ๋ธ์ด๊ณ , champion๊ณผ candidate๋Š” ์ด ๊ตฌ์กฐ๋ฅผ ์‚ฌ์šฉํ•˜๋Š” ์„œ๋กœ ๋‹ค๋ฅธ ์ฒดํฌํฌ์ธํŠธ์ž…๋‹ˆ๋‹ค.

์„ค์ • ํ•ญ๋ชฉ ํ˜„์žฌ ๊ฐ’ ์‰ฌ์šด ์„ค๋ช…
์ „์ฒด ํŒŒ๋ผ๋ฏธํ„ฐ 512,723,465๊ฐœ ์ „์ฒด ๋ชจ๋ธ์— ๋“ค์–ด ์žˆ๋Š” ๊ฐ€์ค‘์น˜ ์ˆ˜์ž…๋‹ˆ๋‹ค.
๋ชจ๋ธ ๋ณธ์ฒด Transformer ์‹œ๊ฐ„์— ๋”ฐ๋ฅธ ๋ณ€ํ™”์™€ ์ข…๋ชฉ ์‚ฌ์ด ๊ด€๊ณ„๋ฅผ ํ•จ๊ป˜ ์ฒ˜๋ฆฌํ•ฉ๋‹ˆ๋‹ค.
Transformer ์ธต ์ˆ˜ 21์ธต ์ž…๋ ฅ์„ ๋‹จ๊ณ„์ ์œผ๋กœ ๋ถ„์„ํ•˜๋Š” ๋ณธ์ฒด ๋ธ”๋ก ์ˆ˜์ž…๋‹ˆ๋‹ค.
hidden size 1,408 ๊ฐ ์‹œ์ ๊ณผ ์ข…๋ชฉ์„ ๋ชจ๋ธ ๋‚ด๋ถ€์—์„œ ํ‘œํ˜„ํ•˜๋Š” ๋ฒกํ„ฐ์˜ ํญ์ž…๋‹ˆ๋‹ค.
attention head ์ˆ˜ 16๊ฐœ attention ๊ณ„์‚ฐ์„ 16๊ฐœ์˜ ๊ด€์ ์œผ๋กœ ๋‚˜๋ˆ„์–ด ์ˆ˜ํ–‰ํ•ฉ๋‹ˆ๋‹ค.
head dimension 88 head ํ•œ ๊ฐœ์˜ ํ‘œํ˜„ ํฌ๊ธฐ์ž…๋‹ˆ๋‹ค. 1,408 รท 16 = 88์ž…๋‹ˆ๋‹ค.
FFN ํฌ๊ธฐ 5,632 ๊ฐ Transformer ๋ธ”๋ก ์•ˆ์—์„œ ์ •๋ณด๋ฅผ ํ•œ ๋ฒˆ ๋” ๋ณ€ํ™˜ํ•˜๋Š” ์ค‘๊ฐ„์ธต์˜ ํญ์ž…๋‹ˆ๋‹ค.
์‹œ๊ฐ„์ถ• attention ์ธต 11๊ฐœ ๊ฐ ์ข…๋ชฉ์˜ ๊ณผ๊ฑฐ ๋ด‰๊ณผ ์‹œ๊ฐ„์— ๋”ฐ๋ฅธ ํ๋ฆ„์„ ์‚ดํ•๋‹ˆ๋‹ค.
์ข…๋ชฉ ๊ฐ„ attention ์ธต 10๊ฐœ ์—ฌ๋Ÿฌ ์ข…๋ชฉ๊ณผ ์‹œ์žฅ ์‚ฌ์ด์˜ ์›€์ง์ž„ ๊ด€๊ณ„๋ฅผ ์‚ดํ•๋‹ˆ๋‹ค.
์ž…๋ ฅ sequence 128๋ด‰ ํ•œ ๋ฒˆ์˜ ํŒ๋‹จ์—์„œ ์ฐธ๊ณ ํ•˜๋Š” ๊ณผ๊ฑฐ ์‹œ๊ฐ„ ๊ตฌ๊ฐ„ ๊ธธ์ด์ž…๋‹ˆ๋‹ค.
์ข…๋ชฉ๋ณ„ ์ž…๋ ฅ feature 17๊ฐœ ๊ฐ€๊ฒฉยท๊ฑฐ๋ž˜ ๋“ฑ ์ข…๋ชฉ๋งˆ๋‹ค ๋“ค์–ด์˜ค๋Š” ์ž…๋ ฅ ํ•ญ๋ชฉ ์ˆ˜์ž…๋‹ˆ๋‹ค.
market context 16๊ฐœ ์ง€์ˆ˜์™€ ์‹œ์žฅ ์ „๋ฐ˜ ์ƒํ™ฉ์ฒ˜๋Ÿผ ์—ฌ๋Ÿฌ ์ข…๋ชฉ์— ๊ณตํ†ต์ธ ์ž…๋ ฅ ํ•ญ๋ชฉ ์ˆ˜์ž…๋‹ˆ๋‹ค.
ํ–‰๋™ ์ ์ˆ˜ SELL / HOLD / BUY logits ์ข…๋ชฉ๋ณ„๋กœ ์„ธ ํ–‰๋™์˜ ์ ์ˆ˜๋ฅผ ์ถœ๋ ฅํ•ฉ๋‹ˆ๋‹ค.
๊ฐ€์น˜ ์ถœ๋ ฅ value ํ˜„์žฌ ์ƒํƒœ์˜ ์žฅ๊ธฐ์ ์ธ ๊ธฐ๋Œ€ ๊ฐ€์น˜๋ฅผ ์ถ”์ •ํ•˜๋Š” ์ถœ๋ ฅ์ž…๋‹ˆ๋‹ค.
๋ฐฐ๋ถ„ ์ถœ๋ ฅ portfolio allocation ํฌํŠธํด๋ฆฌ์˜ค ์ž๊ธˆ์„ ์ข…๋ชฉ๋ณ„๋กœ ์–ด๋–ป๊ฒŒ ๋ฐฐ๋ถ„ํ• ์ง€ ๋‚˜ํƒ€๋‚ด๋Š” ์ถœ๋ ฅ์ž…๋‹ˆ๋‹ค.
๊ธฐ๋ณธ ์ •๋ฐ€๋„ FP16 ๋ชจ๋ธ ์—ฐ์‚ฐ์— ์‚ฌ์šฉํ•˜๋Š” ๊ธฐ๋ณธ ์ˆ˜์น˜ ์ •๋ฐ€๋„์ž…๋‹ˆ๋‹ค.

์ž…๋ ฅ ๊ธธ์ด 128๋ด‰์˜ ์˜๋ฏธ

128๋ด‰์€ ์ข…๋ชฉ ์ˆ˜๊ฐ€ ์•„๋‹ˆ๋ผ ์‹œ๊ฐ„์ถ• ๊ธธ์ด์ž…๋‹ˆ๋‹ค. ์˜ˆ๋ฅผ ๋“ค์–ด ๋ชจ๋ธ์€ ๊ฐ ๋Œ€์ƒ ์ข…๋ชฉ์— ๋Œ€ํ•ด ์ตœ๊ทผ 128๊ฐœ์˜ ์‹œ๊ฐ„ ๋‹จ์œ„ ๋ด‰์„ ์ฐธ๊ณ ํ•ฉ๋‹ˆ๋‹ค. ์ข…๋ชฉ๋ณ„ feature 17๊ฐœ๋Š” ๊ฐ ์ข…๋ชฉ์˜ ์ž…๋ ฅ์ด๊ณ , market context 16๊ฐœ๋Š” ๊ฐ™์€ ์‹œ์ ์˜ ์‹œ์žฅ ์ „์ฒด ์ƒํ™ฉ์„ ์„ค๋ช…ํ•˜๋Š” ์ž…๋ ฅ์ž…๋‹ˆ๋‹ค.

๋”ฐ๋ผ์„œ โ€œ128๋ด‰์„ ๋ณธ๋‹คโ€๋Š” ๋ง์€ โ€œ์ตœ๋Œ€ 128์ข…๋ชฉ๋งŒ ์ฒ˜๋ฆฌํ•œ๋‹คโ€๋Š” ๋œป์ด ์•„๋‹™๋‹ˆ๋‹ค. ์—ฌ๋Ÿฌ ์ข…๋ชฉ์˜ ์ž…๋ ฅ์„ ํ•จ๊ป˜ ๋†“๊ณ  ์ข…๋ชฉ ๊ฐ„ ๊ด€๊ณ„๋ฅผ ๋ถ„์„ํ•˜๋Š” ์ธต๋„ ๋ณ„๋„๋กœ ์žˆ์Šต๋‹ˆ๋‹ค. ์‹œ๊ฐ„์ถ• ์ฒ˜๋ฆฌ์™€ ์ข…๋ชฉ ๊ฐ„ ์ฒ˜๋ฆฌ์˜ ์„ค์ •์€ ์„œ๋กœ ๋‹ค๋ฅธ ์—ญํ• ์ž…๋‹ˆ๋‹ค.

Transformer 21์ธต์˜ ๊ตฌ์„ฑ

์ „์ฒด 21์ธต์€ ๋‘ ์ข…๋ฅ˜์˜ ๋ถ„์„์„ ๋‚˜๋ˆ„์–ด ๋‹ด๋‹นํ•ฉ๋‹ˆ๋‹ค. ์•ฝ 11๊ฐœ ์‹œ๊ฐ„์ถ• attention ์ธต์€ ํ•œ ์ข…๋ชฉ์˜ ๊ณผ๊ฑฐ ์›€์ง์ž„์„ ์ฝ๊ณ , ์•ฝ 10๊ฐœ ์ข…๋ชฉ ๊ฐ„ attention ์ธต์€ ๋‹ค๋ฅธ ์ข…๋ชฉ ๋ฐ ์‹œ์žฅ ๋ฌธ๋งฅ๊ณผ์˜ ๊ด€๊ณ„๋ฅผ ์ฝ์Šต๋‹ˆ๋‹ค. ์ด ๊ตฌ๋ถ„์„ ํ†ตํ•ด ํ•œ ์ข…๋ชฉ ์ž์ฒด์˜ ํŒจํ„ด๊ณผ ์‹œ์žฅ ์ „์ฒด์—์„œ์˜ ์ƒ๋Œ€์  ์›€์ง์ž„์„ ํ•จ๊ป˜ ํ‘œํ˜„ํ•˜๋ ค๋Š” ๊ตฌ์กฐ์ž…๋‹ˆ๋‹ค.

hidden size 1,408์€ ์ด ๋ถ„์„ ๊ณผ์ •์—์„œ ์œ ์ง€๋˜๋Š” ๋‚ด๋ถ€ ํ‘œํ˜„์˜ ํญ์ž…๋‹ˆ๋‹ค. 16๊ฐœ attention head๊ฐ€ ์ด ํ‘œํ˜„์„ ๋‚˜๋ˆ„์–ด ๋ณด๊ณ , ๊ฐ head๋Š” 88์ฐจ์›์„ ์‚ฌ์šฉํ•ฉ๋‹ˆ๋‹ค. FFN ํฌ๊ธฐ 5,632๋Š” attention์„ ํ†ต๊ณผํ•œ ์ •๋ณด๋ฅผ ๋ธ”๋ก ์•ˆ์—์„œ ํ™•์žฅํ•ด ๋‹ค์‹œ ๋ณ€ํ™˜ํ•˜๋Š” ๋ถ€๋ถ„์˜ ํฌ๊ธฐ์ž…๋‹ˆ๋‹ค.

๋ชจ๋ธ์ด ๋‚ด๋Š” ํ–‰๋™

๋ชจ๋ธ์€ ๊ฐ ์ข…๋ชฉ์— ๋Œ€ํ•ด SELL, HOLD, BUY ์„ธ ํ–‰๋™์˜ logits๋ฅผ ์ถœ๋ ฅํ•ฉ๋‹ˆ๋‹ค. logits๋Š” ํ–‰๋™๋ณ„ ์›์ ์ˆ˜์ด๋ฉฐ, ๊ทธ ์ž์ฒด๊ฐ€ ์ฒด๊ฒฐ ๊ฐ€๊ฒฉ์ด๋‚˜ ์ˆ˜์ต ๋ณด์žฅ์€ ์•„๋‹™๋‹ˆ๋‹ค. value ์ถœ๋ ฅ์€ ์ƒํƒœ์˜ ๊ฐ€์น˜๋ฅผ ๋‚˜ํƒ€๋‚ด๋Š” ํ•™์Šต ์‹ ํ˜ธ์ด๊ณ , portfolio allocation ์ถœ๋ ฅ์€ ์ข…๋ชฉ๋ณ„ ์ž๊ธˆ ๋ฐฐ๋ถ„ ๊ฒฐ์ •์„ ํ‘œํ˜„ํ•ฉ๋‹ˆ๋‹ค.

์‹ค์ œ ๊ฐ€์ƒ ์ฃผ๋ฌธ๊ณผ ๊ณ„์ขŒ ๋ณ€ํ™”๋Š” ์ด ์ถœ๋ ฅ์„ ํ˜„๊ธˆ, ๋ณด์œ ์ˆ˜๋Ÿ‰, ๊ฐ€๊ฒฉ, ๊ฑฐ๋ž˜ ๊ฐ€๋Šฅ ์—ฌ๋ถ€ ๋ฐ ์ˆ˜์ˆ˜๋ฃŒยท์„ธ๊ธˆยท์Šคํ”„๋ ˆ๋“œยท์Šฌ๋ฆฌํ”ผ์ง€ ๊ทœ์น™๊ณผ ํ•จ๊ป˜ ์ฒ˜๋ฆฌํ•ด ๊ณ„์‚ฐํ•ด์•ผ ํ•ฉ๋‹ˆ๋‹ค. ๋ชจ๋ธ ์ถœ๋ ฅ๋งŒ์œผ๋กœ ๊ฑฐ๋ž˜๊ฐ€ ์ฒด๊ฒฐ๋˜๋Š” ๊ฒƒ์€ ์•„๋‹™๋‹ˆ๋‹ค.

์‹ค์‹œ๊ฐ„ continual ํ•™์Šต ์„ค์ •

์ „์ฒด ๋ชจ๋ธ ํŒŒ๋ผ๋ฏธํ„ฐ๊ฐ€ ๋ชจ๋‘ ๋งค๋ฒˆ ๋ฐ”๋€Œ๋Š” ๊ฒƒ์€ ์•„๋‹™๋‹ˆ๋‹ค. ์‹ค์‹œ๊ฐ„ continual ํ•™์Šต์—์„œ๋Š” ๋ชจ๋ธ ๋ณธ์ฒด ๋Œ€๋ถ€๋ถ„์„ ๊ณ ์ •ํ•˜๊ณ  ๋งˆ์ง€๋ง‰ ๋ถ€๋ถ„๊ณผ ์ถœ๋ ฅ head๋ฅผ ํ•™์Šต ๊ฐ€๋Šฅํ•˜๊ฒŒ ๋‘ก๋‹ˆ๋‹ค. ์ด ๋ฐฉ์‹์œผ๋กœ ์‹œ์žฅ ํŒ๋‹จ์— ์“ฐ๋Š” ํฐ ๋ชจ๋ธ ๊ตฌ์กฐ๋Š” ์œ ์ง€ํ•˜๋ฉด์„œ ์ƒˆ ๊ฒฝํ—˜์— ๋งž์ถฐ ์ •์ฑ…ยท๊ฐ€์น˜ยท๋ฐฐ๋ถ„ ๋ถ€๋ถ„์„ ์—…๋ฐ์ดํŠธํ•ฉ๋‹ˆ๋‹ค.

์‹ค์‹œ๊ฐ„ ํ•™์Šต ํ•ญ๋ชฉ ํ˜„์žฌ ๊ธฐ์ค€ ์˜๋ฏธ
ํ•™์Šต ๊ฐ€๋Šฅ ํŒŒ๋ผ๋ฏธํ„ฐ 95,243,017๊ฐœ ์ „์ฒด 512,723,465๊ฐœ ์ค‘ ํ˜„์žฌ optimizer๊ฐ€ ์—…๋ฐ์ดํŠธํ•  ์ˆ˜ ์žˆ๋„๋ก ํ’€์–ด๋‘” ์ˆ˜์ž…๋‹ˆ๋‹ค.
ํ•™์Šต ๋ฒ”์œ„ ๋งˆ์ง€๋ง‰ Transformer 4๊ฐœ ์ธต + policy/value/portfolio head ์•ž๋ถ€๋ถ„ ๋ณธ์ฒด๋Š” ๊ณ ์ •ํ•˜๊ณ  ๋งˆ์ง€๋ง‰ ๋„ค ์ธต ๋ฐ ํ–‰๋™ยท๊ฐ€์น˜ยท๋ฐฐ๋ถ„ ์ถœ๋ ฅ์„ ํ•™์Šตํ•ฉ๋‹ˆ๋‹ค.
batch size 8 optimizer step ํ•˜๋‚˜์—์„œ replay ๊ฒฝํ—˜ 8๊ฐœ๋ฅผ ๋ฌถ์–ด ์ฒ˜๋ฆฌํ•ฉ๋‹ˆ๋‹ค.
optimizer step ํ•™์Šต 1ํšŒ๋‹น 8ํšŒ ํ•œ ๋ฒˆ์˜ candidate ์—…๋ฐ์ดํŠธ ๋™์•ˆ optimizer๊ฐ€ ์ˆ˜ํ–‰ํ•˜๋Š” ๊ฐฑ์‹  ํšŸ์ˆ˜์ž…๋‹ˆ๋‹ค.
๊ณ ์œ  ๊ฒฝํ—˜ ์ˆ˜ ํ•™์Šต 1ํšŒ๋‹น 64๊ฐœ batch 8๊ฐœ๋ฅผ 8 step ์ฒ˜๋ฆฌํ•˜๋Š” ํ•œ ์—…๋ฐ์ดํŠธ์—์„œ ์‚ฌ์šฉํ•˜๋Š” ๊ฒฝํ—˜ ์ˆ˜ ๊ธฐ์ค€์ž…๋‹ˆ๋‹ค.
์ตœ๊ทผ ํ•™์Šต ์‹œ๊ฐ„ ์•ฝ 246์ดˆ ์ตœ๊ทผ ํ•œ ํ•™์Šต ์—…๋ฐ์ดํŠธ์— ๊ฑธ๋ฆฐ ์‹œ๊ฐ„์ž…๋‹ˆ๋‹ค. ์‹คํ–‰ ์ƒํƒœ์™€ ์‹œ์Šคํ…œ ๋ถ€ํ•˜์— ๋”ฐ๋ผ ๋‹ฌ๋ผ์ง‘๋‹ˆ๋‹ค.
peak VRAM ์•ฝ 4.14 GB ์ตœ๊ทผ ํ•™์Šต ์ค‘ ๊ด€์ธก๋œ GPU ๋ฉ”๋ชจ๋ฆฌ ์ตœ๊ณ  ์‚ฌ์šฉ๋Ÿ‰์ž…๋‹ˆ๋‹ค. GPU์™€ ๋ถ€ํ•˜์— ๋”ฐ๋ผ ๋‹ฌ๋ผ์ง‘๋‹ˆ๋‹ค.
์Šน๊ธ‰ ๊ฒ€์ฆ ๊ตฌ๊ฐ„ 128๋ด‰ ๊ณ ์ •๋œ candidate snapshot๊ณผ champion์„ ๋น„๊ตํ•˜๋Š” ๋ฏธ๋ž˜ paper ๊ตฌ๊ฐ„ ๊ธธ์ด์ž…๋‹ˆ๋‹ค.
๊ฒ€์ฆ ์ค‘ candidate ํ•™์Šต ๊ณ„์† ์ง„ํ–‰ ๊ฒ€์ฆ snapshot์€ ๊ณ ์ •ํ•˜๊ณ , ํ˜„์žฌ candidate ๋ณธ์ฒด๋Š” ์ƒˆ ๊ฒฝํ—˜์œผ๋กœ ํ•™์Šต์„ ์ด์–ด๊ฐ‘๋‹ˆ๋‹ค.
checkpoint ํŒŒ์ผ champion.pt 1๊ฐœ + candidate.pt 1๊ฐœ ์ผ๋ฐ˜ ์šด์˜์—์„œ ์œ ์ง€ํ•˜๋Š” ๋ชจ๋ธ checkpoint๋Š” ์ด ๋‘ ๊ฐœ์ž…๋‹ˆ๋‹ค.

์ „์ฒด ํŒŒ๋ผ๋ฏธํ„ฐ 512,723,465๊ฐœ์™€ ์‹ค์‹œ๊ฐ„ ํ•™์Šต ๊ฐ€๋Šฅ ํŒŒ๋ผ๋ฏธํ„ฐ 95,243,017๊ฐœ๋Š” ์„œ๋กœ ๋‹ค๋ฅธ ์ˆ˜์น˜์ž…๋‹ˆ๋‹ค. ์ฒซ ๋ฒˆ์งธ๋Š” ๋ชจ๋ธ ์ „์ฒด์˜ ํฌ๊ธฐ์ด๊ณ , ๋‘ ๋ฒˆ์งธ๋Š” continual ํ•™์Šต ๋•Œ ์‹ค์ œ๋กœ ์—…๋ฐ์ดํŠธ ๋Œ€์ƒ์œผ๋กœ ํ’€์–ด ๋†“๋Š” ๋ถ€๋ถ„์ž…๋‹ˆ๋‹ค. ๋งˆ์ง€๋ง‰ ๋„ค Transformer ์ธต๊ณผ ์ถœ๋ ฅ head ์™ธ์˜ ํŒŒ๋ผ๋ฏธํ„ฐ๋Š” ์ด ํ•™์Šต ์„ค์ •์—์„œ ๊ณ ์ •๋ฉ๋‹ˆ๋‹ค.

ํ•™์Šต 1ํšŒ๋Š” replay์—์„œ ๊ฒฝํ—˜์„ batch๋กœ ๊ฐ€์ ธ์™€ 8ํšŒ optimizer update๋ฅผ ์ˆ˜ํ–‰ํ•ฉ๋‹ˆ๋‹ค. ์ตœ๊ทผ ์ธก์ •๊ฐ’์€ ์•ฝ 246์ดˆ์™€ peak VRAM 4.14 GB์ž…๋‹ˆ๋‹ค. ์ด๋Š” ํŠน์ • ์‹คํ–‰์˜ ๊ด€์ธก๊ฐ’์ด๋ฉฐ, ๋‹ค๋ฅธ GPUยท์‹œ์žฅ ์ž…๋ ฅยท์‹œ์Šคํ…œ ๋ถ€ํ•˜์—์„œ ๊ฐ™์€ ๊ฒฐ๊ณผ๋ฅผ ๋ณด์žฅํ•˜์ง€ ์•Š์Šต๋‹ˆ๋‹ค.

Champion, candidate, ๊ฒ€์ฆ์˜ ์—ญํ• 

  • Champion์€ ํ˜„์žฌ ๊ธฐ์ค€ ๋ชจ๋ธ์ž…๋‹ˆ๋‹ค. ์šด์˜ ์‹œ์žฅ ํŒ๋‹จ์„ ๋‹ด๋‹นํ•ฉ๋‹ˆ๋‹ค.
  • Candidate๋Š” ์ƒˆ replay ๊ฒฝํ—˜์„ ํ•™์Šตํ•˜๋Š” ๋ชจ๋ธ์ž…๋‹ˆ๋‹ค. ์—…๋ฐ์ดํŠธ๋œ candidate๊ฐ€ ์ž๋™์œผ๋กœ champion์ด ๋˜๋Š” ๊ฒƒ์€ ์•„๋‹™๋‹ˆ๋‹ค.
  • ๊ฒ€์ฆ snapshot์€ ์‹œํ—˜์„ ์‹œ์ž‘ํ•  ๋•Œ candidate์˜ ๊ฐ€์ค‘์น˜๋ฅผ ๊ณ ์ •ํ•œ ํ‰๊ฐ€๋ณธ์ž…๋‹ˆ๋‹ค. ํ‰๊ฐ€ ์ค‘ ๊ฐ€์ค‘์น˜๊ฐ€ ๋ฐ”๋€Œ์ง€ ์•Š์•„์•ผ ์‹œํ—˜์˜ ์ฒ˜์Œ๊ณผ ๋์ด ๊ฐ™์€ ๋ชจ๋ธ์„ ํ‰๊ฐ€ํ•ฉ๋‹ˆ๋‹ค.
  • ๊ฒ€์ฆ ์ค‘ candidate ๋ณธ์ฒด๋Š” ๋‹ค์Œ ๊ฒฝํ—˜์„ ํ•™์Šตํ•  ์ˆ˜ ์žˆ์Šต๋‹ˆ๋‹ค. ๊ฒ€์ฆ snapshot๊ณผ ํ•™์Šต ์ค‘์ธ candidate๋Š” ์—ญํ• ์ด ๋‹ค๋ฆ…๋‹ˆ๋‹ค.
  • Champion๊ณผ ๊ณ ์ • candidate snapshot์€ ๊ฐ™์€ ๋ฏธ๋ž˜ ์‹œ์žฅ ๊ตฌ๊ฐ„, ๊ฐ™์€ ์‹œ์ž‘ ๊ณ„์ขŒ ์กฐ๊ฑด, ๊ฐ™์€ ๊ฑฐ๋ž˜๋น„์šฉ ์กฐ๊ฑด์œผ๋กœ ๋น„๊ตํ•ด์•ผ ํ•ฉ๋‹ˆ๋‹ค.
  • ์Šน๊ธ‰ ๊ธฐ์ค€์€ ๊ฑฐ๋ž˜ ๋น„์šฉ์„ ์ฐจ๊ฐํ•œ paper-account ์ˆœ์†์ต ๊ฐœ์„ ์ž…๋‹ˆ๋‹ค. ์‹œํ—˜์—์„œ ์ด๊ฒผ๋‹ค๋Š” ๊ทผ๊ฑฐ๊ฐ€ ์—†์œผ๋ฉด champion์„ ๋ฐ”๊พธ์ง€ ์•Š์Šต๋‹ˆ๋‹ค.

๊ฒ€์ฆ ๊ตฌ๊ฐ„์˜ ๊ธธ์ด์™€ ํ˜„์žฌ ์„ค์ •์€ 128๋ด‰์ž…๋‹ˆ๋‹ค. ๊ฒ€์ฆ์— ์‚ฌ์šฉํ•œ ๊ณผ๊ฑฐ ๊ตฌ๊ฐ„์„ ๋‹ค์‹œ ์ƒˆ๋กœ์šด ๋ฏธํ•™์Šต ์‹œํ—˜์ด๋ผ๊ณ  ์ทจ๊ธ‰ํ•˜์ง€ ์•Š๊ณ , ๋‹ค์Œ ์‹œํ—˜์€ ์ดํ›„์˜ ์ƒˆ ๊ตฌ๊ฐ„์—์„œ ์ง„ํ–‰ํ•ด์•ผ ํ•ฉ๋‹ˆ๋‹ค.

๊ฒฝํ—˜์—์„œ ์Šน๊ธ‰๊นŒ์ง€์˜ ํ๋ฆ„

์‹œ์žฅ ๊ด€์ฐฐ โ†’ champion์˜ ํ–‰๋™ โ†’ ๊ฐ€์ƒ ์ฒด๊ฒฐ โ†’ paper ๊ณ„์ขŒ ํ˜„๊ธˆยท๋ณด์œ ๋Ÿ‰ยท์†์ต ๊ฐฑ์‹  โ†’ ๋น„์šฉ ์ฐจ๊ฐ ์ˆœ์†์ต์œผ๋กœ ๊ฒฝํ—˜ ๊ธฐ๋ก โ†’ replay batch ํ•™์Šต โ†’ candidate snapshot ๊ฒ€์ฆ โ†’ ๊ฐœ์„  ํ™•์ธ ์‹œ champion ์Šน๊ฒฉ

ํ•™์Šต ๋ณด์ƒ๊ณผ paper ๊ณ„์ขŒ ์†์ต์€ ๊ตฌ๋ถ„ํ•ด ๊ธฐ๋กํ•ด์•ผ ํ•ฉ๋‹ˆ๋‹ค. ๊ฑฐ๋ž˜๋ณ„ ๋น„์šฉ์„ ๋ฐ˜์˜ํ•˜๊ณ , ์ข…๋ชฉ๋ณ„ ์ฒด๊ฒฐ ๊ธฐ์—ฌ๋„์™€ ๊ณ„์ขŒ ์ „์ฒด ์„ฑ๊ณผ๋ฅผ ํ•จ๊ป˜ ์‚ดํŽด candidate์˜ ๋ณ€ํ™”๊ฐ€ ์‹ค์ œ ํฌํŠธํด๋ฆฌ์˜ค ๊ฒฐ๊ณผ๋ฅผ ๊ฐœ์„ ํ–ˆ๋Š”์ง€ ํ‰๊ฐ€ํ•ฉ๋‹ˆ๋‹ค.

์‹ค์ œ ์ฃผ๋ฌธ์€ ๊ธฐ๋ณธ OFF์ž…๋‹ˆ๋‹ค. ์ด ์ €์žฅ์†Œ์˜ ์ฒดํฌํฌ์ธํŠธ๋ฅผ ๋‚ด๋ ค๋ฐ›๋Š” ๊ฒƒ๋งŒ์œผ๋กœ ์ฃผ๋ฌธ์ด ๋ฐœ์ƒํ•˜์ง€ ์•Š์Šต๋‹ˆ๋‹ค. ํ”„๋กœ์ ํŠธ ์šด์˜์—์„œ๋„ ์‚ฌ์šฉ์ž๊ฐ€ ์‹ค์ œ ์ฃผ๋ฌธ ๊ธฐ๋Šฅ์„ ๋ช…์‹œ์ ์œผ๋กœ ํ—ˆ์šฉํ•˜๊ธฐ ์ „๊นŒ์ง€ ๊ฐ€์ƒ ๋งค๋งค๋งŒ ์‚ฌ์šฉํ•œ๋‹ค๋Š” ๊ฒƒ์ด ์›์น™์ž…๋‹ˆ๋‹ค.

์ด ์ €์žฅ์†Œ์˜ ํŒŒ์ผ

ํŒŒ์ผ ์—ญํ• 
champion.pt ํ˜„์žฌ champion ๋ชจ๋ธ์˜ ๊ฐ€์ค‘์น˜ checkpoint์ž…๋‹ˆ๋‹ค.
candidate.pt ํ˜„์žฌ candidate ๋ชจ๋ธ์˜ ๊ฐ€์ค‘์น˜ checkpoint์ž…๋‹ˆ๋‹ค. ํ•™์Šต ๋˜๋Š” ๊ฒ€์ฆ ์ƒํƒœ์— ๋”ฐ๋ผ champion๊ณผ ๋‹ค๋ฅธ ๊ฐ€์ค‘์น˜๋ฅผ ๊ฐ€์งˆ ์ˆ˜ ์žˆ์Šต๋‹ˆ๋‹ค.
README.md ํ”„๋กœ์ ํŠธ ๋ชฉํ‘œ, ๋ชจ๋ธ ์„ค์ •, ์˜จ๋ผ์ธ ํ•™์Šต ๋ฐ ํ‰๊ฐ€ ๋ฐฉ์‹์„ ์„ค๋ช…ํ•ฉ๋‹ˆ๋‹ค.

๋‘ checkpoint ํŒŒ์ผ์€ ๊ฐ๊ฐ ์•ฝ 1.03GB์ž…๋‹ˆ๋‹ค. ์ €์žฅ์†Œ์—๋Š” champion๊ณผ candidate ๊ฐ ํ•˜๋‚˜๋ฅผ ๋‘๋ฉฐ, replay ๊ฒฝํ—˜์ด๋‚˜ ๊ณ„์ขŒ runtime ๋ฐ์ดํ„ฐ๋Š” ๋ชจ๋ธ checkpoint์™€ ๋ณ„๊ฐœ์˜ ์šด์˜ ์ž๋ฃŒ์ž…๋‹ˆ๋‹ค.

์‚ฌ์šฉ ์ „ ์ฃผ์˜์‚ฌํ•ญ

  • ์ด ์ €์žฅ์†Œ์—๋Š” ์ฒดํฌํฌ์ธํŠธ์™€ ์„ค๋ช…๋ฌธ์ด ์žˆ์Šต๋‹ˆ๋‹ค. ์‹œ์žฅ ๋ฐ์ดํ„ฐ ์ˆ˜์ง‘, ๊ฐ€์ƒ ๊ณ„์ขŒ, ๋น„์šฉ ๊ณ„์‚ฐ, replay, ํ•™์Šต ๋ฐ ๊ฒ€์ฆ์„ ์‹คํ–‰ํ•˜๋ ค๋ฉด ํ”„๋กœ์ ํŠธ ์†Œ์Šค์™€ ๊ทธ์— ๋งž๋Š” ์„ค์ •์ด ํ•„์š”ํ•ฉ๋‹ˆ๋‹ค.
  • ์ฒดํฌํฌ์ธํŠธ๋ฅผ ๋ถˆ๋Ÿฌ์˜ค๊ธฐ ์ „์— ์ฝ”๋“œ ๋ฒ„์ „, ๋ชจ๋ธ ์ž…๋ ฅ feature ์ˆœ์„œ์™€ ํฌ๊ธฐ, ๋ชจ๋ธ ๊ตฌ์กฐ๊ฐ€ ๋งž๋Š”์ง€ ํ™•์ธํ•˜์„ธ์š”.
  • ์ด ๋ชจ๋ธ์˜ ์„ฑ๊ณผ์™€ ์ˆ˜์ต์„ฑ์€ ๋ณด์žฅ๋˜์ง€ ์•Š์Šต๋‹ˆ๋‹ค. ๊ณผ๊ฑฐยทํ˜„์žฌ์˜ paper ๊ฒฐ๊ณผ๊ฐ€ ๋ฏธ๋ž˜ ์ˆ˜์ต์„ ๋ณด์žฅํ•˜์ง€ ์•Š์Šต๋‹ˆ๋‹ค.
  • ์‹ค์ œ ์ž๊ธˆ์œผ๋กœ ์‚ฌ์šฉํ•˜๊ธฐ ์ „์— ๋…๋ฆฝ ๊ฒ€์ฆ๊ณผ ์œ„ํ—˜ ๊ด€๋ฆฌ๊ฐ€ ํ•„์š”ํ•ฉ๋‹ˆ๋‹ค. ์ด ์ €์žฅ์†Œ๋Š” ํˆฌ์ž ์กฐ์–ธ์„ ์ œ๊ณตํ•˜์ง€ ์•Š์Šต๋‹ˆ๋‹ค.
  • ์ฒดํฌํฌ์ธํŠธ๋Š” PyTorch ํŒŒ์ผ์ž…๋‹ˆ๋‹ค. ์‹ ๋ขฐํ•  ์ˆ˜ ์žˆ๋Š” ์ถœ์ฒ˜์˜ ํŒŒ์ผ๋งŒ ์‚ฌ์šฉํ•˜๊ณ , ์•ˆ์ „ํ•˜์ง€ ์•Š์€ ํ™˜๊ฒฝ์—์„œ ์ž„์˜์˜ checkpoint๋ฅผ ๋กœ๋“œํ•˜์ง€ ๋งˆ์„ธ์š”.

โ€œDear teachers, please send your feedback to hushmind@gmail.com

Downloads last month

-

Downloads are not tracked for this model. How to track
Video Preview
loading