大语言模型的训练方法及装置

本说明书实施例披露一种大语言模型的训练方法及装置。该训练方法包括：首先，在大语言模型的目标注意力头中，分别利用查询变换矩阵、键变换矩阵和值变换矩阵处理上一层针对输入文本的输出，得到查询矩阵、键矩阵和值矩阵；然后，将所述查询矩阵压缩为全局查询向量，并对所述全局查询向量与所述键矩阵中的多个键向量分别进行融合处理，得到注意力矩阵；接着，将所述注意力矩阵压缩为全局注意力向量，并通过对所述全局注意力向量与所述值矩阵中的多个值向量分别进行融合处理，得到所述目标注意力头的输出，以确定所述大语言模型的目标输出；之后，基于所述目标输出，训练所述大语言模型。