Clustering and Ranking: Diversity-preserved Instruction Selection through Expert-aligned Quality Estimation

Yuan Ge, Yilun Liu, Chi Hu, Weibin Meng, Shimin Tao +8 more
2/28/2024
cs.CL

No official code found. Best implementation: IronBeliever/CaR (91 stars, last commit 2024-11-13).

Abstract

With contributions from the open-source community, a vast amount of instruction tuning (IT) data has emerged. Given the significant resource allocation required for training and evaluating models, it is advantageous to have an efficient method for selecting high-quality IT data. However, existing methods for instruction data selection have limitations such as relying on fragile external APIs, being affected by biases in GPT models, or reducing the diversity of the selected instruction dataset. In this paper, we propose an industrial-friendly, expert-aligned and diversity-preserved instruction data selection method: Clustering and Ranking (CaR). CaR employs a two-step process: first, it ranks instruction pairs using a high-accuracy (84.25%) scoring model aligned with expert preferences; second, it preserves dataset diversity through clustering. In our experiment, CaR efficiently selected a mere 1.96% of Alpaca's IT data, yet the resulting AlpaCaR model surpassed Alpaca's performance by an average of 32.1% in GPT-4 evaluations. Moreover, we find that data selecting is a consistent paradigm whether the pre-trained model is more capable or the model parameters scaling up. Our approach employs compact models with 550M parameters and incurs just 11.2% of the financial outlay of current methods, enhancing its industrial deployability.

View on arXivView PDF

Code Implementations(1)

IronBeliever/CaRCommunity100%

Clustering and Ranking: Diversity-preserved Instruction Selection through Expert-aligned Quality Estimation

913CSS, PythonFeb 14, 20241 years ago

Similar Papers

Cite this paper

@article{ge2024clustering,
  title  = {Clustering and Ranking: Diversity-preserved Instruction Selection through Expert-aligned Quality Estimation},
  author = {Yuan Ge and Yilun Liu and Chi Hu and Weibin Meng and Shimin Tao and Xiaofeng Zhao and Hongxia Ma and Li Zhang and Boxing Chen and Hao Yang and Bei Li and Tong Xiao and Jingbo Zhu},
  year   = {2024},
  eprint = {2402.18191},
  archivePrefix = {arXiv},
  url    = {http://arxiv.org/abs/2402.18191v3}
}

Add a badge to your README

Shows whether official code is linked for this paper.

Code of Paper badge
[![Code of Paper](https://codeofpaper.com/badge/2402.18191.svg)](https://codeofpaper.com/paper/2402.18191)

Discussion