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

在离散潜在空间中进行下一步概念预测可使语言模型更强大

Next Concept Prediction in Discrete Latent Space Leads to Stronger Language Models

LUMIA Lab(LUMIA实验室);School of Artificial Intelligence(人工智能学院);Shanghai Jiao Tong University(上海交通大学);Shanghai AI Laboratory(上海人工智能实验室);Department of Electronic Engineering(电子工程系);College of Artificial Intelligence(人工智能学院);Tsinghua University(清华大学);Shanghai Innovation Institute(上海创新研究院)

Yuliang Liu, Yunchong Song, Yixuan Wang, Kewen Ge, Alex Lamb, Qipeng Guo, Kai Chen, Bowen Zhou, Zhouhan Lin

AI总结:

通过在离散潜在空间中进行下一步概念预测,ConceptLM在预训练任务中实现了更强大的语言模型性能。

摘要:

我们提出了Next Concept Prediction(NCP),一种基于Next Token Prediction(NTP)的生成预训练范式。NCP预测跨越多个token的概念,从而形成更具有挑战性的预训练目标。我们的模型ConceptLM通过向量量化对隐藏状态进行量化,并构建了一个概念词汇表。它利用NCP和NTP来驱动参数更新,并生成一个概念以指导后续token的生成。我们从70M到1.5B参数的规模上从头开始训练ConceptLM,训练数据高达300B,包括Pythia和GPT-2的基座模型。在13个基准测试中,结果表明NCP在传统token级模型上实现了一致的性能提升。此外,对一个8B参数Llama模型的持续预训练实验表明,NCP可以进一步改进已训练的NTP模型。我们的分析表明,通过引入更困难的预训练任务,NCP能够产生更强大的语言模型,为更好的语言建模提供了一条有希望的路径。

英文摘要:

We propose Next Concept Prediction (NCP), a generative pretraining paradigm built on top of Next Token Prediction (NTP). NCP predicts discrete concepts that span multiple tokens, thereby forming a more challenging pretraining objective. Our model, ConceptLM, quantizes hidden states using Vector Quantization and constructs a concept vocabulary. It leverages both NCP and NTP to drive parameter updates and generates a concept to guide the generation of the following tokens. We train ConceptLM from scratch at scales ranging from 70M to 1.5B parameters with up to 300B training data, including Pythia and GPT-2 backbones. Results on 13 benchmarks show that NCP yields consistent performance gains over traditional token-level models. Furthermore, continual pretraining experiments on an 8B-parameter Llama model indicate that NCP can further improve an NTP-trained model. Our analysis suggests that NCP leads to more powerful language models by introducing a harder pretraining task, providing a promising path toward better language modeling.

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

↑