TensorX
返回文献探索

Paper · arXiv 2307.06290

Instruction Mining: High-Quality Instruction Data Selection for Large Language Models

Yihan Cao, Yanbin Kang, Lichao Sun

10 upvotesJuly 12, 2023arXiv 预印本
AI 摘要

InstructMining, a linear rule using natural language indicators, effectively selects high-quality instruction-following data for enhancing language model finetuning performance.

large language modelspretrainingfinetuninginstruction finetuninginstruction-following datanatural language indicators

Abstract

Large language models typically undergo two training stages, pretraining and finetuning. Despite that large-scale pretraining endows the model with strong capabilities to generate natural language responses, these pretrained models can still fail to understand human instructions at times. To enhance language models' ability of interpreting and responding to instructions, instruction finetuning has emerged as a critical method in this area. Recent studies found that large language models can be finetuned to perform well even with a small amount of high-quality instruction-following data. However, the selection of high-quality datasets for finetuning language models still lacks clear guidelines to follow. In this paper, we propose InstructMining, a linear rule for evaluating instruction-following data quality. We formulate InstructMining using specific natural language indicators. To investigate the relationship between data quality and these indicators, we further conduct extensive finetuning experiments. The experiment results are then applied to estimating parameters in InstructMining. To further investigate its performance, we use InstructMining to select high-quality data from unseen datasets. Results demonstrate that InstructMining can help select relatively high-quality samples from various instruction-following datasets. Compared to models finetuned on unfiltered datasets, models finetuned on InstructMining selected datasets perform better on 42.5% cases.

北京市昌平区探索星信息技术及软件开发工作室

京ICP备2026059466号
Instruction Mining: High-Quality Instruction Data Selection for Large Language Models | TensorX