在数据增强、蒸馏剪枝下ERNIE3.0分类模型性能提升

46 阅读1分钟

在数据增强、蒸馏剪枝下ERNIE3.0模型性能提升

项目链接:
aistudio.baidu.com/aistudio/pr…

以CBLUE数据集中医疗搜索检索词意图分类为例:

本项目首先讲解了数据增强和数据蒸馏的方案,并在后面章节进行效果展示,结果预览:

模型ACCPrecisionRecallF1average_of_acc_and_f1
ERNIE 3.0 Base0.802550.93171470.9082840.9198500.86120
ERNIE 3.0 Base+数据增强0.79795390.9010040.928990.914780.8563
ERNIE 3.0 Base+剪裁保留比0.50.798460.9512570.894970.922250.8603
ERNIE 3.0 Base +剪裁保留比2/30.80920710.94153840.9053250.9230760.86614