Our new X account is live! Follow @wizwand_team for updates
WorkDL logo mark

Instruction Mining: Instruction Data Selection for Tuning Large Language Models

About

Large language models (LLMs) are initially pretrained for broad capabilities and then finetuned with instruction-following datasets to improve their performance in interacting with humans. Despite advances in finetuning, a standardized guideline for selecting high-quality datasets to optimize this process remains elusive. In this paper, we first propose InstructMining, an innovative method designed for automatically selecting premium instruction-following data for finetuning LLMs. Specifically, InstructMining utilizes natural language indicators as a measure of data quality, applying them to evaluate unseen datasets. During experimentation, we discover that double descent phenomenon exists in large language model finetuning. Based on this observation, we further leverage BlendSearch to help find the best subset among the entire dataset (i.e., 2,532 out of 100,000). Experiment results show that InstructMining-7B achieves state-of-the-art performance on two of the most popular benchmarks: LLM-as-a-judge and Huggingface OpenLLM leaderboard.

Yihan Cao, Yanbin Kang, Chi Wang, Lichao Sun• 2023

Related benchmarks

TaskDatasetResultRank
Language ModelingWikiText2
Perplexity12.02
1875
Language ModelingWikiText-2 (test)
PPL17.83
1541
Commonsense ReasoningHellaSwag
Accuracy75.51
1460
Language ModelingWikiText-2
Perplexity (PPL)13.67
841
Commonsense ReasoningWinoGrande
Accuracy68.56
776
Language ModelingPTB
Perplexity21.51
650
Commonsense ReasoningPIQA
Accuracy77.58
647
Language ModelingPTB (test)
Perplexity28.87
471
Question AnsweringOBQA
Accuracy37.9
276
Question AnsweringARC-E
Accuracy66.96
242
Showing 10 of 29 rows

Other info

Follow for update