网站设计是什么意思,中信建设有限责任公司海南分公司,网站模板破解版,苏州网站建设联系苏州梦易行本文是LLM系列文章#xff0c;针对《NarrowBERT: Accelerating Masked Language Model Pretraining and Inference》的翻译。 NarrowBERT#xff1a;加速掩蔽语言模型的预训练和推理 摘要1 引言2 NarrowBERT3 实验4 讨论与结论局限性 摘要
大规模语言模型预训练是自然语言处…本文是LLM系列文章针对《NarrowBERT: Accelerating Masked Language Model Pretraining and Inference》的翻译。 NarrowBERT加速掩蔽语言模型的预训练和推理 摘要1 引言2 NarrowBERT3 实验4 讨论与结论局限性 摘要
大规模语言模型预训练是自然语言处理中一种非常成功的自监督学习形式但随着时间的推移模型和预训练语料库变得越来越大执行成本越来越高。我们提出了NarrowBERT这是一种改进的transformer编码器它将掩蔽语言模型预训练的吞吐量提高了2倍以上。NarrowBERT稀疏transformer模型使得自注意查询和前馈层在预训练期间仅对每个句子的掩蔽标记进行操作而不是像通常的transformer编码器那样对所有标记进行操作。我们还表明在MNLI等句子编码任务上NarrowBERT在推理时的吞吐量增加了3.5倍性能下降最小或没有。最后我们检查了NarrowBERT在IMDB和Amazon评论分类和CoNLL NER任务上的性能并表明它也与标准BERT性能相当。
1 引言
2 NarrowBERT
3 实验
4 讨论与结论
我们已经探索了在掩蔽语言模型损失计算中利用稀疏性的两种直接方法重新排列transformer编码器的层以允许前馈组件避免在非掩蔽位置上进行计算以及在注意力机制中稀疏查询以仅将掩蔽位置上下文化。NarrowBERT变体可以将训练速度提高约2倍将推理速度提高约3倍同时在GLUE、IMDB、Amazon和CoNLL NER任务上保持非常相似的性能。基于第3节中速度和性能之间的有利权衡我们建议从业者在缩小范围之前考虑使用具有2或3层的SparseQueries NarrowBERT模型。
局限性
由于我们的预算限制我们只对基本尺寸的transformer模型进行了预训练和下游实验。我们也只应用了掩蔽语言建模目标但还有其他有效的预训练目标。尽管如此由于我们在架构中引入了最小的更改我们希望后续工作将从我们的缩小操作中受益并进行更广泛的预训练和下游实验。虽然预训练的模型可以应用于更多的下游任务但我们在这项工作中设计了一个合理的任务集包括GLUE句子分类和CoNLL-NER顺序分类任务。
