https://github.com/microsoft/unilm/tree/master/bitnet


๋งˆ์ดํฌ๋กœ์†Œํ”„ํŠธ ์–ธ์–ด๋ชจ๋ธ ๊ด€๋ จ ๊นƒํ—ˆ๋ธŒ์— ์˜ค๋Š˜ ์ถ”๊ฐ€ ์ž๋ฃŒ ์˜ฌ๋ผ์˜ด


The Era of 1-bit LLMs: Training Tips, Code and FAQ


๊ทธ๋ฆฌ ๊ธธ์ง€๋Š” ์•Š์•„์„œ ์ „์ฒด์ ์œผ๋กœ ๋ณด๊ณ  ์™”์Œ

์ €์ž๋“ค๋„ ํšŒ์˜์ ์ธ ๋ฐ˜์‘๋“ค์ด ๊ฝค ์žˆ์–ด์„œ ๊ทธ๋žฌ๋Š”์ง€,

๋…ผ๋ฌธ์— 'Believing is seeing.' ์ด๋ผ๊ณ  ๋ฐ•์•„ ๋†“์Œ

(์ง์ ‘ ๋ณด๋ฉด ์•ˆ ๋ฏฟ์„ ์ˆ˜ ์—†์„ ๊ฒƒ์ด๋‹ค.)


์–ด๋–ป๊ฒŒ ํ•™์Šตํ–ˆ๋Š”์ง€์™€ ํ•™์Šต ๊ณผ์ •์— ๋Œ€ํ•œ ๋ณด๋‹ค ๊ตฌ์ฒด์ ์ธ ์ž๋ฃŒ, ํ•˜์ดํผํŒŒ๋ผ๋ฏธํ„ฐ ๋“ฑ ๊ณต๊ฐœํ•˜๊ณ 

์–ด๋–ป๊ฒŒ ํŒŒ์ดํ† ์น˜๋กœ ๊ตฌํ˜„ํ•˜๋Š”์ง€๋„ ๊ณต๊ฐœํ•จ

์ด๋ฒˆ ๋ณด๊ณ ์„œ์—์„œ ๊ณต๊ฐœํ•œ ๊ทธ๋ž˜ํ”„

์ €์ž๋“ค์€ 1.58 BitNet์ด S์žํ˜• ์†์‹ค ๊ณก์„ ์„ ๋ณด์—ฌ์ค€๋‹ค๊ณ  ๋งํ•จ

๊ทธ๋ฆฌ๊ณ  ์ด ๊ทธ๋ž˜ํ”„๊ฐ€ ์ด ๋ณด๊ณ ์„œ์˜ ํ•ต์‹ฌ์ด๋ผ๊ณ  ๋ณผ ์ˆ˜๋„ ์žˆ๋Š”๋ฐ,

๋ชจ๋ธ ํฌ๊ธฐ๊ฐ€ ์ปค์ง€๋ฉด ์ปค์งˆ ์ˆ˜๋ก ์ „์ฒด ์ •๋ฐ€๋„ ๋ชจ๋ธ๊ณผ ํ•™์Šต ์†์‹ค ์ฐจ์ด๊ฐ€ ์ค„์–ด๋“œ๋Š” ๊ฒฝํ–ฅ์„ ๋ณด์—ฌ์คŒ




์•„๋ž˜๋Š” ์ €์ž๋“ค์ด ์จ๋†“์€ FAQ


1. ์‚ผํ•ญ {-1, 0, 1} ๋ง๊ณ  ๋”ด๊ฑด ์™œ ์•ˆ์”€?


{-1, 1} : ์›๋ž˜ BitNet b1(์ €์ž๋“ค์˜ ์ด์ „ ๋…ผ๋ฌธ)์—์„œ ๊ตฌํ˜„์ด์—ˆ๋Š”๋ฐ ์„ฑ๋Šฅ์ด ์‚ผํ•ญ๋ณด๋‹ค ๋”ธ๋ ธ์Œ

{0, 1} : ์ตœ์ ํ™”๊ฐ€ ๋งค์šฐ ๋ถˆ์•ˆ์ •ํ•จ

{-2, -1, 0, 1} or {-2, -1, 0, 1, 2}๊ฐ™์ด ์ถ”๊ฐ€์ ์ธ ๋น„ํŠธ ์‚ฌ์šฉ : ์‚ผํ•ญ {-1, 0, 1}ํ•˜๊ณ  ๋ณ„ ์ฐจ์ด ์—†์–ด์„œ ์•ˆ์”€


2. ํ›ˆ๋ จ๋„ ๋นจ๋ผ์ง?


ํ˜„์žฌ ๊ตฌํ˜„์€ ์—ฌ์ „ํžˆ FP16/BF16์— ์žˆ์–ด์„œ ์‹คํ—˜์—์„œ ์‹ค์ œ ์†๋„ ํ–ฅ์ƒ์€ ์—†์Œ

ํ•˜์ง€๋งŒ ๋Œ€ํ˜• ๋ชจ๋ธ์˜ ๊ฒฝ์šฐ ๊ฐ€์†ํ™”ํ•  ์ˆ˜ ์žˆ๋Š” ์ƒ๋‹นํ•œ ๊ธฐํšŒ๊ฐ€ ์žˆ์Œ


3. BitNet์ด ๋” ํฐ ๋ชจ๋ธ์—๋„ ์ž‘๋™ํ•จ?


"The Era of 1-bit LLMs" ๋…ผ๋ฌธ์—์„œ ๊ณต๊ฐœํ•œ ๊ฒƒ์ฒ˜๋Ÿผ ์™„์ „ ์ •๋ฐ€๋„ LLM๊ณผ BitNet 1.58 ์‚ฌ์ด์˜ ๊ฐ„๊ฒฉ์ด ๋ชจ๋ธ ํฌ๊ธฐ๊ฐ€ ์ปค์ง์— ๋”ฐ๋ผ ์ž‘์•„์ง€๋Š” ๊ฒƒ์„ ๋ณด์—ฌ์ฃผ๋Š” ๋ช…ํ™•ํ•œ ์ถ”์„ธ๊ฐ€ ์žˆ์Œ


์ด๋Š” BitNet์ด ๋” ํฐ ๋ชจ๋ธ์— ๋”์šฑ ํšจ๊ณผ์ ์ด๋ผ๋Š” ๊ฒƒ์„ ์˜๋ฏธํ•จ


์˜คํžˆ๋ ค BitNet์€ ๋” ํฐ ๋ชจ๋ธ ํฌ๊ธฐ์—์„œ ๋” ๋‚˜์€ ์„ฑ๋Šฅ์„ ๋ฐœํœ˜ํ•จ

1.58 ๋น„ํŠธ ๋ชจ๋ธ์€ ๋” ๋‚˜์€ ์ผ๋ฐ˜ํ™” ๊ธฐ๋Šฅ์„ ์ œ๊ณตํ•˜๊ณ  ๊ณผ์ ํ•ฉ์ด ๋œ ๋ฐœ์ƒํ•  ์ˆ˜ ์žˆ์Œ



๊ทธ๋ฆฌ๊ณ  ์•„๋ž˜์— ํŒŒ์ดํ† ์น˜ ๊ธฐ๋ฐ˜ ๊ตฌํ˜„ ์ฝ”๋“œ๋„ ๊ณต๊ฐœํ–ˆ๋Š”๋ฐ, ๋˜๊ฒŒ ๊ฐ„๋‹จํ•ด์„œ ๊ธˆ๋ฐฉ ์‚ฌ๋žŒ๋“ค์ด ์‹คํ—˜ํ•ด๋ณผ ๋“ฏ


์ด๋ฒˆ์—๋„ ์—ฌ์ „ํžˆ ์ €์ž๋“ค์˜ ์ฃผ์žฅ์ด๊ธฐ๋Š” ํ•˜๋‚˜

์ €์ž๋“ค์€ ๋ฏฟ์Œ์ด ๊ต‰์žฅํžˆ ๊ตณ๊ฑดํ•ด๋ณด์ธ๋‹ค.


๋—