Smaller Language Models are capable of selecting Instruction-Tuning Training Data for Larger Language Models

Fuente: arXiv
Salvato in:
Dettagli Bibliografici
Autori principali: Mekala, Dheeraj, Nguyen, Alex, Shang, Jingbo
Natura: Preprint
Pubblicazione: 2024
Soggetti:
Accesso online:
Tags: Aggiungi Tag
Nessun Tag, puoi essere il primo ad aggiungerne!!
_version_ 1866911778253832192
author Mekala, Dheeraj
Nguyen, Alex
Shang, Jingbo
author_facet Mekala, Dheeraj
Nguyen, Alex
Shang, Jingbo
contents Instruction-tuning language models has become a crucial step in aligning them for general use. Typically, this process involves extensive training on large datasets, incurring high training costs. In this paper, we introduce a novel training data selection based on the learning percentage of the samples. We assert that current language models possess the capability to autonomously select high-quality training data, leading to comparable or improved performance compared to training on the entire dataset. Our experiments span different-sized models, revealing that this characteristic holds for models ranging from 1B (small) to 13B (large) in size. Moreover, we demonstrate an interesting finding that the data hardness transfers across model sizes, and a smaller 350M model can effectively curate high-quality training data with hard samples for a larger 13B model, resulting in an equally or superior instruction-tuned model compared to training on the complete dataset. Utilizing open-sourced OPT and Llama-2 models up to 13B in size, two publicly available instruction-tuning training datasets and evaluated by both automatic metrics & humans, our paper introduces a novel approach to training data selection, showcasing a more efficient alternative.
format Preprint
id arxiv_https___arxiv_org_abs_2402_10430
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle Smaller Language Models are capable of selecting Instruction-Tuning Training Data for Larger Language Models
Mekala, Dheeraj
Nguyen, Alex
Shang, Jingbo
Computation and Language
Instruction-tuning language models has become a crucial step in aligning them for general use. Typically, this process involves extensive training on large datasets, incurring high training costs. In this paper, we introduce a novel training data selection based on the learning percentage of the samples. We assert that current language models possess the capability to autonomously select high-quality training data, leading to comparable or improved performance compared to training on the entire dataset. Our experiments span different-sized models, revealing that this characteristic holds for models ranging from 1B (small) to 13B (large) in size. Moreover, we demonstrate an interesting finding that the data hardness transfers across model sizes, and a smaller 350M model can effectively curate high-quality training data with hard samples for a larger 13B model, resulting in an equally or superior instruction-tuned model compared to training on the complete dataset. Utilizing open-sourced OPT and Llama-2 models up to 13B in size, two publicly available instruction-tuning training datasets and evaluated by both automatic metrics & humans, our paper introduces a novel approach to training data selection, showcasing a more efficient alternative.
title Smaller Language Models are capable of selecting Instruction-Tuning Training Data for Larger Language Models
topic Computation and Language
url https://arxiv.org/abs/2402.10430