TuneAhead: Predicting Fine-tuning Performance Before Full Training Begins
Not on the map. Why?
Abstract (arXiv)
Fine-tuning large language models (LLMs) is compute-intensive and error-prone: model performance depends sensitively on data quality and hyperparameter choices, and na"ive runs can even degrade model performance. This raises a practical question:can we predict fine-tuning performance before committing to a full training run? We present TUNEAHEAD, a lightweight framework for pre-hoc prediction of fine-tuning performance. TUNEAHEAD encodes each candidate run as a meta-feature vector that combines static dataset descriptors with dynamic probe features from a short standardized probe. A predictor maps these features to performance estimates, while SHAP-based attributions provide interpretable diagnostics that reveal which specific features drive the prediction. Across 1,300+ fine-tuning runs on Qwen2.5-7B-Instruct, TUNEAHEAD consistently outperforms strong baselines such as Early-Stop Extrapolation and ProxyLM. On a held-out test set of 370 runs, TUNEAHEAD achieves an RMSE of 1.47 percentage points and places 95.1% of predictions within +3/-3 percentage points of the true score. These accurate continuous predictions support practical go/no-go screening policies that can reduce unnecessary full fine-tuning while retaining most promising runs.
Key facts. Cited by 1 paper in the map (data through 13 Oct 2026). Builds on 56 of its 92 references that are in the map. First posted on arXiv 16 Jun 2026.
Builds on
56 of 92 references.
References · 56
- 2019Exploring the Limits of Transfer Learning with a Unified Text-to-Text Transformercited by 14,549
- 2017A Unified Approach to Interpreting Model Predictionscited by 6,515
- 2020Scaling Laws for Neural Language Modelscited by 6,494
- 2017Model-Agnostic Meta-Learning for Fast Adaptation of Deep Networkscited by 4,870
- 2024Qwen2.5 Technical Reportcited by 4,840
- 2021TruthfulQA: Measuring How Models Mimic Human Falsehoodscited by 2,949
- 2019The Curious Case of Neural Text Degenerationcited by 2,893
- 2018Neural Tangent Kernel: Convergence and Generalization in Neural Networkscited by 2,610
- 2018The Lottery Ticket Hypothesis: Finding Sparse, Trainable Neural Networkscited by 2,257
- 2019Fine-Tuning Language Models from Human Preferencescited by 2,139
- 2022Survey of Hallucination in Natural Language Generationcited by 2,068
- 2018Don't Give Me the Details, Just the Summary! Topic-Aware Convolutional Neural Networks for Extreme Summarizationcited by 1,312
38 more
- 2015A Diversity-Promoting Objective Function for Neural Conversation Modelscited by 1,071
- 2018Deep Anomaly Detection with Outlier Exposurecited by 941
- 2017A Closer Look at Memorization in Deep Networkscited by 939
- 2023Extracting Training Data from Diffusion Modelscited by 729
- 2021Deduplicating Training Data Makes Language Models Bettercited by 710
- 2019Data Shapley: Equitable Valuation of Data for Machine Learningcited by 621
- 2017Deep Learning Scaling is Predictable, Empiricallycited by 569
- 2019SAMSum Corpus: A Human-annotated Dialogue Dataset for Abstractive Summarizationcited by 533
- 2021Deep Learning on a Data Diet: Finding Important Examples Early in Trainingcited by 532
- 2019Rigging the Lottery: Making All Tickets Winnerscited by 493
- 2021Documenting Large Webtext Corpora: A Case Study on the Colossal Clean Crawled Corpuscited by 485
- 2021MAUVE: Measuring the Gap Between Neural Text and Human Text using Divergence Frontierscited by 468
- 2019Confident Learning: Estimating Uncertainty in Dataset Labelscited by 411
- 2023SemDeDup: Data-efficient learning at web-scale through semantic deduplicationcited by 300
- 2021GRAD-MATCH: Gradient Matching based Data Subset Selection for Efficient Deep Model Trainingcited by 273
- 2016Exploring the Limits of Language Modelingcited by 172
- 2020Unsupervised Domain Clusters in Pretrained Language Modelscited by 160
- 2020The Break-Even Point on Optimization Trajectories of Deep Neural Networkscited by 150
- 2018The Dataset Nutrition Label: A Framework To Drive Higher Data Quality Standardscited by 97
- 2024CRAG -- Comprehensive RAG Benchmarkcited by 96
- 2017Towards Understanding Generalization of Deep Learning: Perspective of Loss Landscapescited by 84
- 2019Visualizing and Understanding the Effectiveness of BERTcited by 67
- 2024ChartInsights: Evaluating Multimodal Large Language Models for Low-Level Chart Question Answeringcited by 50
- 2023Efficient Bayesian Learning Curve Extrapolation using Prior-Data Fitted Networkscited by 37
- 2024Selecting Large Language Model to Fine-tune via Rectified Scaling Lawcited by 33
- 2025NL2SQL-BUGs: A Benchmark for Detecting Semantic Errors in NL2SQL Translationcited by 33
- 2024Are Large Language Models Good Statisticians?cited by 26
- 2018A Qualitative Comparison of CoQA, SQuAD 2.0 and QuACcited by 22
- 2025A Multi-Power Law for Loss Curve Prediction Across Learning Rate Schedulescited by 22
- 2024ProxyLM: Predicting Language Model Performance on Multilingual Tasks via Proxy Modelscited by 20
- 2024Machine Translation Meta Evaluation through Translation Accuracy Challenge Setscited by 18
- 2025MEBench: Benchmarking Large Language Models for Cross-Document Multi-Entity Question Answeringcited by 16
- 2023Cost-Effective In-Context Learning for Entity Resolution: A Design Space Explorationcited by 15
- 2025LENSLLM: Unveiling Fine-Tuning Dynamics for LLM Selectioncited by 14
- 2025Massive Supervised Fine-tuning Experiments Reveal How Data, Layer, and Training Factors Shape LLM Alignment Qualitycited by 14
- 2022Predicting Fine-Tuning Performance with Probingcited by 13
- 2025LEAD: Iterative Data Selection for Efficient LLM Instruction Tuningcited by 13
- 2019Towards Understanding Generalization in Gradient-Based Meta-Learningcited by 12
50 most cited of 56 All 56 (JSON)
Built on by
Citing papers per year: 2026: 1.
1 paper.