arXivDaily arXiv每日学术速递 周一至周五更新
arXiv周末暂无论文更新,休息一下吧,周末愉快~~
arXiv 2403.09024cs.CLcs.AI
收藏

半参数令牌-序列协同监督

Semiparametric Token-Sequence Co-Supervision

KAIST AI(韩国科学技术院人工智能学院);Seoul National University(首尔大学);University of Washington(华盛顿大学);Kakao Brain

Hyunji Lee, Doyoung Kim, Jihoon Jun, Sejune Joo, Joel Jang, Kyoung-Woon On, Minjoon Seo

AI总结:

提出半参数令牌-序列协同监督训练方法,结合参数化令牌空间与非参数序列空间的损失,提升语言模型泛化能力与稳定性。

摘要:

在这项工作中,我们引入了一种半参数令牌-序列协同监督训练方法。该方法通过同时利用来自传统下一令牌预测损失(在参数化令牌嵌入空间上计算)和下一序列预测损失(在非参数序列嵌入空间上计算)的监督来训练语言模型。非参数序列嵌入空间由另一个语言模型构建,该模型的任务是将输入文本压缩为单个代表性嵌入。我们的实验表明,通过两种监督训练出的模型始终优于通过每种监督独立训练出的模型。分析表明,这种协同监督促进了模型更广泛的泛化能力。特别是,在预训练阶段建立的参数化令牌空间的鲁棒性往往能有效增强由另一个语言模型建立的非参数序列嵌入空间的稳定性。

英文摘要:

In this work, we introduce a semiparametric token-sequence co-supervision training method. It trains a language model by simultaneously leveraging supervision from the traditional next token prediction loss which is calculated over the parametric token embedding space and the next sequence prediction loss which is calculated over the nonparametric sequence embedding space. The nonparametric sequence embedding space is constructed by a separate language model tasked to condense an input text into a single representative embedding. Our experiments demonstrate that a model trained via both supervisions consistently surpasses models trained via each supervision independently. Analysis suggests that this co-supervision encourages a broader generalization capability across the model. Especially, the robustness of parametric token space which is established during the pretraining step tends to effectively enhance the stability of nonparametric sequence embedding space, a new space established by another language model.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑