arXiv:2603.09405cs.CV2026-03中稿 · CVPR

为YOLO检测器设计首个高效搜索基准,大幅降低模型评估成本。

YOLO-NAS-Bench: A Surrogate Benchmark with Self-Evolving Predictors for YOLO Architecture Search

  • 构建覆盖骨干与颈部的YOLO搜索空间,采样1000个架构训练预测器。
  • 通过自进化机制提升预测器性能,R2达0.815,排名一致性显著增强。
  • 用预测器指导搜索,发现超越官方基线的高效检测模型,适合架构师使用。

目标检测中的神经网络架构搜索(NAS)因评估成本高而受限,每个候选YOLO架构在COCO上完整训练需数天GPU时间。现有NAS基准多针对图像分类,检测领域缺乏可比基准。为此,我们提出首个专为YOLO检测器设计的代理基准YOLO-NAS-Bench。该基准涵盖从YOLOv8到YOLO12的核心模块,定义了包含通道宽度、块深度和算子类型在内的搜索空间。通过随机、分层及拉丁超立方策略采样1000个架构,在COCO-mini上训练并构建LightGBM代理预测器。为提升对高性能区域的预测精度,提出自进化机制:利用预测器自身迭代发现并评估有信息量的架构,使样本池扩充至1500个,使集成预测器的R2从0.770提升至0.815,稀疏肯德尔τ从0.694升至0.752,证明其具备强预测准确性与排名一致性。将最终预测器作为演化搜索的适应度函数,所发现的架构在同等延迟下优于所有官方YOLOv8-YOLO12基线,验证其对高性能检测架构的判别能力。代码已开源。

原文摘要 · Abstract (English)

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.

YOLONAS架构搜索代理模型

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。