5.7 KiB
DeBERTa-v2
๊ฐ์
DeBERTa ๋ชจ๋ธ์ Pengcheng He, Xiaodong Liu, Jianfeng Gao, Weizhu Chen์ด ์์ฑํ DeBERTa: ๋ถ๋ฆฌ๋ ์ดํ ์ ์ ํ์ฉํ ๋์ฝ๋ฉ ๊ฐํ BERT์ด๋ผ๋ ๋ ผ๋ฌธ์์ ์ ์๋์์ต๋๋ค. ์ด ๋ชจ๋ธ์ 2018๋ Google์ด ๋ฐํํ BERT ๋ชจ๋ธ๊ณผ 2019๋ Facebook์ด ๋ฐํํ RoBERTa ๋ชจ๋ธ์ ๊ธฐ๋ฐ์ผ๋ก ํฉ๋๋ค. DeBERTa๋ RoBERTa์์ ์ฌ์ฉ๋ ๋ฐ์ดํฐ์ ์ ๋ฐ๋ง์ ์ฌ์ฉํ์ฌ ๋ถ๋ฆฌ๋(disentangled) ์ดํ ์ ๊ณผ ํฅ์๋ ๋ง์คํฌ ๋์ฝ๋ ํ์ต์ ํตํด RoBERTa๋ฅผ ๊ฐ์ ํ์ต๋๋ค.
๋ ผ๋ฌธ์ ์ด๋ก์ ๋ค์๊ณผ ๊ฐ์ต๋๋ค:
์ฌ์ ํ์ต๋ ์ ๊ฒฝ๋ง ์ธ์ด ๋ชจ๋ธ์ ์ต๊ทผ ๋ฐ์ ์ ๋ง์ ์์ฐ์ด ์ฒ๋ฆฌ(NLP) ์์ ์ ์ฑ๋ฅ์ ํฌ๊ฒ ํฅ์์์ผฐ์ต๋๋ค. ๋ณธ ๋ ผ๋ฌธ์์๋ ๋ ๊ฐ์ง ์๋ก์ด ๊ธฐ์ ์ ์ฌ์ฉํ์ฌ BERT์ RoBERTa ๋ชจ๋ธ์ ๊ฐ์ ํ ์๋ก์ด ๋ชจ๋ธ ๊ตฌ์กฐ์ธ DeBERTa๋ฅผ ์ ์ํฉ๋๋ค. ์ฒซ ๋ฒ์งธ๋ ๋ถ๋ฆฌ๋ ์ดํ ์ ๋ฉ์ปค๋์ฆ์ผ๋ก, ๊ฐ ๋จ์ด๊ฐ ๋ด์ฉ๊ณผ ์์น๋ฅผ ๊ฐ๊ฐ ์ธ์ฝ๋ฉํ๋ ๋ ๊ฐ์ ๋ฒกํฐ๋ก ํํ๋๋ฉฐ, ๋จ์ด๋ค ๊ฐ์ ์ดํ ์ ๊ฐ์ค์น๋ ๋ด์ฉ๊ณผ ์๋์ ์์น์ ๋ํ ๋ถ๋ฆฌ๋ ํ๋ ฌ์ ์ฌ์ฉํ์ฌ ๊ณ์ฐ๋ฉ๋๋ค. ๋ ๋ฒ์งธ๋ก, ๋ชจ๋ธ ์ฌ์ ํ์ต์ ์ํด ๋ง์คํน๋ ํ ํฐ์ ์์ธกํ๋ ์ถ๋ ฅ ์ํํธ๋งฅ์ค ์ธต์ ๋์ฒดํ๋ ํฅ์๋ ๋ง์คํฌ ๋์ฝ๋๊ฐ ์ฌ์ฉ๋ฉ๋๋ค. ์ฐ๋ฆฌ๋ ์ด ๋ ๊ฐ์ง ๊ธฐ์ ์ด ๋ชจ๋ธ ์ฌ์ ํ์ต์ ํจ์จ์ฑ๊ณผ ๋ค์ด์คํธ๋ฆผ ์์ ์ ์ฑ๋ฅ์ ํฌ๊ฒ ํฅ์์ํจ๋ค๋ ๊ฒ์ ๋ณด์ฌ์ค๋๋ค. RoBERTa-Large์ ๋น๊ตํ์ ๋, ์ ๋ฐ์ ํ์ต ๋ฐ์ดํฐ๋ก ํ์ต๋ DeBERTa ๋ชจ๋ธ์ ๊ด๋ฒ์ํ NLP ์์ ์์ ์ผ๊ด๋๊ฒ ๋ ๋์ ์ฑ๋ฅ์ ๋ณด์ฌ์ฃผ๋ฉฐ, MNLI์์ +0.9%(90.2% vs 91.1%), SQuAD v2.0์์ +2.3%(88.4% vs 90.7%), RACE์์ +3.6%(83.2% vs 86.8%)์ ์ฑ๋ฅ ํฅ์์ ๋ฌ์ฑํ์ต๋๋ค. DeBERTa ์ฝ๋์ ์ฌ์ ํ์ต๋ ๋ชจ๋ธ์ https://github.com/microsoft/DeBERTa ์์ ๊ณต๊ฐ๋ ์์ ์ ๋๋ค.
๋ค์ ์ ๋ณด๋ค์ ์๋ณธ ๊ตฌํ ์ ์ฅ์์์ ๋ณด์ค ์ ์์ต๋๋ค. DeBERTa v2๋ DeBERTa์ ๋๋ฒ์งธ ๋ชจ๋ธ์ ๋๋ค. DeBERTa v2๋ SuperGLUE ๋จ์ผ ๋ชจ๋ธ ์ ์ถ์ ์ฌ์ฉ๋ 1.5B ๋ชจ๋ธ์ ํฌํจํ๋ฉฐ, ์ธ๊ฐ ๊ธฐ์ค์ (๋ฒ ์ด์ค๋ผ์ธ) 89.8์ ๋๋น 89.9์ ์ ๋ฌ์ฑํ์ต๋๋ค. ์ ์์ ๋ธ๋ก๊ทธ์์ ๋ ์์ธํ ์ ๋ณด๋ฅผ ํ์ธํ ์ ์์ต๋๋ค.
v2์ ์๋ก์ด ์ :
- ์ดํ(Vocabulary) v2์์๋ ํ์ต ๋ฐ์ดํฐ๋ก๋ถํฐ ๊ตฌ์ถ๋ 128K ํฌ๊ธฐ์ ์๋ก์ด ์ดํ๋ฅผ ์ฌ์ฉํ๋๋ก ํ ํฌ๋์ด์ ๊ฐ ๋ณ๊ฒฝ๋์์ต๋๋ค. GPT2 ๊ธฐ๋ฐ ํ ํฌ๋์ด์ ๋์ , ์ด์ ๋ ์ผํ ์คํผ์ค ๊ธฐ๋ฐ ํ ํฌ๋์ด์ ๋ฅผ ์ฌ์ฉํฉ๋๋ค.
- nGiE[n๊ทธ๋จ ์ ๋(Induced) ์ ๋ ฅ ์ธ์ฝ๋ฉ] DeBERTa-v2 ๋ชจ๋ธ์ ์ ๋ ฅ ํ ํฐ๋ค์ ์ง์ญ์ ์์กด์ฑ์ ๋ ์ ํ์ตํ๊ธฐ ์ํด ์ฒซ ๋ฒ์งธ ํธ๋์คํฌ๋จธ ์ธต๊ณผ ํจ๊ป ์ถ๊ฐ์ ์ธ ํฉ์ฑ๊ณฑ ์ธต์ ์ฌ์ฉํฉ๋๋ค.
- ์ดํ ์ ์ธต์์ ์์น ํฌ์ ํ๋ ฌ๊ณผ ๋ด์ฉ ํฌ์ ํ๋ ฌ ๊ณต์ ์ด์ ์คํ๋ค์ ๊ธฐ๋ฐ์ผ๋ก, ์ด๋ ์ฑ๋ฅ์ ์ํฅ์ ์ฃผ์ง ์์ผ๋ฉด์ ๋งค๊ฐ๋ณ์๋ฅผ ์ ์ฝํ ์ ์์ต๋๋ค.
- ์๋์ ์์น๋ฅผ ์ธ์ฝ๋ฉํ๊ธฐ ์ํ ๋ฒํท ์ ์ฉ DeBERTa-v2 ๋ชจ๋ธ์ T5์ ์ ์ฌํ๊ฒ ์๋์ ์์น๋ฅผ ์ธ์ฝ๋ฉํ๊ธฐ ์ํด ๋ก๊ทธ ๋ฒํท์ ์ฌ์ฉํฉ๋๋ค.
- 900M ๋ชจ๋ธ & 1.5B ๋ชจ๋ธ 900M๊ณผ 1.5B, ๋ ๊ฐ์ง ์ถ๊ฐ ๋ชจ๋ธ ํฌ๊ธฐ๊ฐ ์ ๊ณต๋๋ฉฐ, ์ด๋ ๋ค์ด์คํธ๋ฆผ ์์ ์ ์ฑ๋ฅ์ ํฌ๊ฒ ํฅ์์ํต๋๋ค.
DeBERTa ๋ชจ๋ธ์ ํ ์ํ๋ก 2.0 ๊ตฌํ์ kamalkraj๊ฐ ๊ธฐ์ฌํ์ต๋๋ค. ์๋ณธ ์ฝ๋๋ ์ด๊ณณ์์ ํ์ธํ์ค ์ ์์ต๋๋ค.
์๋ฃ
- ํ ์คํธ ๋ถ๋ฅ ์์ ๊ฐ์ด๋
- ํ ํฐ ๋ถ๋ฅ ์์ ๊ฐ์ด๋
- ์ง์์๋ต ์์ ๊ฐ์ด๋
- ๋ง์คํฌ ์ธ์ด ๋ชจ๋ธ๋ง ์์ ๊ฐ์ด๋
- ๋ค์ค ์ ํ ์์ ๊ฐ์ด๋
DebertaV2Config
autodoc DebertaV2Config
DebertaV2Tokenizer
autodoc DebertaV2Tokenizer - get_special_tokens_mask - save_vocabulary
DebertaV2TokenizerFast
autodoc DebertaV2TokenizerFast
DebertaV2Model
autodoc DebertaV2Model - forward
DebertaV2PreTrainedModel
autodoc DebertaV2PreTrainedModel - forward
DebertaV2ForMaskedLM
autodoc DebertaV2ForMaskedLM - forward
DebertaV2ForSequenceClassification
autodoc DebertaV2ForSequenceClassification - forward
DebertaV2ForTokenClassification
autodoc DebertaV2ForTokenClassification - forward
DebertaV2ForQuestionAnswering
autodoc DebertaV2ForQuestionAnswering - forward
DebertaV2ForMultipleChoice
autodoc DebertaV2ForMultipleChoice - forward