察元AI教程库 · 文章 · 大模型

通过自监督早期退出加速大语言模型推理

本文提出了一种模块化方法,通过在变换器层间的中间层添加早期退出头来加速大型语言模型(LLMs)的推理过程。每个头以自监督方式训练,以模仿主模型的预测,当经过校准的阈值被满足时,计算过程可以提前终止

作者:Florian Valade · 语言:en

前往原址学习

简介

本文提出了一种模块化方法,通过在变换器层间的中间层添加早期退出头来加速大型语言模型(LLMs)的推理过程。每个头以自监督方式训练,以模仿主模型的预测,当经过校准的阈值被满足时,计算过程可以提前终止

返回教程库 · 察元AI产品家族