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

基于 Transformer 的因果语言模型执行聚类

Transformer-based Causal Language Models Perform Clustering

University of Illinois Urbana-Champaign(伊利诺伊大学厄巴纳-香槟分校)

Xinbo Wu, Lav R. Varshney

AI总结:

本文通过简化指令遵循任务和合成数据分析 Transformer 因果语言模型,揭示其在隐藏空间中动态聚类数据以学习任务信息,并探讨该机制对泛化、预训练与对齐的启示。

摘要:

尽管大型语言模型(LLM)已在解决各类自然语言任务方面展现出卓越能力,但 LLM 遵循人类指令的能力仍令人担忧。近期研究表明,通过针对指令遵循任务进行额外训练,指令遵循能力获得了显著提升。然而,促成有效指令遵循能力的机制仍未得到充分理解。为此,我们引入一个简化的指令遵循任务,并使用合成数据集分析基于 Transformer 的因果语言模型。研究发现表明,该模型通过在其隐藏空间中对数据进行聚类来学习任务特定信息,而且这一聚类过程会在学习期间动态演化。我们还展示了该现象如何帮助模型处理未见过的实例,并在更贴近现实的设置中验证了结果。此外,我们提出了由此获得启发的、关于预训练与对齐的应用。

英文摘要:

Even though large language models (LLMs) have demonstrated remarkable capability in solving various natural language tasks, the capability of an LLM to follow human instructions is still a concern. Recent works have shown great improvements in the instruction-following capability via additional training for instruction-following tasks. However, the mechanisms responsible for effective instruction-following capabilities remain inadequately understood. Here, we introduce a simplified instruction-following task and use synthetic datasets to analyze a Transformer-based causal language model. Our findings suggest that the model learns task-specific information by clustering data within its hidden space, with this clustering process evolving dynamically during learning. We also demonstrate how this phenomenon assists the model in handling unseen instances, and validate our results in a more realistic setting. Furthermore, we present inspired applications regarding pre-training and alignment.

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

↑