*์–ด๋ ต๊ฑฐ๋‚˜ ์‹œ๊ฐ„์—†๋Š” ์‚ฌ๋žŒ๋“ค์€

๋งจ ์•„๋ž˜์ชฝ์— 4์ค„ ์š”์•ฝ ์žˆ์Œ!


๋…ผ๋ฌธ ์ œ๋ชฉ : LEVERAGING UNPAIRED DATA FOR

VISION-LANGUAGE GENERATIVE MODELS VIA CYCLE CONSISTENCY


์ €์ž :

1.MIT CSAIL

2.Mila

3.Google Research

4.Google DeepMind

5.OpenAI


๊ณต๊ฐœ ์ผ์‹œ : 2023.10.05 https://browse.arxiv.org/pdf/2310.03734.pdf



Abstract ์š”์•ฝ


1. ํ˜„์žฌ ๋น„์ „-์–ธ์–ด ๋ชจ๋ธ์€ ๋ฐฉ๋Œ€ํ•œ ์ด๋ฏธ์ง€-ํ…์ŠคํŠธ ์ง ๋ฐ์ดํ„ฐ์— ์˜์กดํ•˜๊ณ  ์žˆ์Œ.


  • 2. ํ•˜์ง€๋งŒ, ์›น ์Šคํฌ๋ž˜ํ•‘์„ ํ†ตํ•ด ๊ทธ๋Ÿฐ ๋ฐฉ๋Œ€ํ•œ ๋ฐ์ดํ„ฐ๋ฅผ ๊ธฐ๊ณ„์ ์œผ๋กœ ์ˆ˜์ง‘ํ•˜๊ธฐ ๋•Œ๋ฌธ์— ์ €ํ’ˆ์งˆ๊ณผ ์ด๋ฏธ์ง€-ํ…์ŠคํŠธ ์—ฐ๊ด€์„ฑ์ด ๋ถ€์กฑํ•œ ๋ฌธ์ œ๊ฐ€ ๋ฐœ์ƒํ•จ


  • 3. ์ธ๊ฐ„์ด ์ง์ ‘ ์ฃผ์„์„ ๋‹ค๋Š” ๋ฐฉ๋ฒ•์ด ๋ณด๋‹ค ์ •ํ™•ํ•˜์ง€๋งŒ, ์—„์ฒญ๋‚œ ์ธ๋ ฅ๊ณผ ๋น„์šฉ์ด ์š”๊ตฌ๋จ


  • 4. ์ด ๋ฌธ์ œ๋ฅผ ํ•ด๊ฒฐํ•˜๊ธฐ ์œ„ํ•ด ๋ณธ ๋…ผ๋ฌธ์—์„  ITIT(InTegrating Image Text)๋ผ๋Š” ๋ฐฉ์‹์„ ์ œ์•ˆํ•จ


  • 5. ๊ธฐ์กด์— '์ง์„ ๋งž์ค€' ์ด๋ฏธ์ง€-ํ…์ŠคํŠธ ๋ฐ์ดํ„ฐ๋กœ ํ•™์Šตํ•˜๋˜ ๊ฒƒ๊ณผ ๋‹ฌ๋ฆฌ, '์ง์ด ์—†๋Š”' ์ด๋ฏธ์ง€, ํ…์ŠคํŠธ ๋ฐ์ดํ„ฐ๋กœ ํ•™์Šต์„ ํ•˜๋Š” ํ˜์‹ ์ ์ธ ํ•™์Šต ํŒจ๋Ÿฌ๋‹ค์ž„์ž„.


  • 6. ITIT๋Š” ์ด๋ฏธ์ง€-ํ…์ŠคํŠธ ์ธ์ฝ”๋”์™€ ํ…์ŠคํŠธ-์ด๋ฏธ์ง€ ์ƒ์„ฑ ๋ชจ๋ธ์„ ์–‘๋ฐฉํ–ฅ์œผ๋กœ ํ•ฉ์ณ๋†“์€ ๋‹จ์ผ ํ”„๋ ˆ์ž„์›Œํฌ์ž„.


  • 7. ํ•™์Šตํ•˜๋Š” ๋™์•ˆ, ITIT๋Š” ์ ์€ ์ˆ˜์˜ ์ด๋ฏธ์ง€-ํ…์ŠคํŠธ ์ง ๋ฐ์ดํ„ฐ๋กœ ์•„์›ƒํ’‹์ด ์ธํ’‹๊ณผ ์ž˜ ๋งค์นญ์ด ๋˜๋Š”์ง€ ํ™•์ธํ•˜๋„๋ก ํ•จ.


  • 8. ๋™์‹œ์—, ๋ชจ๋ธ์€ ์ด๋ฏธ์ง€๋‚˜ ํ…์ŠคํŠธ ๋‘˜ ์ค‘ ํ•˜๋‚˜๋งŒ ์žˆ๋Š” ํ›จ์”ฌ ๋” ํฐ ๋ฐ์ดํ„ฐ์…‹์—์„œ๋„ ํ•™์Šตํ•จ.


  • 9. ์ด๋Š” ์›๋ž˜ ์ง์ด ์—†๋Š” ์ƒ˜ํ”Œ๊ณผ ์‚ฌ์ดํด ์ƒ์—์„œ ์ƒ์„ฑ๋œ ๊ฒƒ๋“ค ์‚ฌ์ด์˜ 'cycle consistency'๋ฅผ ๋งž์ถ”๋„๋ก ๊ฐ•์ œํ•˜์—ฌ ์ด๋ฃจ์–ด์ง.


  • 10. ์˜ˆ๋ฅผ ๋“ค์–ด, ์ฃผ์–ด์ง„ ์ด๋ฏธ์ง€(input)์— ๋Œ€ํ•œ ์บก์…˜์„ ๋ชจ๋ธ์ด ์ƒ์„ฑํ•˜๊ณ , ์ด ์ƒ์„ฑ๋œ ์บก์…˜์„ ๊ฐ€์ง€๊ณ  ๋˜ ์ด๋ฏธ์ง€๋ฅผ ๋‹ค์‹œ ์ƒ์„ฑ(output)ํ•œ ๋‹ค์Œ, input๊ณผ output ์‚ฌ์ด์˜ ์œ ์‚ฌ๋„๊ฐ€ ๋†’๋„๋ก ๊ฐ•์ œํ•˜๋Š” ์‹์œผ๋กœ ํ•™์Šต์ด ์ด๋ฃจ์–ด์ง.


  • 11. ๋ณธ ๋…ผ๋ฌธ์˜ ์‹คํ—˜๊ฒฐ๊ณผ, ITIT๋ฅผ ์‚ฌ์šฉํ•ด ์ง์ด ์—†๋Š” ๋ฐ์ดํ„ฐ์…‹์œผ๋กœ ํ•™์Šตํ•œ ๊ฒฐ๊ณผ๊ฐ€ '๊ณ ํ’ˆ์งˆ์˜ ์ง์ด ์žˆ๋Š” ๋ฐ์ดํ„ฐ'๋ฅผ ์‚ฌ์šฉํ•œ ๊ฒƒ๊ณผ ๋น„์Šทํ•œ ์ˆ˜์ค€์˜ ์Šค์ผ€์ผ ๋ฒ•์น™์„ ๋”ฐ๋ฅด๋Š” ๊ฒƒ์œผ๋กœ ๋ฐํ˜€์ง.


  • 12. ๋ณธ ๋…ผ๋ฌธ์—์„œ๋Š” ์ตœ๊ณ ์˜ ํ…์ŠคํŠธ-์ด๋ฏธ์ง€, ์ด๋ฏธ์ง€-ํ…์ŠคํŠธ ๋ชจ๋ธ๊ณผ ๋™๋“ฑํ•œ ์„ฑ๋Šฅ์„ ๋ณด์ด๋Š”๋ฐ ํ›จ์”ฌ ๋” ์ ์€ '์ง ์ง€์–ด์ง„' (์˜ค์ง 3M๊ฐœ)์˜ ๋ฐ์ดํ„ฐ๋งŒ ์žˆ์œผ๋ฉด ๋œ๋‹ค๋Š” ๊ฒƒ์„ ์ฆ๋ช…ํ•จ.



๋ณธ๋ฌธ ์š”์•ฝ



  • ๊ธฐ์กด ์ด๋ฏธ์ง€-ํ…์ŠคํŠธ ์ƒ์„ฑ ๋ชจ๋ธ๋“ค์˜ ํ•™์Šต์—๋Š” ์ˆ˜๋ฐฑ๋งŒ๊ฐœ์—์„œ ์ˆ˜๋ฐฑ์–ต๊ฐœ์˜ '์ง ์ง€์–ด์ง„' ์ด๋ฏธ์ง€ํ…์ŠคํŠธ ๋ฐ์ดํ„ฐ๊ฐ€ ํ•„์š”ํ–ˆ์Šต๋‹ˆ๋‹ค.

  • ์—ฌ๊ธฐ์„œ ์šฐ๋ฆฌ๋Š” ํ•œ ๊ฐ€์ง€ ์ž์—ฐ์Šค๋Ÿฌ์šด ์งˆ๋ฌธ์„ ํ•  ์ˆ˜ ์žˆ์Šต๋‹ˆ๋‹ค : '์šฐ๋ฆฌ๊ฐ€ '์ง์ด ์—†๋Š”' ์ด๋ฏธ์ง€์™€ ํ…์ŠคํŠธ ๋ฐ์ดํ„ฐ๋ฅผ ๋น„์ „-์–ธ์–ด ๋ชจ๋ธ ํ•™์Šต์— ํ™œ์šฉํ•  ์ˆ˜ ์žˆ๋Š” ๋ฐฉ๋ฒ•์€ ์—†์„๊นŒ?'

  • ์šฐ๋ฆฌ๋Š” ์ด๋ฅผ ํ•ด๊ฒฐํ•˜๊ธฐ ์œ„ํ•ด 'ITIT'๋ฅผ ์ œ์•ˆํ•ฉ๋‹ˆ๋‹ค. ITIT๋Š” 'Cycle consistency losses'๋ฅผ ์‚ฌ์šฉํ•˜๋Š” ์ƒˆ๋กœ์šด ํ•™์Šต ํŒจ๋Ÿฌ๋‹ค์ž„์ž…๋‹ˆ๋‹ค.

7cf3c028e2f206a26d81f6e4478572697e



  • ์œ„ ๊ทธ๋ฆผ์—์„œ ๋ณผ ์ˆ˜ ์žˆ๋“ฏ, ITIT๋Š” ์ฃผ์–ด์ง„ '์ง์ด ์—†๋Š” ํ…์ŠคํŠธ'์— ๋Œ€ํ•ด T2I ๋ชจ๋ธ์„ ๊ฑฐ์ณ ํ•ฉ์„ฑ๋œ ์ด๋ฏธ์ง€์™€ 'ํ•ฉ์„ฑ๋œ ์ด๋ฏธ์ง€์—์„œ I2T ๋ชจ๋ธ์„ ๊ฑฐ์ณ ๋งŒ๋“  ํ…์ŠคํŠธ' ์‚ฌ์ด์˜ Reconstruction Loss๋ฅผ ์ตœ์†Œํ™”ํ•˜๋Š” ๋ฐฉ์‹์œผ๋กœ ํ•™์Šต๋ฉ๋‹ˆ๋‹ค. ์ด๋ฏธ์ง€์˜ ๊ฒฝ์šฐ๋„ ๊ฐ™์€ ์›๋ฆฌ์ž…๋‹ˆ๋‹ค. (*T2I : Text to Image, I2T : Image to Text)


๋‹ค์–‘ํ•œ ๋ฒค์น˜๋งˆํฌ๋ฅผ ํ†ตํ•ด ITIT์˜ ์„ฑ๋Šฅ์„ ์ ๊ฒ€ํ•ด๋ดค์Šต๋‹ˆ๋‹ค. ITIT๋Š” ๊ธฐ์กด์˜ non-cycle ๋ฒ ์ด์Šค๋ผ์ธ๊ณผ ๋น„์Šทํ•œ ์ ์ˆ˜๋ฅผ ์–ป์—ˆ์ง€๋งŒ, ITIT๋Š” ๊ธฐ์กด๋ณด๋‹ค ํ›จ์”ฌ ๋” ์ ์€ ๋ฐ์ดํ„ฐ๋งŒ์„ ์‚ฌ์šฉํ•ด์„œ ์–ป์€ ์„ฑ๊ณผ์ž…๋‹ˆ๋‹ค.


7ff3c028e2f206a26d81f6ec43857365

๊ธฐ์กด๋ณด๋‹ค '#paired data'๊ฐ€ ํ›จ์”ฌ ๋” ์ค„์–ด๋“ค์—ˆ์Œ์—๋„ ๋น„์Šทํ•œ ์„ฑ๋Šฅ์„ ๋ณด์ž„


7ef3c028e2f206a26d81f6e441837d69e6

  • ITIT ํ•™์Šต๊ณผ์ •์„ ๋ณด์—ฌ์ค€๋‹ค. ์บก์…˜์„ ํ†ตํ•ด ์ด๋ฏธ์ง€๋ฅผ ์ƒ์„ฑ->์ƒ์„ฑ๋œ ์ด๋ฏธ์ง€์—์„œ ์บก์…˜์„ ์ƒ์„ฑ->์ƒ์„ฑ๋œ ์บก์…˜์—์„œ ์ด๋ฏธ์ง€๋ฅผ ์ƒ์„ฑ.... ์ด Cycle์„ ๋ฐ˜๋ณตํ•˜๋ฉฐ reconstruction loss๋ฅผ ์ตœ์†Œํ™”ํ•˜๋Š” ์‹์œผ๋กœ ํ•™์Šต์„ ์ง„ํ–‰ํ•œ๋‹ค.



๊ฒฐ๋ก 

  • ์ด ๋…ผ๋ฌธ์—์„œ๋Š” '์ง์ด ์—†๋Š”' ์ด๋ฏธ์ง€์™€ ํ…์ŠคํŠธ ๋ฐ์ดํ„ฐ๋ฅผ ๋น„์ „-์–ธ์–ด ์ƒ์„ฑ ํ•™์Šต์— ํ™œ์šฉํ•˜๋Š” ์ƒˆ๋กœ์šด ํ•™์Šต ๋ฐฉ์‹์„ ์ฒ˜์Œ์œผ๋กœ ์ œ์‹œํ–ˆ๋‹ค.

  • ์—ฌ๋Ÿฌ ์‹คํ—˜ ๊ฒฐ๊ณผ T2I2T cycle๊ณผ I2T2I cycle์€ ํ…์ŠคํŠธ-์ด๋ฏธ์ง€, ์ด๋ฏธ์ง€-ํ…์ŠคํŠธ ํ•ฉ์„ฑ ์„ฑ๋Šฅ์„ ํšจ๊ณผ์ ์œผ๋กœ ๊ฐœ์„ ํ•œ๋‹ค๋Š” ๊ฒƒ์„ ์ฆ๋ช…ํ–ˆ๋‹ค.

  • ITIT๋Š” SOTA ๋น„์ „-์–ธ์–ด ์ƒ์„ฑ ๋ชจ๋ธ๋“ค๊ณผ ๊ฒฝ์Ÿํ• ๋งŒํ•œ ์„ฑ๋Šฅ์„ ๋ณด์—ฌ์คฌ์ง€๋งŒ, ํ•™์Šต์— ์‚ฌ์šฉํ•œ '์ง ์ง€์–ด์ง„' ์ด๋ฏธ์ง€-ํ…์ŠคํŠธ ์ƒ˜ํ”Œ์€ ๋‹จ 3M๊ฐœ์— ๋ถˆ๊ณผํ–ˆ๋‹ค.

  • ์šฐ๋ฆฌ๊ฐ€ ์ œ์‹œํ•œ ๋ฐฉ๋ฒ•์€ '์ง ์ง€์–ด์ง„' ์ด๋ฏธ์ง€-ํ…์ŠคํŠธ ๋ฐ์ดํ„ฐ๊ฐ€ ์กฐ๊ธˆ์€ ์กด์žฌํ•ด์•ผ ์‚ฌ์šฉํ•  ์ˆ˜ ์žˆ๋‹ค. ๋˜ํ•œ ์ง ์ง€์–ด์ง„ ๋ฐ์ดํ„ฐ์˜ ํ’ˆ์งˆ์ด ๋‚ฎ์„ ๋•Œ ํŠน๋ณ„ํžˆ ๋” ๋งŽ์€ ๋„์›€์ด ๋œ๋‹ค.

  • ์ถ”ํ›„ ์—ฐ๊ตฌ์—์„œ๋Š” ITIT๋ฅผ ๋” ๋งŽ์€ '์ง์ด ์—†๋Š”' ์ด๋ฏธ์ง€์™€ ํ…์ŠคํŠธ ๋ฐ์ดํ„ฐ๋ฅผ ์‚ฌ์šฉํ•˜์—ฌ ๊ทœ๋ชจ๋ฅผ ๋„“ํ˜€๊ฐ€๋Š” ํ•œํŽธ, ๋” ํฐ ๋ชจ๋ธ๊ณผ ๋” ๋‹ค์–‘ํ•œ ๋ฐ์ดํ„ฐ์…‹์„ ์‚ฌ์šฉํ•ด๋ณผ ์˜ˆ์ •์ด๋‹ค.




๋‚ด์šฉ๋“ค์ด ํ•˜๋‚˜ํ•˜๋‚˜ ๋‹ค ์ค‘์š”ํ•ด์„œ ๋งŽ์ด ๋ชป ์ค„์˜€๋‹ค.
4์ค„๋กœ ์š”์•ฝํ•ด๋ด„.

1. ๊ธฐ์กด ๋น„์ „-์–ธ์–ด ๋ชจ๋ธ ํ•™์Šต์—๋Š” ๋‹น์—ฐํžˆ ์ง ์ง€์–ด์ง„(paired) ์ด๋ฏธ์ง€-ํ…์ŠคํŠธ ์Œ์ด ํ•™์Šต ๋ฐ์ดํ„ฐ๋กœ ํ•„์š”ํ–ˆ์Œ

2. ๊ทธ๋Ÿฐ๋ฐ ์ง์ด ์—†์ด ์ด๋ฏธ์ง€, ํ…์ŠคํŠธ ๋‹จ๋…์œผ๋กœ๋งŒ ์žˆ๋Š” (unpaired) ๋ฐ์ดํ„ฐ๋ฅผ ํ•™์Šต์— ์‚ฌ์šฉํ•˜๊ธฐ ์œ„ํ•ด, ์˜ˆ๋ฅผ ๋“ค์–ด ํ…์ŠคํŠธ ๋ฐ์ดํ„ฐ๋ผ๋ฉด ํ…์ŠคํŠธ ๋ฐ์ดํ„ฐ๋ฅผ ํ† ๋Œ€๋กœ T2I ๋ชจ๋ธ๋กœ ์ด๋ฏธ์ง€๋ฅผ ์ƒ์„ฑ -> ์ƒ์„ฑ๋œ ์ด๋ฏธ์ง€๋กœ I2T ๋ชจ๋ธ๋กœ ํ…์ŠคํŠธ ์บก์…˜์„ ์ƒ์„ฑํ•˜๊ณ  ์›๋ณธ ํ…์ŠคํŠธ์™€ ์ƒ์„ฑ๋œ ํ…์ŠคํŠธ ์บก์…˜๊ฐ„์˜ reconstruction loss๋ฅผ ์ตœ์†Œํ™”ํ•˜๋Š” ๋ฐฉ์‹์œผ๋กœ cycle training์„ ์ง„ํ–‰ํ•จ

7ef3c028e2f206a26d81f6e441837d69e6

(ITIT ๊ณผ์ •)


3. ๊ทธ ๊ฒฐ๊ณผ ๊ธฐ์กด ์ˆ˜์–ต~์ˆ˜๋ฐฑ์–ต๊ฐœ์˜ paired ์ด๋ฏธ์ง€-ํ…์ŠคํŠธ ์Œ์ด ์‚ฌ์šฉ๋๋˜ ์—ฐ๊ตฌ๋“ค๊ณผ ๋‹ฌ๋ฆฌ ๋‹จ 300๋งŒ๊ฐœ์˜ ์ด๋ฏธ์ง€-ํ…์ŠคํŠธ ์Œ๊ณผ 5500๋งŒ๊ฐœ์˜ ์ง์ด ์—†๋Š” ์ด๋ฏธ์ง€, ํ…์ŠคํŠธ ๋ฐ์ดํ„ฐ๋งŒ์„ ๊ฐ€์ง€๊ณ ๋„ ๊ธฐ์กด๊ณผ ๋น„์Šทํ•œ ์„ฑ์ ์„ ๊ธฐ๋กํ•จ

4. unpaired data์— ๋Œ€ํ•ด์„œ๋„ ๊ธฐ์กด paired data์™€ ๋งˆ์ฐฌ๊ฐ€์ง€๋กœ '์Šค์ผ€์ผ ๋ฒ•์น™'์ด ๊ทธ๋Œ€๋กœ ์ ์šฉ๋จ์„ ํ™•์ธํ•จ. ๋”ฐ๋ผ์„œ ์ถ”ํ›„ ์—ฐ๊ตฌ์—์„œ ๋ชจ๋ธ์„ ๋” ํฌ๊ฒŒ ๋งŒ๋“ค๊ณ  ํ›จ์”ฌ ๋” ๋งŽ์€ unpaired dataset์„ ์‚ฌ์šฉํ•ด์„œ ์–ผ๋งˆ๋‚˜ ์„ฑ๋Šฅ์ด ์ฆ๊ฐ€ํ•˜๋Š”์ง€ ํ™•์ธํ•  ์˜ˆ์ •์ด๋ผ๊ณ  ํ•จ




๊ฐœ์ธ์ ์œผ๋กœ GPT-1 ๋…ผ๋ฌธ ์ฒ˜์Œ ์ฝ์—ˆ์„ ๋•Œ๋ž‘ ๋น„์Šทํ•œ ๋А๋‚Œ์ด์—ˆ์Œ

์•ž์œผ๋กœ ์ด์ชฝ ๋ถ„์•ผ ์—ฐ๊ตฌ๋ฐฉํ–ฅ์ด ๋งŽ์ด ๋‹ฌ๋ผ์งˆ ์ˆ˜๋„ ์žˆ๋Š” ์ค‘์š”ํ•œ ๋…ผ๋ฌธ์ด๋ผ๊ณ  ์ƒ๊ฐํ–ˆ๋‹ค.


์ด์ œ ๊ทธ๋ƒฅ ํ…์ŠคํŠธ-์ด๋ฏธ์ง€ ์Œ์ด ์•„๋‹ˆ๋ผ ๊ฐ๊ฐ ์žˆ๋Š” ๋ฐ์ดํ„ฐ๋กœ๋„ ๋น„์ „-์–ธ์–ด ๋ชจ๋ธ์„ ํ•™์Šต์‹œํ‚ฌ ์ˆ˜ ์žˆ๊ฒŒ ๋œ๊ฑฐ์ž„.

์ด ์„ธ์ƒ์— ์กด์žฌํ•˜๋Š” Unpaired ๋ฐ์ดํ„ฐ๊ฐ€ ์–ผ๋งˆ๋‚˜ ๋งŽ์€์ง€ ์ƒ๊ฐํ•ด๋ณด๋ฉด, ์—„์ฒญ๋‚œ ์„ฑ๊ณผ๋ผ๊ณ  ๋ณธ๋‹ค.