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

将指令微调与预训练对齐

Aligning Instruction Tuning with Pre-training

School of Artificial Intelligence, University of Chinese Academy of Sciences(中国科学院大学人工智能学院);Institute of Automation, Chinese Academy of Sciences(中国科学院自动化研究所);Peking University(北京大学)
Yiming Liang, Tianyu Zheng, Xinrun Du, Ge Zhang, Jiaheng Liu, Xingwei Qu, Wenqiang Zu, Xingrun Xing, Chujie Zheng, Lei Ma, Guoyin Wang, Zhaoxiang Zhang, Wenhao … 展开作者

Yiming Liang, Tianyu Zheng, Xinrun Du, Ge Zhang, Jiaheng Liu, Xingwei Qu, Wenqiang Zu, Xingrun Xing, Chujie Zheng, Lei Ma, Guoyin Wang, Zhaoxiang Zhang, Wenhao Huang, Xiang Yue, Jiajun Zhang

AI总结:

本文提出AITP方法,通过识别指令微调数据集的覆盖不足并将预训练数据重写为指令-响应对,弥合指令微调与预训练分布的差距,在8个基准上对3个开源LLMs的评估证明其持续提升了模型性能。

摘要:

指令微调通过依赖高质量数据集来引导行为,增强了大型语言模型(LLMs)遵循人类指令完成多样任务的能力。然而,这些数据集无论是人工整理还是合成生成,往往范围狭窄,且与预训练期间捕获的广泛分布不一致,从而限制了LLMs的泛化能力及对预训练知识的有效利用。我们提出了将指令微调与预训练对齐(AITP)方法,该方法通过识别指令微调数据集中的覆盖不足,并将代表性不足的预训练数据重写为高质量的指令-响应对来弥合这一差距。该方法在丰富数据集多样性的同时保留了特定任务目标。在8个基准上对3个完全开源的LLMs进行的评估表明,AITP带来了持续的性能提升。消融实验突出了自适应数据选择、受控重写和平衡整合的优势,强调了将指令微调与预训练分布对齐以释放LLMs全部潜力的重要性。

英文摘要:

Instruction tuning enhances large language models (LLMs) to follow human instructions across diverse tasks, relying on high-quality datasets to guide behavior. However, these datasets, whether manually curated or synthetically generated, are often narrowly focused and misaligned with the broad distributions captured during pre-training, limiting LLM generalization and effective use of pre-trained knowledge. We propose Aligning Instruction Tuning with Pre-training (AITP), a method that bridges this gap by identifying coverage shortfalls in instruction-tuning datasets and rewriting underrepresented pre-training data into high-quality instruction-response pairs. This approach enriches dataset diversity while preserving task-specific objectives. Evaluations on three fully open LLMs across eight benchmarks demonstrate consistent performance improvements with AITP. Ablations highlight the benefits of adaptive data selection, controlled rewriting, and balanced integration, emphasizing the importance of aligning instruction tuning with pre-training distributions to unlock the full potential of LLMs.

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

↑