GLUE tasks
March 5, 2023 ยท View on GitHub
Here are the corresponding GLUE scores on BiGS.
Experiments
GLUE is made up of a total of 9 different tasks. Here is how to run the script on one of them:
We finetune BiGS on TPU-v3 with 8 cores. Since the batch size per device is 2, the total number of batch size is 16.
export TASK_NAME=cola
python run_glue.py \
--model_name_or_path JunxiongWang/BiGS_512 \
--task_name $TASK_NAME \
--max_seq_length 512 \
--learning_rate 2e-5 \
--num_train_epochs 3 \
--per_device_train_batch_size 2 \
--logging_steps 100 \
--eval_steps 500 \
--weight_decay 0.01 \
--output_dir BiGS_$TASK_NAME/
Those give us the following result
| Task | Metric | Result |
|---|---|---|
| CoLA | Matthews corr | 67.9 |
| SST-2 | Accuracy | 93.8 |
| QQP | Accuracy/F1 | 91.4/88.4 |
| MNLI | Matched acc./Mismatched acc. | 86.5 |
| QNLI | Accuracy | 91.6 |
For MRPC, STS-B and RTE, we finetune on the MNLI model
export TASK_NAME=mrpc
python run_glue.py \
--model_name_or_path JunxiongWang/BiGS_512_MNLI \
--task_name $TASK_NAME \
--max_seq_length 512 \
--learning_rate 2e-5 \
--num_train_epochs 3 \
--per_device_train_batch_size 2 \
--logging_steps 100 \
--eval_steps 500 \
--weight_decay 0.01 \
--output_dir BiGS_$TASK_NAME/
| Task | Metric | Result |
|---|---|---|
| MRPC | F1/Accuracy | 88.3/83.3 |
| STS-B | Pearson/Spearman corr. | 89.9/89.7 |
| RTE | Accuracy | 79.8 |