A Generic Graph-based Neural Architecture Encoding Scheme for Predictor-based NAS
About
This work proposes a novel Graph-based neural ArchiTecture Encoding Scheme, a.k.a. GATES, to improve the predictor-based neural architecture search. Specifically, different from existing graph-based schemes, GATES models the operations as the transformation of the propagating information, which mimics the actual data processing of neural architecture. GATES is a more reasonable modeling of the neural architectures, and can encode architectures from both the "operation on node" and "operation on edge" cell search spaces consistently. Experimental results on various search spaces confirm GATES's effectiveness in improving the performance predictor. Furthermore, equipped with the improved performance predictor, the sample efficiency of the predictor-based neural architecture search (NAS) flow is boosted. Codes are available at https://github.com/walkerning/aw_nas.
Related benchmarks
| Task | Dataset | Result | Rank | |
|---|---|---|---|---|
| Image Classification | CIFAR-10 (test) | -- | 3381 | |
| Image Classification | ImageNet (test) | -- | 235 | |
| Accuracy Prediction | NAS-Bench-101 1.0 | Kendall's Tau0.822 | 46 | |
| Architecture Performance Prediction | NAS-Bench-101 (test) | Kendall's Tau0.903 | 40 | |
| Neural Architecture Search (Performance Prediction) | NAS-Bench-201 (test) | Kendall's Tau0.9259 | 18 | |
| Neural Architecture Search | NAS-Bench-101 CIFAR-10 1.0 (S424) | Kendall's Tau66.6 | 11 | |
| Neural Architecture Search | NAS-Bench-101 CIFAR-10 1.0 | Kendall's Tau0.822 | 8 | |
| Neural Architecture Search | NAS-Bench-101 CIFAR-10 1.0 (S848) | Kendall's Tau69.1 | 8 | |
| Neural Architecture Search | NAS-Bench-101 CIFAR-10 1.0 (S100) | Kendall's Tau60.5 | 7 | |
| Neural Architecture Search | NAS-Bench-101 CIFAR-10 1.0 (S172) | Kendall's Tau65.9 | 7 |