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

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.

Paper

References (51)

Scroll for more · 38 remaining

Similar papers

Peer review

Reviewer 4rCk5/10 · confidence 4/52024-06-12

Summary

The paper introduces SelectIT, a novel instruction tuning (IT) method that leverages the intrinsic uncertainty in large language models (LLMs) to select high-quality IT data without requiring additional resources. The authors present a curated IT dataset, Selective Alpaca, derived from the Alpaca-GPT4 dataset using SelectIT. The empirical results demonstrate significant improvements in model performance across various tasks. The paper highlights the robustness and efficiency of SelectIT in enhancing the capabilities of LLMs.

Strengths

- The approach of using the intrinsic uncertainty of LLMs for IT data selection is innovative and eliminates the need for external resources, making it cost-effective and widely applicable. - The submission is technically sound, with well-supported claims through theoretical analysis and comprehensive experimental results.

Weaknesses

- Figure 2 is not clear to me. I suggest giving a brief description of each selection stage in the caption for easy understanding. - It seems like the foundational LLM's capacity becomes the key for this data selection. In this case, the selection method may not be as effective as other external resource methods.

Questions

- How does SelectIT perform when applied to instruction tuning datasets other than Alpaca-GPT4? - Can the authors provide more detailed examples or case studies illustrating the differences between data selected by SelectIT and other methods? And how these different influence the final model performance? - How does the performance of SelectIT change with different values of the uncertainty factor (α) ?

Rating

5

Confidence

4

Soundness

3

Presentation

2

Contribution

3

Limitations

Yes

Authorsrebuttal2024-08-10

Look forward to follow-up discussion.

Dear Reviewer 4rCk, We greatly appreciate your valuable feedback and suggestions! To address your concerns, we have run extensive experiments on various datasets and made additional insightful observations. The results will be a nice plus to improve the clarity and useability of our method. As the discussion period is coming to an end, we kindly request that you review these results. If our results and explanations help address your concerns, we would be grateful if you could acknowledge our rebuttal and consider adjusting your score accordingly. Best wishes, The Authors

Authorsrebuttal2024-08-13

Request to review the rebuttal [Author-Reviewer discussion phase ending soon]

Dear Reviewer 4rCk, Thank you for your time reviewing the paper. Your constructive feedback will help improve the quality of our paper. We have also addressed all your concerns in our rebuttal point-by-point. As we are towards the end of the author-reviewer discussion period, we request you to please go through our rebuttal, and we would be more than happy to address any more concerns you have. We thank you again for your time! Best, The Authors

Area Chair UByw2024-08-13

Dear reviewer 4rCk Could you please take a look at the responses of the authors and let us know your thoughts on them? Are you satisfied with the responses and do you have some updates on your comments? Please also take a look at other reviews and share with us your thoughts on whether the paper is ready for publication. We need your response so that we can make an informed decision on this paper. AC

Authorsrebuttal2024-08-14

Response to follow-up questions.

Thank you for your response. We will give a clearer explanation of SelectIT's preference for longer instruction data. As shown in the "Introduction" section and "Insights of Selective Data Curation" section, the core feature of Selective Alpaca is the increase in the proportion of computational data, which is different from other methods. Computational data is often longer than other types of data because it requires more reasoning steps, which further leads to the longer overall length of Selective Alpca. So, "Length" may still be the important and easy-to-get feature for evaluating the quality of sample data [1-4]. Secondly, SelectIT does not simply prefer longer data. As shown in Table 8 of our paper (also the table below), we compared the SelectIT with the previous work [1], which selects samples only by length. The results show that SelectIT can achieve better results with the same amount of data. | | LLaMA-2-7B | LLaMA-2-13B | ALMA-7B | ALMA-13B | |:----------------------:|:----------:|:-----------:|---------|----------| | Alpace-GPT4 | 34.1 | 44.2 | 29.7 | 31.5 | | Alpace-GPT4 + Length | 35.5 | 47.1 | 30.1 | 31.8 | | Alpace-GPT4 + SelectIT | **36.2** | 47.1 | **30.5** | **32.2** | Finally, we further compared the output lengths of models trained on Selective Alpaca and Alpace-GPT4 on the Alpaca-eval benchmark. As shown in the following table, with the output lengths being similar, Selective Alpaca achieves a 1.5% improvement, which effectively eliminates the bias in sample length and explanation of the doubts about the preference of LLMs for samples with longer length. | | Length | Win Rate | |:----------------:|:------:|:----:| | Alpace-GPT4 | 807 | 34.2 | | Selective Alpaca | 815 | 35.7 | Thank you again for taking the time to read our response! We are more than happy to continue the discussion if you are still confused about the paper. [1] Long Is More for Alignment: A Simple but Tough-to-Beat Baseline for Instruction Fine-Tuning. arXiv, 2024. [2] Zero-Shot Generalization during Instruction Tuning: Insights from Similarity and Granularity. arXiv, 2024. [3] LIMA: Less Is More for Alignment. NeurIPS, 2023. [4] AlpaGasus: Training A Better Alpaca with Fewer Data. ICLR, 2024.

Reviewer CKC25/10 · confidence 4/52024-07-12

Summary

In this paper, authors leverage LLMs to score the training data in terms of the token and sentence and model. They train various models on their select models to show the effectiveness on various benchmarks. Their main contribution is to apply foundations to judge the quality of training data from different granularity without any external resources.

Strengths

1. Their idea is easy to understand and the writing is pretty clear to me. 2. They conduct a lot of experiments on several benchmarks to show the effectiveness.

Weaknesses

1. From the table 3, it seems sentence-R works the best so what if we trained model on the data just from the sentence-R? BTW, what is the ID 6, 7, 8 means in that Table? 2. My main question is that do we really need all three types selection models, can one or two of them already achieve great performance? 3. What is the overhead of this data selection model? Like how long does it take to finish the process on 10K dataset.

Questions

Please refer to the above.

Rating

5

Confidence

4

Soundness

3

Presentation

3

Contribution

2

Limitations

Yes

Authorsrebuttal2024-08-10

Look forward to follow-up discussion.

Dear Reviewer CKC2, We greatly appreciate your valuable feedback and suggestions! To address your concerns, we have run extensive experiments on various datasets and made additional insightful observations. The results provide more evidence for our method justification. As the discussion period is coming to an end, we kindly request that you review these results. If our results and explanations help address your concerns, we would be grateful if you could acknowledge our rebuttal and consider adjusting your score accordingly. Best wishes, The Authors

Authorsrebuttal2024-08-13

Request to review the rebuttal [Author-Reviewer discussion phase ending soon]

Dear Reviewer CKC2, Thank you for your time reviewing the paper. Your constructive feedback will help improve the quality of our paper. We have also addressed all your concerns in our rebuttal point-by-point. As we are towards the end of the author-reviewer discussion period, we request you to please go through our rebuttal, and we would be more than happy to address any more concerns you have. We thank you again for your time! Best, The Authors

Area Chair UByw2024-08-13

Dear reviewer CKC2 Could you please take a look at the responses of the authors and let us know your thoughts on them? Are you satisfied with the responses and do you have some updates on your comments? Please also take a look at other reviews and share with us your thoughts on whether the paper is ready for publication. We need your response so that we can make an informed decision on this paper. AC

Authorsrebuttal2024-08-14

Response to follow-up questions.

Thank you for your response. For the Model-R of SelectIT, we used models to jointly determine high-quality data and reasonably weigh the sample quality according to the model parameters, which is similar to ensemble learning. As shown in "Insights of High-Quality Data in SelectIT" section and Figure 5, SelectIT which used multiple models is more conducive to screening data with biased calculation types. So, SelectIT can improve model reasoning ability and further improve the encoding ability of LLM, which ultimately leads to an overall improvement in comprehensive ability. Secondly, MMLU consists of a set of questions about 57 subjects and its multiple-choice format makes it suitable for probing models’ knowledge [1-2]. So it is often used to evaluate the foundation model competence, as LLMs mainly learn world knowledge in the pre-training stage, not the instruction tuning stage (only containing a few instances). So there is a slight improvement on MMLU which is relatively reasonable in SelectIT, and also consistent with the results of AlpaGasus [3]. Thank you again for taking the time to read our response! We are more than happy to continue the discussion if you are still confused about the paper. [1] Measuring Massive Multitask Language Understanding. ICLR, 2021. [2] How Far Can Camels Go? Exploring the State of Instruction Tuning on Open Resources. NeurIPS, 2023. [3] AlpaGasus: Training A Better Alpaca with Fewer Data. ICLR, 2024.

Reviewer 6ofY7/10 · confidence 4/52024-07-15

Summary

This paper presents SelectIT, a new way to improve how large language models (LLMs) follow instructions using their own uncertainty to pick better training data. This method is cost-effective as it doesn't require extra tools or data. The authors developed a new dataset called Selective Alpaca using SelectIT, which significantly improved LLMs' performance. The study suggests that using longer and more detailed data could be more beneficial for training models.

Strengths

SelectIT is a bew and unique method that improves how large language models (LLMs) learn by using the model's own uncertainty to choose better training data. This approach is special because it doesn't need extra tools or data, making it less expensive and easier to use. The paper includes detailed studies that show the method works well and supports its findings with strong evidence. SelectIT significantly improves how models perform compared to older methods, showing that it's a big improvement in teaching models. The researchers made their methods and data public, allowing others to use and test them. This openness helps the scientific community to work together and trus the results more. The paper is also written in a way that's easy to understand, which helps more people learn about SelectIT. It discusses how using longer and more detailed data can make models learn better. Additionally, the paper compares the costs of using SelectIT with using other common tools like the GPT-4 and ChatGPT APIs, showing that SelectIT can save money. This shows that SelectIT is not only effective but also cost-efficient for training models.

Weaknesses

The paper states initially that it does not use extra resources for its new method. However, it mentions later that they used different sizes of LLaMA 2 models to help choose the best instruction data. This contradicts their first claim about not needing extra resources.

Questions

"The paper states initially that it does not use extra resources for its new method. However, it mentions later that they used different sizes of LLaMA 2 models to help choose the best instruction data. This contradicts their first claim about not needing extra resources. " Clarification? Could you consider applying the SelectIT method to larger datasets, such as the Orca dataset for ChatGPT with 5 million samples and the GPT-4 dataset with 1 million samples? This would help to evaluate whether your method scales effectively with increased data volume.

Rating

7

Confidence

4

Soundness

4

Presentation

4

Contribution

3

Limitations

the paper lacks a separate section on limitation of the approach.

Reviewer KKrK6/10 · confidence 4/52024-07-18

Summary

The paper introduces a data selection approach to instruction tuning (IT) for large language models (LLMs) by leveraging the intrinsic uncertainties within the models themselves. This method, termed SelectIT, aims to enhance IT data selection without the need for additional external resources, making it more cost-effective and accessible. SelectIT utilizes three levels of uncertainty within LLMs—token-level, sentence-level, and model-level—to effectively rate and select high-quality IT data. The paper demonstrates the effectiveness of SelectIT through empirical results, showing improvements in model abilities on the open-instruct benchmark and other domain-specific tasks. Additionally, SelectIT is highlighted as a faster and more resource-efficient method compared to traditional approaches, making it a more practical solution for widespread adoption. --------------------------------------------------------------------------------------------------------------------------------------------------------------- Thank you for your reply and I have updated my score accordingly.

Strengths

**Originality**: The paper presents a data selection approach to instruction tuning by leveraging the intrinsic uncertainties within LLMs, without relying on additional fine-tuned models or API calls of ChatGPT or Claude. **Quality**: The methodology has a clear explanation of the multi-granularity self-reflection technique. The empirical results are robust, demonstrating the effectiveness of SelectIT across various benchmarks and tasks. **Clarity**: The paper is clearly written, with well-organized sections and comprehensive explanations of the proposed method and its implementation. **Significance**: SelectIT addresses a critical limitation in the field of instruction tuning by removing the dependence on additional models or data, making the approach more cost-effective and scalable. The curated Selective Alpaca dataset offers performance improvements for LLMs, highlighting the practical impact of the research.

Weaknesses

1. **Reliance on Tuning K**: The method depends on tuning the parameter \( K \), which can significantly affect model performance. This dependency introduces additional complexity and may require extensive experimentation to optimize. 2. **Use of Multiple Base Models**: Although the paper claims not to use external resources, it actually relies on multiple different base models to obtain the model-level score. This reliance contradicts the claim of being purely self-reflective and can be confusing, as the title suggests the method is entirely self-contained. 3. **Sentence-Level Quality Evaluation**: The sentence-level quality evaluation requires multiple rating prompts. It is unclear whether other methods, such as AlpaGasus, would achieve similar performance if they also used multiple rating prompts. This comparison is missing and could be critical for understanding the true advantage of the proposed method. 4. **Single Dataset Evaluation**: All evaluations are conducted on a single training dataset, making it questionable whether this method can be generalized to other datasets, such as WizardLM, or multi-turn data, like the ShareGPT dataset. Testing on a more diverse set of datasets would strengthen the paper's claims of robustness and general applicability.

Questions

(1) How the α = 0.2 in the experiments is selected? (2) Will optimal K be training dataset dependent?

Rating

6

Confidence

4

Soundness

3

Presentation

4

Contribution

2

Limitations

See more details in Weaknesses section.

Authorsrebuttal2024-08-10

Look forward to follow-up discussion.

Dear Reviewer KKrK, We greatly appreciate your valuable feedback and suggestions! To address your concerns, we have run extensive experiments on various datasets and made additional insightful observations. The results further confirm the generality of our proposed method. As the discussion period is coming to an end, we kindly request that you review these results. If our results and explanations help address your concerns, we would be grateful if you could acknowledge our rebuttal and consider adjusting your score accordingly. Best wishes, The Authors

Authorsrebuttal2024-08-13

Request to review the rebuttal [Author-Reviewer discussion phase ending soon]

Dear Reviewer KKrK, Thank you for your time reviewing the paper. Your constructive feedback will help improve the quality of our paper. We have also addressed all your concerns in our rebuttal point-by-point. As we are towards the end of the author-reviewer discussion period, we request you to please go through our rebuttal, and we would be more than happy to address any more concerns you have. We thank you again for your time! Best, The Authors

Area Chair UByw2024-08-13

Dear reviewer KKrK Could you please take a look at the responses of the authors and let us know your thoughts on them? Are you satisfied with the responses and do you have some updates on your comments? Please also take a look at other reviews and share with us your thoughts on whether the paper is ready for publication. We need your response so that we can make an informed decision on this paper. AC

Reviewer 6ofY2024-08-09

Thank you for addressing most of my concerns. I have increased the OA score.

Reviewer 4rCk2024-08-13

Thanks for your response. Regarding the 'Length' of the selected data, I cannot be convinced that one of the primary contributions is that a longer instruction indicates better quality. LLMs have a verbosity bias, preferring longer output even when it's not necessarily clearer [1]. Therefore, I will maintain my score. [1] ZHng et al. Judging LLM-as-a-Judge with MT-Bench and Chatbot Arena

Reviewer CKC22024-08-13

Thanks for the response and I am thinking maybe we still need to understand why combination of models can yield better results on some benchmarks, for example, we do not see meaningful improvement on MMLU. I will keep my score. thanks!

Program Chairsdecision2024-09-25

Decision

Accept (poster)

© 2026 NYSGPT2525 LLC