CoRA-NAS框架提升神经架构搜索效率
CoRA-NAS: Coarse Ranking and Anchor-Residual Refinement for Neural Architecture Search
这个新框架CoRA-NAS能大幅降低神经架构搜索的成本,通过两阶段方法在保持高准确率的同时节省大量计算资源。
CoRA-NAS是一种新的神经架构搜索(NAS)框架,它通过两阶段方法——CoRA-Rank和CoRA-Refine——来提升搜索效率。CoRA-Rank通过聚合静态代理指标进行粗略排序,而CoRA-Refine则使用约1%的训练成本对候选架构进行微调。在NAS-Bench-201等基准测试中,CoRA-Refine实现了0.946的Spearman相关性,其选出的架构在CIFAR-100上达到73.32%的准确率,接近最佳值73.37%。
CoRA-NAS: Coarse Ranking and Anchor-Residual Refinement for Neural Architecture Search
Zero-cost proxies rank architectures cheaply, but their reliability varies across search spaces. We introduce CoRA-NAS (COarse Ranking + Anchor-residual), a two-stage framework combining a static ranking prior with low-cost learning-curve refinement. CoRA-Rank aggregates capacity and structure-at-initialization proxies through an equal-weight log-rank consensus and a target-free consensus gate. CoRA-Refine samples anchors across this prior, extrapolates their early validation curves, and propagates a learned residual correction with an ExtraTrees model. The refinement uses approximately 1% of the cost of fully training the candidate set. Fully trained architecture-accuracy labels are not used to fit the ranker. One configuration is used across spaces, with space-specific architecture encodings. Across NAS-Bench-201, NAS-Bench-101, TransNAS-Bench-101, and NATS-SSS, CoRA-Refine achieves mean Spearman correlations of 0.946, 0.715, 0.786, and 0.894, respectively. Its worst-space correlation of 0.715 is the highest among the compared methods. On NAS-Bench-201/CIFAR-100, its selected architecture reaches 73.32% accuracy, near the reported ground-truth best of 73.37%. On the pure size space, refinement recovers the static prior's shortfall relative to parameter count, while remaining tied with the strongest capacity proxies within noise. The resulting framework combines cross-space ranking robustness with low-cost architecture selection.