arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2311.14736cs.CLcs.LG

数据多样性对稳健指令微调至关重要

Data Diversity Matters for Robust Instruction Tuning

Georgia Institute of Technology(佐治亚理工学院);Amazon(亚马逊公司)

Alexander Bukharin, Shiyang Li, Zhengyang Wang, Jingfeng Yang, Bing Yin, Xian Li, Chao Zhang, Tuo Zhao, Haoming Jiang

AI总结:

本文提出质量-多样性指令微调(QDIT)算法,同时控制数据多样性与质量,发现多样性提升可显著增强最差情况指令跟随性能,从而提升稳健性。

摘要:

近期研究表明,通过策划高质量且多样化的指令微调数据集,可以显著提升指令跟随能力。然而,创建此类数据集困难重重,多数工作依赖人工筛选或专有语言模型。自动数据策划面临挑战,因为目前尚不清楚如何定义指令微调中的多样性、多样性与质量如何相互依赖,以及如何优化数据集的质量与多样性。为解决这些问题,我们提出了一种新算法——质量-多样性指令微调(QDIT)。QDIT提供了一种简单方法,可同时控制数据集的多样性与质量,使我们能够深入研究多样性和质量对指令微调性能的影响。从这项研究中,我们得出两个关键见解:(1)数据多样性与质量之间存在自然的权衡;(2)增加数据多样性显著提升了最差情况下的指令跟随性能,从而增强了稳健性。我们在多个大规模指令微调数据集上验证了QDIT的性能,发现与质量驱动的数据选择相比,它能显著改善最差情况和平均情况下的性能。

英文摘要:

Recent works have shown that by curating high quality and diverse instruction tuning datasets, we can significantly improve instruction-following capabilities. However, creating such datasets is difficult and most works rely on manual curation or proprietary language models. Automatic data curation is difficult as it is still not clear how we can define diversity for instruction tuning, how diversity and quality depend on one other, and how we can optimize dataset quality and diversity. To resolve these issue, we propose a new algorithm, Quality-Diversity Instruction Tuning (QDIT). QDIT provides a simple method to simultaneously control dataset diversity and quality, allowing us to conduct an in-depth study on the effect of diversity and quality on instruction tuning performance. From this study we draw two key insights (1) there is a natural tradeoff between data diversity and quality and (2) increasing data diversity significantly improves the worst case instruction following performance, therefore improving robustness. We validate the performance of QDIT on several large scale instruction tuning datasets, where we find it can substantially improve worst and average case performance compared to quality-driven data selection.

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

↑