FairNAS: Rethinking Evaluation Fairness of Weight Sharing Neural Architecture Search
FairNAS: Rethinking Evaluation Fairness of Weight Sharing Neural Architecture Search
复制标题
DOI:
10.1109/iccv48922.2021.01202
复制
发表时间:
2019-07
期刊:
影响因子:
--
通讯作者:
Xiangxiang Chu;Bo Zhang;Ruijun Xu;Jixiang Li
中科院分区:
文献类型:
--
作者:
Xiangxiang Chu;Bo Zhang;Ruijun Xu;Jixiang Li
One of the most critical problems in weight-sharing neural architecture search is the evaluation of candidate models within a predefined search space. In practice, a one-shot supernet is trained to serve as an evaluator. A faithful ranking certainly leads to more accurate searching results. However, current methods are prone to making misjudgments. In this paper, we prove that their biased evaluation is due to inherent unfairness in the supernet training. In view of this, we propose two levels of constraints: expectation fairness and strict fairness. Particularly, strict fairness ensures equal optimization opportunities for all choice blocks throughout the training, which neither overestimates nor underestimates their capacity. We demonstrate that this is crucial for improving the confidence of models’ ranking. Incorporating the one-shot supernet trained under the proposed fairness constraints with a multi-objective evolutionary search algorithm, we obtain various state-of-the-art models, e.g., FairNAS-A attains 77.5% top-1 validation accuracy on ImageNet.