YOLO-NAS-Bench: A Surrogate Benchmark with Self-Evolving Predictors for YOLO Architecture Search
Not on the map. Why?
Abstract (arXiv)
Neural Architecture Search (NAS) for object detection is severely bottlenecked by high evaluation cost, as fully training each candidate YOLO architecture on COCO demands days of GPU time. Meanwhile, existing NAS benchmarks largely target image classification, leaving the detection community without a comparable benchmark for NAS evaluation. To address this gap, we introduce YOLO-NAS-Bench, the first surrogate benchmark tailored to YOLO-style detectors. YOLO-NAS-Bench defines a search space spanning channel width, block depth, and operator type across both backbone and neck, covering the core modules of YOLOv8 through YOLO12. We sample 1,000 architectures via random, stratified, and Latin Hypercube strategies, train them on COCO-mini, and build a LightGBM surrogate predictor. To sharpen the predictor in the high-performance regime most relevant to NAS, we propose a Self-Evolving Mechanism that progressively aligns the predictor's training distribution with the high-performance frontier, by using the predictor itself to discover and evaluate informative architectures in each iteration. This method grows the pool to 1,500 architectures and raises the ensemble predictor's R2 from 0.770 to 0.815 and Sparse Kendall Tau from 0.694 to 0.752, demonstrating strong predictive accuracy and ranking consistency. Using the final predictor as the fitness function for evolutionary search, we discover architectures that surpass all official YOLOv8-YOLO12 baselines at comparable latency on COCO-mini, confirming the predictor's discriminative power for top-performing detection architectures. The code is available at https://github.com/VDIGPKU/YOLO-NAS-Bench.
Key facts. No citing papers in the map yet (data through 7 Oct 2026). Builds on 19 of its 24 references that are in the map. First posted on arXiv 10 Mar 2026.
Builds on
19 of 24 references.
References · 19
- 2014Microsoft COCO: Common Objects in Contextcited by 17,498
- 2017mixup: Beyond Empirical Risk Minimizationcited by 5,155
- 2016XGBoost: A Scalable Tree Boosting Systemcited by 5,120
- 2018DARTS: Differentiable Architecture Searchcited by 1,782
- 2020YOLOv4: Optimal Speed and Accuracy of Object Detectioncited by 1,624
- 2016Neural Architecture Search with Reinforcement Learningcited by 1,489
- 2018Regularized Evolution for Image Classifier Architecture Searchcited by 902
- 2020Simple Copy-Paste is a Strong Data Augmentation Method for Instance Segmentationcited by 595
- 2023DETRs Beat YOLOs on Real-time Object Detectioncited by 526
- 2024YOLOv9: Learning What You Want to Learn Using Programmable Gradient Informationcited by 422
- 2020NAS-Bench-201: Extending the Scope of Reproducible Neural Architecture Searchcited by 412
- 2019NAS-Bench-101: Towards Reproducible Neural Architecture Searchcited by 340
7 more
- 2025YOLOv12: Attention-Centric Real-Time Object Detectorscited by 300
- 2019NAS-FPN: Learning Scalable Feature Pyramid Architecture for Object Detectioncited by 289
- 2019NGBoost: Natural Gradient Boosting for Probabilistic Predictioncited by 94
- 2020Surrogate NAS Benchmarks: Going Beyond the Limited Search Spaces of Tabular NAS Benchmarkscited by 67
- 2021NAS-Bench-360: Benchmarking Neural Architecture Search on Diverse Taskscited by 32
- 2021OPANAS: One-Shot Path Aggregation Network Architecture Search for Object Detectioncited by 19
- 2023YOLOBench: Benchmarking Efficient Object Detectors on Embedded Systemscited by 5
Built on by
Not cited yet.