์•ˆ๋…•ํ•˜์„ธ์š”, ์—ฌ๋Ÿฌ๋ถ„! ์˜ค๋Š˜์€ ๊ตฌ๊ธ€์—์„œ ์ตœ๊ทผ ๋ฐœํ‘œํ•œ ํฅ๋ฏธ๋กœ์šด ์—ฐ๊ตฌ์— ๋Œ€ํ•ด ์†Œ๊ฐœํ•˜๋ ค๊ณ  ํ•ฉ๋‹ˆ๋‹ค. ๋…ผ๋ฌธ์˜ ์ œ๋ชฉ์€ "Leave No Context Behind: Efficient Infinite Context Transformers with Infini-attention"์ด๋ฉฐ, Tsendsuren Munkhdalai, Manaal Faruqui, Siddharth Gopal ๋“ฑ ๊ตฌ๊ธ€ ์†Œ์† ์—ฐ๊ตฌ์›๋“ค์— ์˜ํ•ด ์ž‘์„ฑ๋˜์—ˆ์Šต๋‹ˆ๋‹ค. ์ด ๋…ผ๋ฌธ์€ ์•„์ง ์‹ฌ์‚ฌ ์ค‘์ธ ๋‹จ๊ณ„์ด๋ฉฐ, arXiv๋ผ๋Š” ์˜จ๋ผ์ธ ๋…ผ๋ฌธ ์ €์žฅ์†Œ์— ๊ณต๊ฐœ๋˜์–ด ์žˆ์Šต๋‹ˆ๋‹ค.


์ด ์—ฐ๊ตฌ์—์„œ ์ œ์•ˆ๋œ "Infini-attention"์ด๋ผ๋Š” ์ƒˆ๋กœ์šด ๊ธฐ์ˆ ์€ ์ธ๊ณต์ง€๋Šฅ ๋ชจ๋ธ, ํŠนํžˆ Transformer ๊ธฐ๋ฐ˜ LLM(Large Language Model)์˜ ํ•œ๊ณ„๋ฅผ ๊ทน๋ณตํ•  ์ˆ˜ ์žˆ๋Š” ํš๊ธฐ์ ์ธ ๋ฐฉ๋ฒ•์œผ๋กœ ์ฃผ๋ชฉ๋ฐ›๊ณ  ์žˆ์Šต๋‹ˆ๋‹ค.


๊ธฐ์กด์˜ Transformer ๋ชจ๋ธ์€ "์–ดํ…์…˜(attention)" ๋ฉ”์ปค๋‹ˆ์ฆ˜์„ ํ†ตํ•ด ํ…์ŠคํŠธ ๋ฐ์ดํ„ฐ์˜ ๊ฐ ๋ถ€๋ถ„ ๊ฐ„ ๊ด€๊ณ„๋ฅผ ํŒŒ์•…ํ•˜๋Š”๋ฐ, ์ด ๊ณผ์ •์—์„œ ์ž…๋ ฅ ๋ฐ์ดํ„ฐ๊ฐ€ ๊ธธ์–ด์งˆ์ˆ˜๋ก ๊ณ„์‚ฐ๋Ÿ‰์ด ์—„์ฒญ๋‚˜๊ฒŒ ์ฆ๊ฐ€ํ•˜๋Š” ๋ฌธ์ œ๊ฐ€ ์žˆ์—ˆ์Šต๋‹ˆ๋‹ค. ์ด๋Š” ๊ณง ๊ธด ๊ธ€, ๊ฐ€๋ น ์†Œ์„ค์ด๋‚˜ ๋ณด๊ณ ์„œ ์ „์ฒด๋ฅผ ์ดํ•ดํ•˜๊ธฐ ์œ„ํ•ด์„œ๋Š” ๋ง‰๋Œ€ํ•œ ์ปดํ“จํŒ… ์ž์›์ด ํ•„์š”ํ•˜๋‹ค๋Š” ๊ฒƒ์„ ์˜๋ฏธํ•ฉ๋‹ˆ๋‹ค.


Infini-attention์€ ์ด๋Ÿฐ ๋ฌธ์ œ๋ฅผ ํ•ด๊ฒฐํ•˜๊ธฐ ์œ„ํ•ด ๊ณ ์•ˆ๋œ ๋ฐฉ๋ฒ•์œผ๋กœ, ์••์ถ• ๋ฉ”๋ชจ๋ฆฌ, ๋งˆ์Šคํฌ ๋กœ์ปฌ ์–ดํ…์…˜, ์žฅ๊ธฐ ์„ ํ˜• ์–ดํ…์…˜์ด๋ผ๋Š” ์„ธ ๊ฐ€์ง€ ์š”์†Œ๋ฅผ ๊ฒฐํ•ฉํ•œ ๊ฒƒ์ด ํŠน์ง•์ž…๋‹ˆ๋‹ค.


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


Infini-attention์€ ์ด ์„ธ ๊ฐ€์ง€๋ฅผ ํ•˜๋‚˜๋กœ ๋ฌถ์–ด์„œ, ์•„์ฃผ ๊ธด ํ…์ŠคํŠธ๋„ ํšจ์œจ์ ์œผ๋กœ ์ฒ˜๋ฆฌํ•  ์ˆ˜ ์žˆ๊ฒŒ ๋งŒ๋“ญ๋‹ˆ๋‹ค. ์ด๋Š” ๋งˆ์น˜ ์šฐ๋ฆฌ๊ฐ€ ๋‘๊บผ์šด ์ฑ…์„ ์š”์•ฝ ๋…ธํŠธ ํ•˜๋‚˜๋กœ ์ •๋ฆฌํ•˜๊ณ , ์ค‘์š”ํ•œ ๋ถ€๋ถ„์„ ์ฐพ์•„๋ณด๋ฉฐ, ์•ž๋’ค ๋‚ด์šฉ์„ ์—ฐ๊ฒฐํ•ด์„œ ์ฑ…์˜ ์ „์ฒด ๋‚ด์šฉ์„ ์ดํ•ดํ•˜๋Š” ๊ฒƒ๊ณผ ์œ ์‚ฌํ•ฉ๋‹ˆ๋‹ค.


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


์—ฐ๊ตฌ์ง„์€ ๋‹ค์–‘ํ•œ ์‹คํ—˜์„ ํ†ตํ•ด Infini-attention์˜ ์„ฑ๋Šฅ์„ ์ž…์ฆํ–ˆ์Šต๋‹ˆ๋‹ค. ํŠนํžˆ ๊ธด ์†Œ์„ค์„ ์š”์•ฝํ•˜๋Š” ์ž‘์—…์—์„œ๋Š” ์ตœ๊ณ  ์„ฑ๋Šฅ์„ ๋ณด์—ฌ์ฃผ์—ˆ๋Š”๋ฐ์š”, ๊ธฐ์กด์— 5,000๋‹จ์–ด ๋ถ„๋Ÿ‰์˜ ๊ธ€์„ ์ดํ•ดํ•˜๋˜ ๋ชจ๋ธ์— Infini-attention์„ ์ ์šฉํ–ˆ๋”๋‹ˆ 100๋งŒ ๋‹จ์–ด๊ฐ€ ๋„˜๋Š” ๊ธ€๋„ ๋ฌธ์ œ์—†์ด ์ดํ•ดํ•  ์ˆ˜ ์žˆ๊ฒŒ ๋˜์—ˆ๋‹ค๊ณ  ํ•ฉ๋‹ˆ๋‹ค. ์ด๋Š” ์ •๋ง ๋†€๋ผ์šด ์„ฑ๊ณผ๋ผ๊ณ  ํ•  ์ˆ˜ ์žˆ์Šต๋‹ˆ๋‹ค.


Infini-attention์˜ ๋˜ ๋‹ค๋ฅธ ์žฅ์ ์€ ๊ธฐ์กด ์ธ๊ณต์ง€๋Šฅ ๋ชจ๋ธ์— ์‰ฝ๊ฒŒ ์ ์šฉํ•  ์ˆ˜ ์žˆ๋‹ค๋Š” ์ ์ž…๋‹ˆ๋‹ค. ํ˜„์žฌ ์‚ฌ์šฉ ์ค‘์ธ ์ธ๊ณต์ง€๋Šฅ ์‹œ์Šคํ…œ์ด ์žˆ๋‹ค๋ฉด Infini-attention์œผ๋กœ ์—…๊ทธ๋ ˆ์ด๋“œํ•˜๋Š” ๊ฒƒ๋งŒ์œผ๋กœ๋„ ๊ธด ๊ธ€ ์ดํ•ด ๋Šฅ๋ ฅ์„ ํฌ๊ฒŒ ํ–ฅ์ƒ์‹œํ‚ฌ ์ˆ˜ ์žˆ์Šต๋‹ˆ๋‹ค.


์ด๋Ÿฐ ํŠน์ง• ๋•๋ถ„์— Infini-attention์€ ๊ธด ๋ฌธ์„œ ์š”์•ฝ, ์ฑ… ์ „์ฒด์— ๋Œ€ํ•œ ์งˆ์˜์‘๋‹ต, ์žฅ๋ฌธ ๋ฒˆ์—ญ ๋“ฑ ๋‹ค์–‘ํ•œ ๋ถ„์•ผ์—์„œ ํ™œ์šฉ๋  ๊ฒƒ์œผ๋กœ ๊ธฐ๋Œ€๋ฉ๋‹ˆ๋‹ค. ์ง€๊ธˆ๊นŒ์ง€๋Š” ๊ธฐ์ˆ ์  ํ•œ๊ณ„๋กœ ์‹œ๋„ํ•˜๊ธฐ ์–ด๋ ค์› ๋˜ ์ผ๋“ค์ด ๊ฐ€๋Šฅํ•ด์ง€๋Š” ๊ฒƒ์ด์ฃ .


๋งŽ์€ ์ „๋ฌธ๊ฐ€๋“ค์€ Infini-attention์ด ์ธ๊ณต์ง€๋Šฅ ๊ธฐ์ˆ ์˜ ์ƒˆ๋กœ์šด ํŒจ๋Ÿฌ๋‹ค์ž„์„ ์—ด ๊ฒƒ์ด๋ผ๊ณ  ํ‰๊ฐ€ํ•˜๊ณ  ์žˆ์Šต๋‹ˆ๋‹ค. ์šฐ๋ฆฌ ์‚ถ์— ์ธ๊ณต์ง€๋Šฅ์ด ๋”์šฑ ๊นŠ์ˆ™์ด ์Šค๋ฉฐ๋“œ๋Š” ์‹œ๋Œ€์— Infini-attention๊ณผ ๊ฐ™์€ ํ˜์‹  ๊ธฐ์ˆ ์ด ๊ทธ ๋ณ€ํ™”๋ฅผ ์ด๋Œ์–ด๊ฐˆ ๊ฒƒ์œผ๋กœ ๋ณด์ž…๋‹ˆ๋‹ค. ์•ž์œผ๋กœ ์ธ๊ณต์ง€๋Šฅ์ด ์šฐ๋ฆฌ์—๊ฒŒ ์–ด๋–ค ๋†€๋ผ์šด ์ผ๋“ค์„ ๋ณด์—ฌ์ค„์ง€ ์ •๋ง ๊ธฐ๋Œ€๊ฐ€ ๋ฉ๋‹ˆ๋‹ค!


๊ตฌ๊ธ€์—์„œ ๋ฐœํ‘œํ•œ ์ด ํš๊ธฐ์ ์ธ ์—ฐ๊ตฌ๊ฐ€ ์ธ๊ณต์ง€๋Šฅ ๋ถ„์•ผ์— ๋ฏธ์น  ์˜ํ–ฅ์ด ๋งค์šฐ ํด ๊ฒƒ์œผ๋กœ ์˜ˆ์ƒ๋˜๋Š”๋ฐ์š”, Infini-attention์ด๋ผ๋Š” ์ƒˆ๋กœ์šด ๊ธฐ์ˆ ์„ ํ†ตํ•ด LLM์˜ ์„ฑ๋Šฅ๊ณผ ํšจ์œจ์ด ํฌ๊ฒŒ ๊ฐœ์„ ๋  ๊ฒƒ์œผ๋กœ ๋ณด์ž…๋‹ˆ๋‹ค. ์ด๋Š” ์ž์—ฐ์–ด ์ฒ˜๋ฆฌ ๋ถ„์•ผ ์ „๋ฐ˜์— ๊ฑธ์ณ ํฐ ๋ณ€ํ™”๋ฅผ ๊ฐ€์ ธ์˜ฌ ์ˆ˜ ์žˆ์œผ๋ฉฐ, ์šฐ๋ฆฌ ์ผ์ƒ์ƒํ™œ์—๋„ ๊ธ์ •์ ์ธ ์˜ํ–ฅ์„ ๋ฏธ์น  ๊ฒƒ์œผ๋กœ ๊ธฐ๋Œ€๋ฉ๋‹ˆ๋‹ค.


๊ตฌ๊ธ€์˜ ์ด๋ฒˆ ์—ฐ๊ตฌ๊ฐ€ ๊ฐ€์ ธ์˜ฌ ๋ณ€ํ™”์™€ ๊ทธ ํŒŒ๊ธ‰ํšจ๊ณผ์— ๋Œ€ํ•ด ์•ž์œผ๋กœ๋„ ์ง€์†์ ์ธ ๊ด€์‹ฌ์„ ๊ฐ€์ ธ๋ณด๋ฉด ์ข‹๊ฒ ์Šต๋‹ˆ๋‹ค. ์ธ๊ณต์ง€๋Šฅ ๊ธฐ์ˆ ์˜ ๋ฐœ์ „์ด ์šฐ๋ฆฌ ์‚ถ์„ ์–ด๋–ป๊ฒŒ ๋ฐ”๊ฟ”๋‚˜๊ฐˆ์ง€, ํ•จ๊ป˜ ์ง€์ผœ๋ณด๋Š” ๊ฒƒ๋„ ํฅ๋ฏธ๋กœ์šธ ๊ฒƒ ๊ฐ™๋„ค์š”!