Share your thoughts, 1 month free Claude Pro on usSee more
WorkDL logo mark

Learning Multi-Indicator Weights for Data Selection: A Joint Task-Model Adaptation Framework with Efficient Proxies

About

Data selection is a key component of efficient instruction tuning for large language models, as recent work has shown that data quality often matters more than data quantity. Accordingly, prior studies have introduced various multi-dimensional heuristics to evaluate and filter instruction data. However, most existing methods rely on static task-agnostic and model-agnostic weighting schemes, which overlook the varying requirements of specific downstream tasks and the differing pre-existing capabilities of models. In this paper, we propose a framework for learning multi-indicator weights that jointly adapts data selection to both the downstream task and the specific model. Our method identifies optimal weight configurations without full-scale fine-tuning by utilizing in-context learning (ICL) signals on compact tiny-validation sets. These signals serve as efficient performance proxies that ensure high-fidelity evaluation at minimal computational cost. Experiments across multiple benchmarks and model families, including Mistral, Qwen, and Llama, show that the approach achieves performance comparable to or exceeding full-dataset tuning while using only 30\% of the training samples on GSM8K. Furthermore, our analysis reveals a trade-off between semantic diversity and logical complexity in reasoning tasks, highlighting the necessity of joint task-model adaptation.

Jingze Song, Zihao Chen, Wenqing Chen, Zibin Zheng• 2026

Related benchmarks

TaskDatasetResultRank
Mathematical ReasoningGSM8K (test)
EM Accuracy79
41
Multistep Soft ReasoningMuSR
Accuracy (Multi-choice)50.77
27
ReasoningGPQA
Accuracy (Multi-choice)32.74
27
Mathematical ReasoningGSM8K DeepSeek-distilled (test)
Accuracy71.34
16
Science Question AnsweringDeepSeek-distilled ARC-C (test)
Accuracy77.05
16
Showing 5 of 5 rows

Other info

Follow for update