https://syncedreview.com/2022/03/18/google-idsias-block-recurrent-transformer-dramatically-outperforms-transformers-over-very-long-sequences/

Google ๋ฐ IDSIA์˜ ๋ธ”๋ก ์ˆœํ™˜ ๋ณ€์••๊ธฐ๋Š” ๋งค์šฐ ๊ธด ์‹œํ€€์Šค์— ๊ฑธ์ณ ๋ณ€์••๊ธฐ๋ณด๋‹ค ๊ทน์ ์œผ๋กœ ์„ฑ๋Šฅ์ด ๋›ฐ์–ด๋‚ฉ๋‹ˆ๋‹ค.

Google Research์™€ Swiss AI Lab IDSIA์˜ ํŒ€์€ ๊ธฐ์กด์˜ ๋ณ€ํ™˜๊ธฐ ๋ ˆ์ด์–ด์™€ ๊ณ„์‚ฐ ์‹œ๊ฐ„๊ณผ ๋งค๊ฐœ๋ณ€์ˆ˜ ์ˆ˜ ๋น„์šฉ์ด ๋™์ผํ•˜์ง€๋งŒ ์–ธ์–ด ๋ชจ๋ธ๋ง ์ž‘์—…์˜ ๋ณต์žก์„ฑ ๊ฐœ์„ ์„ ๋‹ฌ์„ฑํ•˜๋Š” ์ƒˆ๋กœ์šด ๊ธด ์‹œํ€€์Šค ์ฒ˜๋ฆฌ ์ ‘๊ทผ ๋ฐฉ์‹์ธ Block-Recurrent Transformer๋ฅผ ์ œ์•ˆํ•ฉ๋‹ˆ๋‹ค. ๋งค์šฐ ๊ธด ์‹œํ€€์Šค.



์ž์—ฐ์–ด ์ฒ˜๋ฆฌ(NLP) ๋ฐ ๊ธฐํƒ€ AI ์—ฐ๊ตฌ ๋ถ„์•ผ์—์„œ ํŠธ๋žœ์Šคํฌ๋จธ ์•„ํ‚คํ…์ฒ˜์˜ ์ธ๊ธฐ๊ฐ€ ๋†’์•„์ง€๋Š” ๊ฒƒ์€ ๊ธด ์ž…๋ ฅ ์‹œํ€€์Šค๋ฅผ ์ฒ˜๋ฆฌํ•  ๋•Œ ๋›ฐ์–ด๋‚œ ํ‘œํ˜„ ๋Šฅ๋ ฅ์— ํฌ๊ฒŒ ๊ธฐ์ธํ•ฉ๋‹ˆ๋‹ค. ๋ณ€ํ™˜๊ธฐ ๋ฐฐ์น˜๋ฅผ ์ œํ•œํ•˜๋Š” ์ฃผ์š” ๋‹จ์ ์€ self-attention ๋ฉ”์ปค๋‹ˆ์ฆ˜์˜ ๊ณ„์‚ฐ ๋ณต์žก์„ฑ์ด ์‹œํ€€์Šค์˜ ํฌ๊ธฐ์— ๋Œ€ํ•ด 2์ฐจ์ ์ด๋ผ๋Š” ๊ฒƒ์ž…๋‹ˆ๋‹ค. ๋ณ€ํ™˜๊ธฐ์˜ ์ฃผ์˜ ์ฐฝ ํฌ๊ธฐ๋Š” ์‰ฝ๊ฒŒ ํ™•์žฅํ•  ์ˆ˜ ์—†๊ธฐ ๋•Œ๋ฌธ์— ์ฑ…์ด๋‚˜ ์†Œ์Šค ์ฝ”๋“œ ์ €์žฅ์†Œ์™€ ๊ฐ™์ด ๋งค์šฐ ๊ธด ๋ฌธ์„œ๋ฅผ ์ฒ˜๋ฆฌํ•  ๋•Œ ์ปดํ“จํŒ… ๋น„์šฉ์ด ์—„์ฒญ๋‚˜๊ฒŒ ๋น„์Œ‰๋‹ˆ๋‹ค.


์ƒˆ๋กœ์šด ๋…ผ๋ฌธ Block-Recurrent Transformers ์—์„œ Google Research์™€ Swiss AI Lab IDSIA ํŒ€์€ ๊ธฐ์กด ๋ณ€์••๊ธฐ ๋ ˆ์ด์–ด์™€ ๊ณ„์‚ฐ ์‹œ๊ฐ„๊ณผ ๋งค๊ฐœ๋ณ€์ˆ˜ ์ˆ˜ ๋น„์šฉ์ด ๋™์ผํ•˜์ง€๋งŒ ์ƒ๋‹นํ•œ ๋ณต์žก์„ฑ ๊ฐœ์„ . ์—ฐ๊ตฌ์›๋“ค์€ ์ด ๋ฐฉ๋ฒ•์„ "๊ธด ์‹œํ€€์Šค์—์„œ ์–ธ์–ด ๋ชจ๋ธ๋ง ๋ณต์žก์„ฑ์„ ๊ฐœ์„ ํ•˜๋Š” ์ €๋ ดํ•˜๊ณ  ์œ ์พŒํ•œ ๋ฐฉ๋ฒ•"์œผ๋กœ ์„ค๋ช…ํ•ฉ๋‹ˆ๋‹ค.

์ œ์•ˆํ•˜๋Š” Block-Recurrent Transformer๋Š” ์‹œํ€€์Šค๋ฅผ ๋”ฐ๋ผ ์ˆœํ™˜ ๋ฐฉ์‹์œผ๋กœ ๋ณ€์••๊ธฐ ๋ ˆ์ด์–ด๋ฅผ ์ ์šฉํ•ฉ๋‹ˆ๋‹ค. ์ด์ „์˜ ๋ชจ๋“  ํ† ํฐ์— ์ง์ ‘ ์ฐธ์—ฌํ•˜๋Š” ๊ธฐ์กด์˜ ๋ณ€ํ™˜๊ธฐ ๋ชจ๋ธ๊ณผ ๋‹ฌ๋ฆฌ Block-Recurrent Transformer๋Š” ๋ชจ๋ธ์ด ์ง€๊ธˆ๊นŒ์ง€ ๋ณธ ์‹œํ€€์Šค๋ฅผ ์š”์•ฝํ•˜๊ธฐ ์œ„ํ•ด ๊ณ ์ • ํฌ๊ธฐ ์ƒํƒœ๋ฅผ ๊ตฌ์„ฑ ๋ฐ ์œ ์ง€ํ•˜๋ฏ€๋กœ ์ˆœํ™˜ ์…€์ด ๋‹ค์Œ์„ ์ˆ˜ํ–‰ํ•  ์ˆ˜ ์žˆ๋Š” ํ† ํฐ ๋ธ”๋ก์—์„œ ์ž‘๋™ํ•  ์ˆ˜ ์žˆ์Šต๋‹ˆ๋‹ค. ํฐ ํฌ๊ธฐ๋กœ ํ™•์žฅ๋ฉ๋‹ˆ๋‹ค.


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


ํŠนํžˆ, ์ˆœํ™˜ ์…€์€ ๋‹จ์ผ ๋ฒกํ„ฐ๊ฐ€ ์•„๋‹Œ ์ƒํƒœ ๋ฒกํ„ฐ ๋ธ”๋ก์—์„œ๋„ ์ž‘๋™ํ•˜๊ธฐ ๋•Œ๋ฌธ์— ์ˆœํ™˜ ์ƒํƒœ์˜ ํฌ๊ธฐ๋Š” LSTM(์žฅ๊ธฐ ๊ธฐ์–ต ์žฅ์น˜) ์ˆœํ™˜ ์‹ ๊ฒฝ๋ง์˜ ํฌ๊ธฐ๋ณด๋‹ค ์ˆ˜์‹ญ ๋ฐฐ ๋” ํฝ๋‹ˆ๋‹ค. ๊ธด ์‹œํ€€์Šค ์ฒ˜๋ฆฌ โ€” ๊ณผ๊ฑฐ ์บก์ฒ˜๋ฅผ ์œ„ํ•œ ๋ชจ๋ธ ๊ธฐ๋Šฅ์ด ํฌ๊ฒŒ ํ–ฅ์ƒ๋˜์—ˆ์Šต๋‹ˆ๋‹ค.


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


ํŒ€์€ PG19, arXiv ๋ฐ GitHub์˜ ์„ธ ๊ฐ€์ง€ ๊ธด ๋ฌธ์„œ์—์„œ Block-Recurrent Transformer๋ฅผ ํ‰๊ฐ€ํ–ˆ์Šต๋‹ˆ๋‹ค. ๊ทธ๋“ค์€ ๋ชฉํ‘œ๊ฐ€ ์‹œํ€€์Šค์˜ ๋‹ค์Œ ํ† ํฐ์„ ์˜ˆ์ธกํ•˜๋Š” ์ž๋™ ํšŒ๊ท€ ์–ธ์–ด ๋ชจ๋ธ๋ง ์ž‘์—…์„ ์‚ฌ์šฉํ–ˆ์Šต๋‹ˆ๋‹ค.

์‹คํ—˜์—์„œ Block-Recurrent Transformer๋Š” ์ฐฝ ํฌ๊ธฐ๊ฐ€ 2048์ธ Transformer XL ๋ชจ๋ธ๋ณด๋‹ค 2๋ฐฐ ์ด์ƒ์˜ ๋Ÿฐํƒ€์ž„ ์†๋„ ํ–ฅ์ƒ์„ ๋‹ฌ์„ฑํ•˜๋ฉด์„œ ๋œ ๋ณต์žกํ•จ(๋‚ฎ์„์ˆ˜๋ก ์ข‹์Œ)์„ ๋ณด์—ฌ์ฃผ์—ˆ์Šต๋‹ˆ๋‹ค.


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


ํŒ€์€ ํ–ฅํ›„ ์—ฐ๊ตฌ ๋ฐ ๊ฐœ์„ ์ด ์ œ์•ˆ๋œ ์ˆœํ™˜ ์•„ํ‚คํ…์ฒ˜์˜ ์ž ์žฌ๋ ฅ์„ ์ตœ๋Œ€ํ•œ ์‹คํ˜„ํ•  ์ˆ˜ ์žˆ๊ธฐ๋ฅผ ๋ฐ”๋ž๋‹ˆ๋‹ค.