Skip to content

Latest commit

 

History

History
77 lines (60 loc) · 5.88 KB

README.md

File metadata and controls

77 lines (60 loc) · 5.88 KB

한국어 | English

Finetuning (Benchmark on subtask)

  • Transformers examples를 참고하여 제작
  • Finetuning에는 discriminator를 사용
  • Single GPU 기준으로 코드 작성
  • KoELECTRA의 라이센스와는 별개로 데이터별로도 라이센스가 별도로 존재합니다

Updates

July 7, 2020

  • transformers v3에 맞춰서 코드를 수정하였습니다. 기존에는 ElectraForSequenceClassification 등이 지원되지 않아 직접 구현하였지만, 최근에는 공식 라이브러리에 구현이 되어 있어서 이를 사용하는 방향으로 수정하였습니다.
  • 기존 src/model.py에서의 ElectraForSequenceClassification과 modeling_electra.py의 ElectraForSequenceClassification 구현 방식이 상이합니다.

Requirements

torch==1.6.0
transformers==3.3.1
seqeval
fastprogress
attrdict

How to Run

$ python3 run_seq_cls.py --task {$TASK_NAME} --config_file {$CONFIG_FILE}
$ python3 run_seq_cls.py --task nsmc --config_file koelectra-base.json
$ python3 run_seq_cls.py --task kornli --config_file koelectra-base.json
$ python3 run_seq_cls.py --task paws --config_file koelectra-base.json
$ python3 run_seq_cls.py --task question-pair --config_file koelectra-base-v2.json
$ python3 run_seq_cls.py --task korsts --config_file koelectra-small-v2.json
$ python3 run_ner.py --task naver-ner --config_file koelectra-small.json
$ python3 run_squad.py --task korquad --config_file xlm-roberta.json

Result

Base Model

Size NSMC
(acc)		 Naver NER
(F1)		 PAWS
(acc)		 KorNLI
(acc)		 KorSTS
(spearman)		 Question Pair
(acc)		 KorQuaD (Dev)
(EM/F1)		 Korean-Hate-Speech (Dev)
(F1)
KoBERT 351M 89.59 87.92 81.25 79.62 81.59 94.85 51.75 / 79.15 66.21
XLM-Roberta-Base 1.03G 89.03 86.65 82.80 80.23 78.45 93.80 64.70 / 88.94 64.06
HanBERT 614M 90.06 87.70 82.95 80.32 82.73 94.72 78.74 / 92.02 68.32
KoELECTRA-Base 423M 90.33 87.18 81.70 80.64 82.00 93.54 60.86 / 89.28 66.09
KoELECTRA-Base-v2 423M 89.56 87.16 80.70 80.72 82.30 94.85 84.01 / 92.40 67.45
KoELECTRA-Base-v3 431M 90.63 88.11 84.45 82.24 85.53 95.25 84.83 / 93.45 67.61

Small Model

Size NSMC
(acc)		 Naver NER
(F1)		 PAWS
(acc)		 KorNLI
(acc)		 KorSTS
(spearman)		 Question Pair
(acc)		 KorQuaD (Dev)
(EM/F1)		 Korean-Hate-Speech (Dev)
(F1)
DistilKoBERT 108M 88.60 84.65 60.50 72.00 72.59 92.48 54.40 / 77.97 60.72
KoELECTRA-Small 53M 88.83 84.38 73.10 76.45 76.56 93.01 58.04 / 86.76 63.03
KoELECTRA-Small-v2 53M 88.83 85.00 72.35 78.14 77.84 93.27 81.43 / 90.46 60.14
KoELECTRA-Small-v3 54M 89.36 85.40 77.45 78.60 80.79 94.85 82.11 / 91.13 63.07

Reference