https://arxiv.org/pdf/2311.00871.pdf


Pretraining Data Mixtures Enable Narrow Model Selection Capabilities in Transformer Models Steve Yadlowsky, Lyric Doshi, Nilesh Tripuraneni {yadlowsky, lyric, nileshtrip}@google.com Google DeepMind November 3, 2023


์ผ์ฃผ์ผ ์ „ ๋…ผ๋ฌธ


์ดˆ๋ก


ํŠธ๋žœ์Šคํฌ๋จธ ๋ชจ๋ธ, ํŠนํžˆ ๋Œ€๊ทœ๋ชจ ์–ธ์–ด ๋ชจ๋ธ(LLM)์€ ๋ณธ์งˆ์ ์ธ ๋ชจ๋ธ ํ›ˆ๋ จ ์—†์ด๋„ ๋ณธ ์  ์—†๋Š” ์ž…๋ ฅ-์ถœ๋ ฅ ์˜ˆ์‹œ๊ฐ€ ์ œ์‹œ๋  ๋•Œ ์ƒˆ๋กœ์šด ์ž‘์—…์„ ์ˆ˜ํ–‰ํ•˜๋Š” ์ธ-์ปจํ…์ŠคํŠธ ํ•™์Šต(ICL)์„ ์ˆ˜ํ–‰ํ•  ์ˆ˜ ์žˆ๋Š” ๋›ฐ์–ด๋‚œ ๋Šฅ๋ ฅ์„ ๊ฐ€์ง€๊ณ  ์žˆ์Šต๋‹ˆ๋‹ค. ์ด ์—ฐ๊ตฌ์—์„œ ์šฐ๋ฆฌ๋Š” ํŠธ๋žœ์Šคํฌ๋จธ๊ฐ€ ๋‹ค์–‘ํ•œ ๊ณ ์œ ํ•œ ์ž‘์—… ํŒจ๋ฐ€๋ฆฌ๋กœ ๊ตฌ์„ฑ๋œ ์‚ฌ์ „ ํ›ˆ๋ จ ๋ฐ์ดํ„ฐ ํ˜ผํ•ฉ๋ฌผ ์‚ฌ์ด๋ฅผ ํšจ๊ณผ์ ์œผ๋กœ ์—ฐ๊ฒฐํ•˜์—ฌ ์‚ฌ์ „ ํ›ˆ๋ จ ๋ถ„ํฌ ๋‚ด๋ถ€์™€ ์™ธ๋ถ€์— ์žˆ๋Š” ์ƒˆ๋กœ์šด ์ž‘์—…์„ ์ธ-์ปจํ…์ŠคํŠธ์—์„œ ์‹๋ณ„ํ•˜๊ณ  ํ•™์Šตํ•  ์ˆ˜ ์žˆ๋Š”์ง€๋ฅผ ์—ฐ๊ตฌํ•ฉ๋‹ˆ๋‹ค. ์ด์ „์˜ ์ž‘์—…์„ ๊ธฐ๋ฐ˜์œผ๋กœ, ์šฐ๋ฆฌ๋Š” ์ž์—ฐ์–ด๊ฐ€ ์•„๋‹Œ (x, f(x)) ์Œ์˜ ์ˆœ์„œ์— ๋Œ€ํ•ด ํ›ˆ๋ จ๋œ ํŠธ๋žœ์Šคํฌ๋จธ ๋ชจ๋ธ์„ ์—ฐ๊ตฌํ•˜๋Š” ์ œ์–ด๋œ ํ™˜๊ฒฝ์—์„œ ์ด ์งˆ๋ฌธ์„ ์กฐ์‚ฌํ•ฉ๋‹ˆ๋‹ค. ์šฐ๋ฆฌ์˜ ๊ฒฝํ—˜์  ๊ฒฐ๊ณผ๋Š” ํŠธ๋žœ์Šคํฌ๋จธ๊ฐ€ ์‚ฌ์ „ ํ›ˆ๋ จ ๋ฐ์ดํ„ฐ์—์„œ ์ž˜ ํ‘œํ˜„๋œ ์ž‘์—… ํŒจ๋ฐ€๋ฆฌ๋ฅผ ์ธ-์ปจํ…์ŠคํŠธ์—์„œ ๋จผ์ € ์‹๋ณ„ํ•˜๊ณ  ๊ทธ ์•ˆ์—์„œ ํ•™์Šตํ•  ์ˆ˜ ์žˆ๋Š” ๋Šฅ๋ ฅ์— ์žˆ์–ด์„œ ๊ฑฐ์˜ ์ตœ์ ์˜ ๋น„๊ฐ๋… ๋ชจ๋ธ ์„ ํƒ ๋Šฅ๋ ฅ์„ ๋ณด์—ฌ์ค€๋‹ค๋Š” ๊ฒƒ์„ ๋ณด์—ฌ์ค๋‹ˆ๋‹ค. ๊ทธ๋Ÿฌ๋‚˜ ์‚ฌ์ „ ํ›ˆ๋ จ ๋ฐ์ดํ„ฐ์˜ ๋„๋ฉ”์ธ ์™ธ๋ถ€์— ์žˆ๋Š” ์ž‘์—…์ด๋‚˜ ํ•จ์ˆ˜๊ฐ€ ์ œ์‹œ๋  ๋•Œ ์šฐ๋ฆฌ๋Š” ํŠธ๋žœ์Šคํฌ๋จธ์˜ ๋‹ค์–‘ํ•œ ์‹คํŒจ ๋ชจ๋“œ์™€ ๊ฐ„๋‹จํ•œ ์™ธ์‚ฝ ์ž‘์—…์— ๋Œ€ํ•œ ์ผ๋ฐ˜ํ™”์˜ ์ €ํ•˜๋ฅผ ๋ณด์—ฌ์ค๋‹ˆ๋‹ค. ํ•จ๊ป˜ ์šฐ๋ฆฌ์˜ ๊ฒฐ๊ณผ๋Š” ๋†’์€ ์šฉ๋Ÿ‰์˜ ์ˆœ์„œ ๋ชจ๋ธ์˜ ์ธ์ƒ์ ์ธ ICL ๋Šฅ๋ ฅ์ด ๊ทผ๋ณธ์ ์ธ ์ผ๋ฐ˜ํ™” ๋Šฅ๋ ฅ์„ ์ฐฝ์ถœํ•˜๋Š” ๊ท€๋‚ฉ์  ํŽธํ–ฅ๋ณด๋‹ค๋Š” ๊ทธ๋“ค์˜ ์‚ฌ์ „ ํ›ˆ๋ จ ๋ฐ์ดํ„ฐ ํ˜ผํ•ฉ๋ฌผ์˜ ๋ฒ”์œ„์— ๋” ๋ฐ€์ ‘ํ•˜๊ฒŒ ์—ฐ๊ฒฐ๋  ์ˆ˜ ์žˆ์Œ์„ ๊ฐ•์กฐํ•ฉ๋‹ˆ๋‹ค.





๊ฒฐ๋ก 


์šฐ๋ฆฌ๋Š” ์‚ฌ์ „ ํ›ˆ๋ จ๋œ ํŠธ๋žœ์Šคํฌ๋จธ๊ฐ€ ์‚ฌ์ „ ํ›ˆ๋ จ ๋ฐ์ดํ„ฐ ๋ถ„ํฌ์˜ ์ง€์› ๋‚ด๋ถ€์™€ ์™ธ๋ถ€์— ์žˆ๋Š” ํ•จ์ˆ˜ ํด๋ž˜์Šค๋ฅผ ์ธ-์ปจํ…์ŠคํŠธ ํ•™์Šตํ•  ์ˆ˜ ์žˆ๋Š” ๋Šฅ๋ ฅ์— ์‚ฌ์ „ ํ›ˆ๋ จ ๋ฐ์ดํ„ฐ ๊ตฌ์„ฑ์ด ๋ฏธ์น˜๋Š” ์—ญํ• ์„ ๊ฒฝํ—˜์ ์œผ๋กœ ํƒ๊ตฌํ–ˆ์Šต๋‹ˆ๋‹ค. ์‚ฌ์ „ ํ›ˆ๋ จ ํ˜ผํ•ฉ๋ฌผ์— ์ž˜ ํ‘œํ˜„๋œ ์ž‘์—… ํŒจ๋ฐ€๋ฆฌ๋‚˜ ํ•จ์ˆ˜ ํด๋ž˜์Šค์˜ ๊ฒฝ์šฐ, ์ธ-์ปจํ…์ŠคํŠธ ํ•™์Šต์— ์‚ฌ์šฉํ•  ์ ์ ˆํ•œ ํ•จ์ˆ˜ ํด๋ž˜์Šค๋ฅผ ์„ ํƒํ•˜๋Š” ๋น„์šฉ์ด ๊ฑฐ์˜ ์ œ๋กœ์— ๊ฐ€๊น๋‹ค๋Š” ๊ฒƒ์„ ์šฐ๋ฆฌ๋Š” ๊ฒฝํ—˜์ ์œผ๋กœ ๋ณด์—ฌ์ฃผ์—ˆ์Šต๋‹ˆ๋‹ค. ๋‹ค์Œ์œผ๋กœ ์šฐ๋ฆฌ๋Š” ๋‘ ๊ฐ€์ง€ ์‹œ๋‚˜๋ฆฌ์˜ค์—์„œ์˜ ์ผ๋ฐ˜ํ™” ๊ฐ€๋Šฅ์„ฑ์„ ํƒ๊ตฌํ–ˆ์Šต๋‹ˆ๋‹ค: (1) ์‚ฌ์ „ ํ›ˆ๋ จ๋œ ํŠธ๋žœ์Šคํฌ๋จธ๋“ค์ด ์‚ฌ์ „ ํ›ˆ๋ จ ํ•จ์ˆ˜ ํด๋ž˜์Šค์—์„œ ๊ฐ€์ ธ์˜จ ํ•จ์ˆ˜๋“ค์˜ ๋ณผ๋ก ์กฐํ•ฉ์„ ์˜ˆ์ธกํ•˜๋Š” ๋ฐ ์–ด๋ ค์›€์„ ๊ฒช๋Š”๋‹ค๋Š” ๊ฒƒ์„ ๋ฐœ๊ฒฌํ–ˆ๊ณ , (2) ํŠธ๋žœ์Šคํฌ๋จธ๋“ค์ด ํ•จ์ˆ˜ ํด๋ž˜์Šค ๊ณต๊ฐ„์˜ ๋” ๋“œ๋ฌธ ๋ถ€๋ถ„์—์„œ ํšจ๊ณผ์ ์œผ๋กœ ์ผ๋ฐ˜ํ™”ํ•  ์ˆ˜ ์žˆ์ง€๋งŒ ์ž‘์—…์ด ๋ถ„ํฌ์—์„œ ๋ฒ—์–ด๋‚จ์— ๋”ฐ๋ผ ๋ถ•๊ดด๋˜๋Š” ๊ฒƒ์„ ๊ด€์ฐฐํ–ˆ์Šต๋‹ˆ๋‹ค. ์ค‘์š”ํ•œ ์งˆ๋ฌธ์€ ์šฐ๋ฆฌ๊ฐ€ ์—ฌ๊ธฐ์„œ ํ•˜๋Š” ๊ด€์ฐฐ์ด ํ† ํฐํ™”๋œ ๋ชจ๋ธ๊ณผ ์ž์—ฐ์–ด๋กœ ํ‘œํ˜„๋œ ์งˆ๋ฌธ์— ์–ด๋–ป๊ฒŒ ์ ์šฉ๋˜๋Š”์ง€ ์ดํ•ดํ•˜๋Š” ๊ฒƒ์ž…๋‹ˆ๋‹ค. ์šฐ๋ฆฌ๋Š” ์„น์…˜ 4์—์„œ ์ œ์‹œ๋œ ์ผ์ฐจ์› ์˜ˆ์‹œ๋ฅผ ํ›ˆ๋ จํ•˜๊ธฐ ์œ„ํ•ด ์Šค์นผ๋ผ ๊ฐ’์„ ๋ฒ„ํ‚ท์œผ๋กœ ๋ถ„๋ฅ˜ํ•˜๊ณ , ๋ฒ„ํ‚ท ์ธ๋ฑ์Šค๋ฅผ ํŠธ๋žœ์Šคํฌ๋จธ ๊ธฐ๋ฐ˜ ์–ธ์–ด ๋ชจ๋ธ ์ž…๋ ฅ์„ ์œ„ํ•œ ํ† ํฐ์œผ๋กœ ์ทจ๊ธ‰ํ•˜๋Š” ์‹คํ—˜์„ ์‹œ๋„ํ–ˆ์Šต๋‹ˆ๋‹ค. ์šฐ๋ฆฌ๋Š” ์ด ๋ชจ๋ธ์„ 5M ์—ํฌํฌ ๋™์•ˆ ์ผ๋ฐ˜์ ์œผ๋กœ ์–ธ์–ด ๋ชจ๋ธ์—์„œ ์‚ฌ์šฉ๋˜๋Š” ํฌ๋กœ์Šค ์—”ํŠธ๋กœํ”ผ ์†์‹ค๋กœ ํ›ˆ๋ จํ–ˆ์ง€๋งŒ ์†์‹ค์„ ์ƒ๋‹นํžˆ ๊ฐ์†Œ์‹œํ‚ค์ง€ ๋ชปํ–ˆ์Šต๋‹ˆ๋‹ค. ์ด๋Ÿฌํ•œ ๋ชจ๋ธ์„ ํ›ˆ๋ จํ•˜๋Š” ๋ฐ ์žˆ์–ด ๋„์ „๊ณผ์ œ๋ฅผ ์ดํ•ดํ•˜๊ณ , ์ด๋Ÿฌํ•œ ๊ตฌ์„ฑ์ด ๋ชจ๋ธ ์„ ํƒ์ด๋‚˜ ๋ถ„ํฌ ์™ธ ์ผ๋ฐ˜ํ™” ํŠน์„ฑ์ด ๋‹ค๋ฅธ์ง€ ํ‰๊ฐ€ํ•˜๋Š” ๊ฒƒ์€ ์ค‘์š”ํ•œ ๋ฏธ๋ž˜์˜ ์ž‘์—…์ž…๋‹ˆ๋‹ค. ์ž์—ฐ์–ด ์„ค์ •์—์„œ ์šฐ๋ฆฌ์˜ ๊ฒฝ์šฐ์ฒ˜๋Ÿผ ์ ์ ˆํ•˜๊ฒŒ ์ž‘์—… ํŒจ๋ฐ€๋ฆฌ(ํ•จ์ˆ˜ ํด๋ž˜์Šค)๋ฅผ ์ •์˜ํ•˜๋Š” ๋ฐฉ๋ฒ•, ์ •ํ™•ํ•œ ์‚ฌ์ „ ํ›ˆ๋ จ ํ˜ผํ•ฉ๋ฌผ, ๊ทธ๋ฆฌ๊ณ  ๋ณผ๋ก ์กฐํ•ฉ์˜ ์ง๊ด€์ ์ธ ๊ฐœ๋…์€ ๋ชจ๋‘ ๋œ ๋ช…ํ™•ํ•ฉ๋‹ˆ๋‹ค. ์—ฌ๊ธฐ์„œ ์ œ์‹œ๋œ ๊ฐœ๋…๊ณผ ์–ธ์–ด ๋ชจ๋ธ๋ง ์‚ฌ์ด์˜ ๊ฐ„๊ทน์„ ๋ฉ”์šฐ๋Š” ๊ฒƒ์ด ICL์˜ ํž˜์„ ์ดํ•ดํ•˜๊ณ  ํšจ๊ณผ์ ์œผ๋กœ ๊ฐ€๋Šฅํ•˜๊ฒŒ ํ•˜๋Š” ๋ฐฉ๋ฒ•์„ ๊ฐœ์„ ํ•˜๋Š” ๋ฐ ๋„์›€์ด ๋  ๊ฒƒ์ด๋ผ๊ณ  ๋ฏฟ์Šต๋‹ˆ๋‹ค.