Search papers, labs, and topics across Lattice.
YOLO-NAS-Bench is introduced as a surrogate benchmark for neural architecture search (NAS) of YOLO-style object detectors, addressing the lack of efficient benchmarks in this domain. A LightGBM surrogate predictor is trained on 1,500 YOLO architectures sampled from a defined search space and trained on COCO-mini. A novel Self-Evolving Mechanism is used to iteratively refine the predictor by focusing on high-performance architectures, leading to improved R2 and Sparse Kendall Tau metrics, and enabling the discovery of architectures that outperform YOLOv8-YOLO12 baselines.
YOLO architecture search can now be sped up dramatically: a new surrogate benchmark lets you evaluate designs without full training, and it's good enough to find architectures that beat YOLOv12.
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.
