LLM2D
GreenFactory:聚合零成本代理以估计神经网络性能
GreenFactory: Ensembling Zero-Cost Proxies to Estimate Performance of Neural Networks
作者: Gabriel Cort\^es, Nuno Louren\c{c}o, Paolo Romano, Penousal Machado
发布日期: 5/15/2025
arXiv ID: oai:arXiv.org:2505.09344v1

摘要

arXiv:2505.09344v1 交叉公告类型 摘要:在神经架构搜索过程中确定深度神经网络的性能对于识别最优架构和超参数至关重要。传统的方法需要对每个网络进行训练和评估,这既耗时又资源密集。零成本代理通过不进行训练来估计性能,作为传统训练的一种替代方案。然而,最近的代理往往在多种场景下缺乏概括性,并且只能提供相对排名而不是预测的准确度。为了解决这些限制,我们提出了一种名为GreenFactory的零成本代理集合,利用随机森林回归器整合多个预测器的优势,直接预测模型测试准确度。我们利用NATS-Bench评估了GreenFactory,使其在多个数据集上表现出稳健的结果。具体来说,GreenFactory在NATS-Bench-SSS上实现了高Kendall相关性,表明其预测分数与实际性能之间有显著的一致性:CIFAR-10上的0.907,CIFAR-100上的0.945,以及ImageNet-16-120上的0.920。同样,在NATS-Bench-TSS上,我们实现了CIFAR-10上的0.921,CIFAR-100上的0.929,以及ImageNet-16-120上的0.908的相关性,展示了其在两个搜索空间中的可靠性。