Paper Reading AI Learner

GreenFactory: Ensembling Zero-Cost Proxies to Estimate Performance of Neural Networks

2025-05-14 12:40:34
Gabriel Cort\^es, Nuno Louren\c{c}o, Paolo Romano, Penousal Machado

Abstract

Determining the performance of a Deep Neural Network during Neural Architecture Search processes is essential for identifying optimal architectures and hyperparameters. Traditionally, this process requires training and evaluation of each network, which is time-consuming and resource-intensive. Zero-cost proxies estimate performance without training, serving as an alternative to traditional training. However, recent proxies often lack generalization across diverse scenarios and provide only relative rankings rather than predicted accuracies. To address these limitations, we propose GreenFactory, an ensemble of zero-cost proxies that leverages a random forest regressor to combine multiple predictors' strengths and directly predict model test accuracy. We evaluate GreenFactory on NATS-Bench, achieving robust results across multiple datasets. Specifically, GreenFactory achieves high Kendall correlations on NATS-Bench-SSS, indicating substantial agreement between its predicted scores and actual performance: 0.907 for CIFAR-10, 0.945 for CIFAR-100, and 0.920 for ImageNet-16-120. Similarly, on NATS-Bench-TSS, we achieve correlations of 0.921 for CIFAR-10, 0.929 for CIFAR-100, and 0.908 for ImageNet-16-120, showcasing its reliability in both search spaces.

Abstract (translated)

在神经架构搜索过程中,确定深层神经网络的性能对于识别最优架构和超参数至关重要。传统的方法需要对每个网络进行训练和评估,这既耗时又耗费资源。零成本代理(zero-cost proxies)通过不需训练就能估计性能来作为替代方案,但最近的一些代理方法在不同场景下缺乏泛化能力,并且只能提供相对排名而非预测准确率。为了解决这些问题,我们提出了GreenFactory,这是一个结合了多种零成本代理的集合体,利用随机森林回归器将多个预测器的优势整合起来,并直接预测模型的测试准确性。 我们在NATS-Bench上对GreenFactory进行了评估,在多数据集上取得了稳健的结果。具体来说,在NATS-Bench-SSS中,GreenFactory在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,这表明GreenFactory在不同的搜索空间内都具有可靠的预测性能。

URL

https://arxiv.org/abs/2505.09344

PDF

https://arxiv.org/pdf/2505.09344.pdf


Tags
3D Action Action_Localization Action_Recognition Activity Adversarial Agent Attention Autonomous Bert Boundary_Detection Caption Chat Classification CNN Compressive_Sensing Contour Contrastive_Learning Deep_Learning Denoising Detection Dialog Diffusion Drone Dynamic_Memory_Network Edge_Detection Embedding Embodied Emotion Enhancement Face Face_Detection Face_Recognition Facial_Landmark Few-Shot Gait_Recognition GAN Gaze_Estimation Gesture Gradient_Descent Handwriting Human_Parsing Image_Caption Image_Classification Image_Compression Image_Enhancement Image_Generation Image_Matting Image_Retrieval Inference Inpainting Intelligent_Chip Knowledge Knowledge_Graph Language_Model LLM Matching Medical Memory_Networks Multi_Modal Multi_Task NAS NMT Object_Detection Object_Tracking OCR Ontology Optical_Character Optical_Flow Optimization Person_Re-identification Point_Cloud Portrait_Generation Pose Pose_Estimation Prediction QA Quantitative Quantitative_Finance Quantization Re-identification Recognition Recommendation Reconstruction Regularization Reinforcement_Learning Relation Relation_Extraction Represenation Represenation_Learning Restoration Review RNN Robot Salient Scene_Classification Scene_Generation Scene_Parsing Scene_Text Segmentation Self-Supervised Semantic_Instance_Segmentation Semantic_Segmentation Semi_Global Semi_Supervised Sence_graph Sentiment Sentiment_Classification Sketch SLAM Sparse Speech Speech_Recognition Style_Transfer Summarization Super_Resolution Surveillance Survey Text_Classification Text_Generation Time_Series Tracking Transfer_Learning Transformer Unsupervised Video_Caption Video_Classification Video_Indexing Video_Prediction Video_Retrieval Visual_Relation VQA Weakly_Supervised Zero-Shot