From c3427e0baba8d425e9a339d4f9b7722967e3fc42 Mon Sep 17 00:00:00 2001 From: FeiyuXiao Date: Sun, 2 Aug 2020 22:22:46 +0800 Subject: [PATCH] change file --- DistillBert/FastBert/README.md | 67 ---------------------------------- 1 file changed, 67 deletions(-) diff --git a/DistillBert/FastBert/README.md b/DistillBert/FastBert/README.md index 7b89653..bb2e8e6 100644 --- a/DistillBert/FastBert/README.md +++ b/DistillBert/FastBert/README.md @@ -1,70 +1,3 @@ ### FastBert, 复现ACL2020论文 [FastBERT: a Self-distilling BERT with Adaptive Inference Time](https://arxiv.org/pdf/2004.02178.pdf) -### 简介 - 相比一众BERT蒸馏方法,FastBERT的蒸馏过程和具体任务一起进行, 且主干BERT网络不变,有如下优点: - 1. 准确率损失较小, 加速比约1~10倍, - 2. 和BERT预训练模型兼容,可灵活替换各种预训练模型,如ernie - 3. 使用简单, 较为实用 - 适用于文本分类任务,想要用BERT提升效果但受限于机器资源的场景 - -### 使用方法 - 下载pretrain的bert模型: -
- [bert中文(微云)](https://share.weiyun.com/5goxygS),[bert中文(百度网盘),提取码:mxfb](https://pan.baidu.com/s/15YpZB7uxuiCjEFZh1zx38w) - 放至目录./pretrained_model/bert-chinese/bert-pytorch-google/下 -
- [百度ernie(微云)](https://share.weiyun.com/5rYpEBs),[百度ernie(百度网盘),提取码:vui3](https://pan.baidu.com/s/1nD_hIYfis6Qn2JTXvX7dGg) - 放至目录./pretrained_model/ernie/ERNIE_stable-1.0.1-pytorch/下 -```bash - 1. 初始训练: - sh run_scripts/script_train_stage0.sh - - 2. 蒸馏训练: - sh run_scripts/script_train_stage1.sh - **注意** :蒸馏阶段输入数据为无监督数据,可依据需要引入更多数据提升鲁棒性 - - 3. 推理: - sh run_scripts/script_infer.sh - 其中 inference_speed参数(0.0~1.0)控制加速程度 - - 4. 部署使用 - python3 predict.py -``` - -```bash -若想替换为ernie模型,只需要把 ---model_config_file='config/fastbert_cls.json' -替换为 ---model_config_file='config/fastbert_cls_ernie.json' -``` - -### 模型思路 -##### 加速思路 - 1. 每一层transformer都接一个子分类器 - 2. 根据样本输入,自适应12层transformer的推理深度,子分类器置信度高则提前返回 -##### 训练思路 - 1. load 预训练好的bert - 2. 跟普通训练一样finetune模型 - 3. freeze主干网络和最后层的teacher分类器,每层的子模型拟合teacher分类器(KL散度为loss) - 4. inference阶段,根据样本输入,子分类器置信度高则提前返回 - -arch - -### 结果对比 -##### 论文结果: -compare - -##### 实测结果: -```bash -ChnSentiCorp: -speed_arg:0.0, time_per_record:0.14725032741416672, acc:0.9400, 基准 -speed_arg:0.1, time_per_record:0.10302954971909761, acc:0.9420, 1.42倍 -speed_arg:0.5, time_per_record:0.03420266199111938, acc:0.9340, 4.29倍 -speed_arg:0.8, time_per_record:0.019530397139952513, acc:0.9160, 7.54倍 -注:speed=0.1的情况下比基准的准确率还高,是有可能的,正则之类的效应 -``` - -1. 对于实际场景任务中,测试了语义理解要求较高的模型,经测试在加速1~6倍时,精度和原始BERT差距很小 -2. 对于稀疏场景模型,线上正例占比很小,则可进一步提高加速比, 某模型负例约17%走到第12层,正例约81%走到第12层 -