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์ ํ์ ์ดํดํ๊ณ ํจ๊ณผ์ ์ผ๋ก ๊ฐ๋ฅํ๊ฒ ํ๋ ๋ฐฉ๋ฒ์ ๊ฐ์ ํ๋ ๋ฐ ๋์์ด ๋ ๊ฒ์ด๋ผ๊ณ ๋ฏฟ์ต๋๋ค.
์ํด... ์ฌํด์์ gpt4ํฐ๋ณด๋ ๋ชป๋๊ธฐ๋ฉด ๋๋จ ๊ทธ๋ฅ ๋ฏฟ๊ฑฐ๋ค ์ด์๋ผ๋ค์ 1๋ ๋ด๋ด gpt4 ๋ชป๋์ผ๋ฉด์ ์๋ฅด์ฟค์ง๊ฑฐ๋ฆฌ๋ ํ๊ณ ์๋
์ดํ.. ๋ฐ๋๋ ๋ฅ๋ง์ธ๋ ์์์ด ์๋๋ค - dc App
๋ง์ค์ํ์ง ๋ง๊ฑฐ๋ผ - dc App
๊ทธ๋ ์ ๋ฐ ์ ์ฆํด์ค๋ด๋ผ ๋ญ๊ฐ๋ฌ๋ gpt4ํฐ๋ณด๋ณด๋ค ์ฑ๋ฅ ๋ค์ณ์ง๋ฉด์ ์ ๋ฐ๊ตฌ๋ก ์ ๋งํธ์์ผ๋ฉด ํ์ฌ๋น์ค๋ ์ฌ์ธต๊ฐํ๋ก ์ก๋คํ ํ์คํฌ๋ ๋ง์ง์ค์์ง llm์ดํด๋ ใ ๋์๋๊ฑฐ๋ฐ์ ์๋๋๊ฑฐ์