Predicting Deep Neural Network Training Outcomes from Early Training Telemetry
Not on the map. Why?
Abstract (arXiv)
Large hyperparameter sweeps for deep neural networks spend substantial compute on configurations that are effectively doomed from the first few epochs. We study whether a single training run's own early telemetry - per-epoch loss, training accuracy, gradient signal-to-noise ratio, weight-norm growth, and an activation-saturation snapshot - together with its sampled hyperparameters, can predict that run's eventual outcome without reference to other runs. We evaluate three prediction tasks: final test accuracy, relative performance within a domain, and training-dynamics failure, including numerical divergence. Across 23,788 training runs spanning six architecture/dataset combinations, gradient-boosted trees using only the first five epochs of telemetry achieve R^2 = 0.92-0.99 for final-accuracy regression and ROC-AUC = 0.983-0.998 for relative classification on a permanently held-out set of hyperparameter configurations. Useful prediction is already available after a single epoch. A paired ablation shows that gradient- and weight-level telemetry provides a statistically consistent improvement over loss and accuracy curves alone, although the practical gain varies by domain. Transfer is strong between similar architectures, while cross-dataset transfer is limited mainly by differences in accuracy scale rather than loss of the underlying relationship. These results suggest that early-training telemetry can provide a practical decision-support signal for compute allocation while motivating human oversight for any automated intervention.
Key facts. Cited by 1 paper in the map (data through 7 Oct 2026). Builds on 8 of its 12 references that are in the map. First posted on arXiv 4 Aug 2026.
Builds on
8 of 12 references.
References · 8
- 2015Deep Residual Learning for Image Recognitioncited by 51,809
- 2016Hyperband: A Novel Bandit-Based Approach to Hyperparameter Optimizationcited by 688
- 2018BOHB: Robust and Efficient Hyperparameter Optimization at Scalecited by 466
- 2019NAS-Bench-101: Towards Reproducible Neural Architecture Searchcited by 340
- 2018An Empirical Model of Large-Batch Trainingcited by 268
- 2015Non-stochastic Best Arm Identification and Hyperparameter Optimizationcited by 219
- 2021Zero-Cost Proxies for Lightweight NAScited by 194
- 2020Predicting Neural Network Accuracy from Weightscited by 107
Built on by
Citing papers per year: 2026: 1.
1 paper.