对比八种损失函数,发现组合使用能提升神经网络架构搜索效率
Loss Functions for Predictor-based Neural Architecture Search
- 分类三类损失函数:回归、排序、加权,评估其在架构搜索中的表现
- 13个任务测试显示,特定组合可显著提升预测准确率
- 为不同任务选择合适损失函数提供实证指导,适合NAS研究者参考
神经架构搜索(NAS)中的评估环节成本高昂。性能预测器通过直接估计架构表现来降低评估开销,其效果高度依赖损失函数的选择。传统方法采用回归损失评估架构的绝对准确率,近期研究则探索了成对与列表排序损失等基于排序的方法,以关注架构性能的相对排名。尽管这些方法在NAS中取得成功,但其有效性与特性尚未被系统研究。本文首次对性能预测器中的损失函数进行综合分析,将其分为三类:回归、排序与加权损失函数。我们在五个搜索空间的13个任务上,使用多种NAS相关指标评估了八种损失函数。结果表明,特定类别损失函数的组合能有效提升基于预测器的NAS表现。研究结果可为不同任务选择合适的损失函数提供实践指导,期望为NAS社区中预测方法的损失函数设计提供有益洞察。
原文摘要 · Abstract (English)
Evaluation is a critical but costly procedure in neural architecture search (NAS). Performance predictors have been widely adopted to reduce evaluation costs by directly estimating architecture performance. The effectiveness of predictors is heavily influenced by the choice of loss functions. While traditional predictors employ regression loss functions to evaluate the absolute accuracy of architectures, recent approaches have explored various ranking-based loss functions, such as pairwise and listwise ranking losses, to focus on the ranking of architecture performance. Despite their success in NAS, the effectiveness and characteristics of these loss functions have not been thoroughly investigated. In this paper, we conduct the first comprehensive study on loss functions in performance predictors, categorizing them into three main types: regression, ranking, and weighted loss functions. Specifically, we assess eight loss functions using a range of NAS-relevant metrics on 13 tasks across five search spaces. Our results reveal that specific categories of loss functions can be effectively combined to enhance predictor-based NAS. Furthermore, our findings could provide practical guidance for selecting appropriate loss functions for various tasks. We hope this work provides meaningful insights to guide the development of loss functions for predictor-based methods in the NAS community.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。