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

SelectIT:通过不确定性感知自反思实现大语言模型的选择性指令微调

SelectIT: Selective Instruction Tuning for LLMs via Uncertainty-Aware Self-Reflection

Institute of Computing and Intelligence, Harbin Institute of Technology(哈尔滨工业大学计算与智能学院);University of Macau(澳门大学)

Liangxin Liu, Xuebo Liu, Derek F. Wong, Dongfang Li, Ziyi Wang, Baotian Hu, Min Zhang

AI总结:

本研究针对现有指令微调数据筛选依赖额外资源的问题,提出利用大语言模型内在不确定性进行数据筛选的SelectIT方法,构建了Selective Alpaca数据集,证实其能有效提升模型性能且鲁棒性良好。

摘要:

指令微调(Instruction Tuning,IT)对让大语言模型(large language models,LLMs)适配以人为本的交互至关重要。近期研究进展表明,精心挑选一小部分高质量指令微调数据能显著提升大语言模型的性能。尽管如此,常见方法往往依赖额外模型或数据,这增加了成本,限制了广泛应用。在本研究中,我们提出了一种名为SelectIT的新方法,该方法利用大语言模型自身的基础能力。具体而言,我们利用大语言模型中存在的内在不确定性,更有效地筛选高质量指令微调数据,无需额外资源。此外,我们构建了一个经过筛选的指令微调数据集Selective Alpaca,它是将SelectIT应用于Alpaca-GPT4数据集得到的。实验结果表明,使用Selective Alpaca进行指令微调能大幅提升模型能力。SelectIT的鲁棒性也在多种基础模型和特定领域任务中得到了验证。我们的研究结果表明,更长、计算量更大的指令微调数据可能是更优质的指令微调数据来源,这为该领域的未来研究提供了宝贵见解。数据、代码和脚本可在https://github.com/Blue-Raincoat/SelectIT免费获取。

英文摘要:

Instruction tuning (IT) is crucial to tailoring large language models (LLMs) towards human-centric interactions. Recent advancements have shown that the careful selection of a small, high-quality subset of IT data can significantly enhance the performance of LLMs. Despite this, common approaches often rely on additional models or data, which increases costs and limits widespread adoption. In this work, we propose a novel approach, termed SelectIT, that capitalizes on the foundational capabilities of the LLM itself. Specifically, we exploit the intrinsic uncertainty present in LLMs to more effectively select high-quality IT data, without the need for extra resources. Furthermore, we introduce a curated IT dataset, the Selective Alpaca, created by applying SelectIT to the Alpaca-GPT4 dataset. Empirical results demonstrate that IT using Selective Alpaca leads to substantial model ability enhancement. The robustness of SelectIT has also been corroborated in various foundation models and domain-specific tasks. Our findings suggest that longer and more computationally intensive IT data may serve as superior sources of IT, offering valuable insights for future research in this area. Data, code, and scripts are freely available at https://github.com/Blue-Raincoat/SelectIT.

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

↑