01 · Paper · SIAM SDM 2023 · co-author
A General-Purpose Transferable Predictor for Neural Architecture Search
Can one performance predictor rank networks from search spaces it was never built for?
Architecture search spends most of its budget scoring candidates. Predictors make that cheap, but the predictors of the time each understood one search space, because they read a fixed recipe of blocks.
The paper rewrites any convolutional network as a computation graph of primitive operations, then learns that graph with contrastive learning on unlabelled architectures from several families. A small head, fine-tuned on a few scored networks, ranks a family the predictor was not trained on.
I am a co-author, not the lead author. It was written at Huawei's Canadian lab in Edmonton, with the University of Alberta and UBC.
- 0.917
Spearman on NAS-Bench-201
- 0.892
Spearman on NAS-Bench-301, where zero-cost proxies go negative
- 94.23%
best NAS-Bench-101 architecture found in 700 queries, on average the 2nd best in the space
- 79.2%
ImageNet top-1 for a MobileNetV3 found with the transferred predictor
Spearman figures are means over five fine-tunes, with a spread of about 0.01. Search and ImageNet numbers are the single results the paper reports.