标签:

• bert
• mobilebert
• oBERT 语言: 英文 数据集: squad

mobilebert-uncased-finetuned-squadv1

该模型是基于mobilebert-uncased模型在SQuADv1任务上微调的版本。
为确保这个在TPU上训练的模型能在GPU的PyTorch环境中稳定运行，原始模型在微调SQuADv1任务前，额外以关闭dropout的方式在BookCorpus和英文维基百科上进行了1个epoch的预训练。

此模型是论文The Optimal BERT Surgeon: Scalable and Accurate Second-Order Pruning for Large Language Models研究工作的成果之一。

SQuADv1开发集表现：

精确匹配率(EM) = 83.96
模糊匹配率(F1) = 90.90

代码库：https://github.com/neuralmagic/sparseml/tree/main/research/optimal_BERT_surgeon_oBERT

如果您认为该模型有帮助，请考虑引用我们的工作。

引用信息

@article{kurtic2022optimal,
  title={The Optimal BERT Surgeon: Scalable and Accurate Second-Order Pruning for Large Language Models},
  author={Kurtic, Eldar and Campos, Daniel and Nguyen, Tuan and Frantar, Elias and Kurtz, Mark and Fineran, Benjamin and Goin, Michael and Alistarh, Dan},
  journal={arXiv preprint arXiv:2203.07259},
  year={2022}
}