实验指标测试和记录

July 11, 2024 · View on GitHub

1. 文本分类

1.1 不同预训练模型的指标对比

solutionepochvalid_acctest_acccomment
albert_small10/1094.4693.98small版本
bert6/1094.7294.11——
robert4/1094.7794.64——
nezha7/1095.0794.72——
xlnet6/1095.0794.77——
electra10/1094.9494.78——
roformer9/1094.8594.42——
roformer_v23/1095.7896.09——
gau_alpha2/1095.2594.46——
deberta_v2(10/2/5)/1094.55/95.16/94.8594.99/94.68/94.33分别为97M/320M/710M, 97M的transformer版本指标为94.90/94.81

1.2 不同trick下的指标对比

solutionepochvalid_acctest_acccomment
bert10/1094.9094.78——
fgm4/1095.3494.99——
pgd6/1095.3494.64——
gradient_penalty7/1095.0794.81——
vat8/1095.2195.03——
ema7/1095.2194.86——
ema+warmup7/1095.5195.12——
mix_up6/1095.1294.42——
R-drop9/1095.2594.94——
UDA8/1094.9095.56——
semi-vat10/1095.3495.38——
temporal_ensembling8/1094.9494.90——

2. 序列标注

solutionepochf1_tokenf1_entitycomment
bert+crf18/2096.8996.05——
bert+crf+init18/2096.9396.08用训练数据初始化crf权重
bert+crf+freeze11/2096.8996.13用训练数据生成crf权重(不训练)
bert+cascade+crf5/2098.1096.26crf类别少所以f1_token偏高
bert+crf+posseg13/2097.3296.55加了词性输入
bert+global_pointer18/20——95.66——
bert+efficient_global_pointer17/20——96.55——
bert+mrc7/20——95.75——
bert+span13/20——96.31——
bert+tplinker_plus20/20——95.71长度限制明显
uie20/20——96.57zeroshot:f1=60.8, fewshot-100样本:f1=85.82, 200样本:f1=86.40
W2NER18/2097.3796.32对显存要求较高
CNN_Nested_NER19/2098.0696.11
LEAR8/20——96.52

3. 文本表示

3.1 无监督语义相似度

  • bert预训练模型 + 无监督finetune + cls位句向量(PromptBert除外)
  • 五个中文数据集 + 5个epoch取最优值 + valid的spearmanr相关系数
  • 继续finetune, 部分数据集有小幅提升
  • 实验显示dropout_rate对结果影响较大
solutionATECBQLCQMCPAWSXSTS-Bcomment
Bert-whitening26.7931.8156.3417.2267.45cls+不降维
CT30.6544.5068.6716.2069.27dropout=0.1, 收敛慢跑了10个epoch
CT_In_Batch_Neg32.4747.0968.5627.5074.00dropout=0.1
TSDAE——46.6565.3012.54——dropout=0.1, ——表示该指标异常未记录
SimCSE33.9050.2971.8113.1471.09dropout=0.3
ESimCSE34.0550.5471.5812.5371.27dropout=0.3
DiffSCE33.0448.1771.5112.9171.10dropout=0.3, 没啥效果
PromptBert33.9849.8973.1813.3073.42dropout=0.3

3.2 有监督语义相似度

  • bert预训练模型 + 训练数据finetune + cls位句向量
  • 五个中文数据集 + 5个epoch取最优值 + valid/test的spearmanr相关系数
  • STS-B任务是5分类,其余是2分类
solutionATECBQLCQMCPAWSXSTS-Bcomment
CoSENT50.61 / 49.8172.84 / 71.6177.79 / 78.7455.00 / 56.0083.48 / 80.06
ContrastiveLoss50.02 / 49.1972.52 / 70.9877.49 / 78.2758.21 / 57.6569.87 / 68.58STS-B转为2分类
InfoNCE47.77 / 46.9969.86 / 68.1471.74 / 74.5452.82 / 54.2183.31 / 78.72STS-B转为2分类
concat CrossEntropy48.71 / 47.6272.16 / 70.0778.44 / 78.7751.46 / 52.2861.31 / 56.62STS-B转为2分类
CosineMSELoss46.89 / 45.8672.27 / 71.3575.29 / 77.1954.92 / 54.3581.64 / 77.76STS-B标准化到0-1

4. 关系提取

solutionf1comment
CasRel81.87
gplinker82.38
tplinker74.49seq_len=64, 未完全收敛
tplinker_plus79.30seq_len=64
SPN4RE77.53
PRGC80.36训练很慢

5. 文本生成

  • CSL数据集,注意是训练集1万左右的版本,分别dev/test指标
solutionRouge-LRouge-1Rouge-2BLEUcomment
bert+unlim63.65 / 63.0166.25 / 66.3454.48 / 54.8144.21 / 44.60
bart64.62 / 64.9967.72 / 68.4056.08 / 57.2646.15 / 47.67
mt567.67 / 65.9870.39 / 69.3659.60 / 59.0550.34 / 50.11
t5_pegasus66.07 / 66.1168.94 / 69.6157.12 / 58.3846.14 / 47.95
uer_t563.59 / 63.1166.56 / 66.4854.65 / 54.8244.27 / 44.60

6. 大模型指令微调

chatglmgpuTime/epoch(s)Rouge-LRouge-1Rouge-2BLEUcomment
hf+pt2 official+v100-int4-bs1————24.9731.127.118.10
hf+pt2 reappear+v100-int4-bs1————24.8030.976.987.85
b4t+pt2+v100+int4+bs1————24.5830.767.128.12
b4t+pt2+T4-int8-bs110G147024.8730.837.148.05
b4t+pt2+A100(pcie 40G)-fp16-bs115G28725.1031.437.308.28
b4t+pt2+A100(pcie 40G)-fp16-bs822G70525.2231.227.388.35
b4t+pt2+A100(pcie 40G)-fp32-bs129G76024.8330.957.188.08
b4t+pt2+A100(pcie 40G)-fp32-bs432G260025.1231.557.218.02
b4t+lora+V100-fp16-bs1628G257024.8931.387.178.15
b4t+qlora+V100-bs1626G538123.9929.526.477.74