https://arxiv.org/abs/2308.03712


์ธ๊ฐ„๊ณผ ๊ฐ™์€ ์‹œ๊ฐ์  ๊ฒฝํ—˜์„ ํ†ตํ•ด ์ธ๊ฐ„ ์ˆ˜์ค€์˜ ๊ฐ์ฒด ์ธ์‹ ๋Šฅ๋ ฅ์„ ๋‹ฌ์„ฑํ•˜๋Š” ๋ฐ ํ•„์š”ํ•œ ๊ฒƒ์€ ํ™•์žฅ๋ฟ์ผ ์ˆ˜ ์žˆ์Šต๋‹ˆ๋‹ค.


์ด ๋…ผ๋ฌธ์—์„œ๋Š” ํ˜„์žฌ์˜ ์ž๊ธฐ ์ง€๋„ ํ•™์Šต ๋ฐฉ๋ฒ•์ด ์ถฉ๋ถ„ํžˆ ํ™•์žฅ๋œ๋‹ค๋ฉด ์ธ๊ฐ„์ด ๋ฐฐ์šฐ๋Š” ๊ฒƒ๊ณผ ๋™์ผํ•œ ์œ ํ˜•๊ณผ ์–‘์˜ ์‹œ๊ฐ์  ๊ฒฝํ—˜์œผ๋กœ ์ธ๊ฐ„ ์ˆ˜์ค€์˜ ์‹œ๊ฐ์  ๊ฐœ์ฒด ์ธ์‹ ๊ธฐ๋Šฅ์— ๋„๋‹ฌํ•  ์ˆ˜ ์žˆ๋Š”์ง€ ๋ฌป์Šต๋‹ˆ๋‹ค. ์ด ์งˆ๋ฌธ์— ๋Œ€ํ•œ ์ด์ „ ์—ฐ๊ตฌ์—์„œ๋Š” ๋ฐ์ดํ„ฐ ํฌ๊ธฐ์˜ ํ™•์žฅ๋งŒ ๊ณ ๋ คํ–ˆ์Šต๋‹ˆ๋‹ค. ์—ฌ๊ธฐ์„œ๋Š” ๋ฐ์ดํ„ฐ ํฌ๊ธฐ, ๋ชจ๋ธ ํฌ๊ธฐ ๋ฐ ์ด๋ฏธ์ง€ ํ•ด์ƒ๋„์˜ ๋™์‹œ ํ™•์žฅ์„ ๊ณ ๋ คํ•ฉ๋‹ˆ๋‹ค. ์šฐ๋ฆฌ๋Š” ์ตœ๋Œ€ 476x476 ํ”ฝ์…€์˜ ์ด๋ฏธ์ง€ ํ•ด์ƒ๋„๋ฅผ ๊ฐ–์ถ˜ ์ตœ๋Œ€ 5,000์‹œ๊ฐ„์˜ ์ธ๊ฐ„๊ณผ ์œ ์‚ฌํ•œ ๋น„๋””์˜ค ๋ฐ์ดํ„ฐ(์žฅ์‹œ๊ฐ„ ์—ฐ์†, ๋Œ€๋ถ€๋ถ„ ์ž๊ธฐ ์ค‘์‹ฌ์  ๋น„๋””์˜ค)๋กœ ํ›ˆ๋ จ๋œ ์ตœ๋Œ€ 633M ๋งค๊ฐœ๋ณ€์ˆ˜ ํฌ๊ธฐ(ViT-H/14)์˜ ๋น„์ „ ๋ณ€ํ™˜๊ธฐ๋ฅผ ์‚ฌ์šฉํ•˜์—ฌ ์Šค์ผ€์ผ๋ง ์‹คํ—˜์„ ์ˆ˜ํ–‰ํ•ฉ๋‹ˆ๋‹ค. . ์ž๊ธฐ ์ง€๋„ ํ•™์Šต ์•Œ๊ณ ๋ฆฌ์ฆ˜์ธ MAE(Masked Autoencoder)์˜ ํšจ์œจ์„ฑ ๋•๋ถ„์— ์ ์€ ํ•™๋น„๋กœ ์ด ํ™•์žฅ ์‹คํ—˜์„ ์‹คํ–‰ํ•  ์ˆ˜ ์žˆ์Šต๋‹ˆ๋‹ค. ์šฐ๋ฆฌ๋Š” ๋ชจ๋ธ ํฌ๊ธฐ, ๋ฐ์ดํ„ฐ ํฌ๊ธฐ ๋ฐ ์ด๋ฏธ์ง€ ํฌ๊ธฐ์˜ ์ธ๊ฐ„ ์ดํ•˜ ๊ทœ๋ชจ์—์„œ ์ด๋Ÿฌํ•œ ์š”์†Œ๊ฐ€ ๋™์‹œ์— ํ™•์žฅ๋˜๋ฉด ์ธ๊ฐ„ ์ˆ˜์ค€์˜ ๊ฐ์ฒด ์ธ์‹ ๋Šฅ๋ ฅ์— ๋„๋‹ฌํ•˜๋Š” ๊ฒƒ์ด ๊ฐ€๋Šฅํ•˜๋‹ค๋Š” ๊ฒƒ์„ ๋ฐœ๊ฒฌํ–ˆ์Šต๋‹ˆ๋‹ค. ๊ตฌ์ฒด์ ์ธ ์˜ˆ๋ฅผ ๋“ค์ž๋ฉด, 952x952 ํ”ฝ์…€์˜ ๊ณต๊ฐ„ ํ•ด์ƒ๋„๋ฅผ ๊ฐ–์ถ˜ ์ธ๊ฐ„๊ณผ ์œ ์‚ฌํ•œ 20,000์‹œ๊ฐ„(2.3๋…„)์˜ ๋น„๋””์˜ค ๋ฐ์ดํ„ฐ๋กœ ํ›ˆ๋ จ๋œ 2.5B ๋งค๊ฐœ๋ณ€์ˆ˜ ViT ๋ชจ๋ธ์ด ImageNet์—์„œ ๋Œ€๋žต ์ธ๊ฐ„ ์ˆ˜์ค€์˜ ์ •ํ™•๋„์— ๋„๋‹ฌํ•  ์ˆ˜ ์žˆ์–ด์•ผ ํ•œ๋‹ค๊ณ  ์ถ”์ •ํ•ฉ๋‹ˆ๋‹ค. ๋”ฐ๋ผ์„œ ์ธ๊ฐ„ ์ˆ˜์ค€์˜ ์—ญ๋Ÿ‰์€ ์‹ค์งˆ์ ์ธ ๊ท€๋‚ฉ์  ํŽธ๊ฒฌ ์—†์ด ๋งค์šฐ ์ผ๋ฐ˜์ ์ธ ํ•™์Šต ์•Œ๊ณ ๋ฆฌ์ฆ˜ ๋ฐ ์•„ํ‚คํ…์ฒ˜๋ฅผ ์‚ฌ์šฉํ•˜์—ฌ ์ธ๊ฐ„๊ณผ ์œ ์‚ฌํ•œ ์ง€๊ฐ ๊ฒฝํ—˜(์–‘๊ณผ ์œ ํ˜• ๋ชจ๋‘์—์„œ ์ธ๊ฐ„๊ณผ ์œ ์‚ฌ)์„ ํ†ตํ•ด ๊ทผ๋ณธ์ ์ธ ์ง€๊ฐ ๋Šฅ๋ ฅ์„ ๋‹ฌ์„ฑํ•  ์ˆ˜ ์žˆ์Šต๋‹ˆ๋‹ค.