CLAILGJan 12, 2024

The Unreasonable Effectiveness of Easy Training Data for Hard Tasks

arXiv:2401.06751v248 citationsh-index: 14Has CodeACL
AI Analysis

This addresses the scalable oversight problem for AI researchers and practitioners by showing that collecting easy data can be more effective than hard data for improving performance on hard tasks, though it is incremental as it builds on existing finetuning methods.

The paper tackles the problem of training models on hard tasks when labeling hard data is difficult, finding that current pretrained language models generalize well from easy to hard data, performing as well as models finetuned on hard data across seven hardness measures and up to 70b parameters.

How can we train models to perform well on hard test data when hard training data is by definition difficult to label correctly? This question has been termed the scalable oversight problem and has drawn increasing attention as language models have continually improved. In this paper, we present the surprising conclusion that current pretrained language models often generalize relatively well from easy to hard data, even performing as well as oracle models finetuned on hard data. We demonstrate this kind of easy-to-hard generalization using simple finetuning methods like in-context learning, linear classifier heads, and QLoRA for seven different measures of datapoint hardness, including six empirically diverse human hardness measures (like grade level) and one model-based measure (loss-based). Furthermore, we show that even if one cares most about model performance on hard data, it can be better to collect easy data rather than hard data for finetuning, since hard data is generally noisier and costlier to collect. Our experiments use open models up to 70b in size and four publicly available question-answering datasets with questions ranging in difficulty from 3rd grade science questions to college level STEM questions and general-knowledge trivia. We conclude that easy-to-hard generalization in LMs is surprisingly strong for the tasks studied. Our code is available at: https://github.com/allenai/easy-to-hard-generalization

Code Implementations1 repo
Foundations

The foundational work for this paper's niche, ranked by how specifically the neighbourhood builds on it — not by global fame.

Your Notes