Niv Nayman, Yonathan Aflalo, Asaf Noy, Lihi Zelnik-Manor
Realistic use of neural networks often requires adhering to multiple constraints on latency, energy and memory among others. A popular approach to find fitting networks is through constrained Neural Architecture Search (NAS), however, previous methods enforce the constraint only softly. Therefore, the resulting networks do not exactly adhere to the resource constraint and their accuracy is harmed. In this work we resolve this by introducing Hard Constrained diffeRentiable NAS (HardCoRe-NAS), that is based on an accurate formulation of the expected resource requirement and a scalable search method that satisfies the hard constraint throughout the search. Our experiments show that HardCoRe-NAS generates state-of-the-art architectures, surpassing other NAS methods, while strictly satisfying the hard resource constraints without any tuning required.
| Task | Dataset | Metric | Value | Model |
|---|---|---|---|---|
| Neural Architecture Search | ImageNet | Accuracy | 80.1 | HardcoreNAS_E_KD |
| Neural Architecture Search | ImageNet | Top-1 Error Rate | 19.9 | HardcoreNAS_E_KD |
| Neural Architecture Search | ImageNet | Accuracy | 79.5 | HardcoreNAS_D_KD |
| Neural Architecture Search | ImageNet | Top-1 Error Rate | 20.5 | HardcoreNAS_D_KD |
| Neural Architecture Search | ImageNet | Accuracy | 78.9 | HardcoreNAS_C_KD |
| Neural Architecture Search | ImageNet | Top-1 Error Rate | 21.1 | HardcoreNAS_C_KD |
| Neural Architecture Search | ImageNet | Accuracy | 78.8 | HardcoreNAS_B_KD |
| Neural Architecture Search | ImageNet | Top-1 Error Rate | 21.2 | HardcoreNAS_B_KD |
| Neural Architecture Search | ImageNet | Accuracy | 78.3 | HardcoreNAS_A_KD |
| Neural Architecture Search | ImageNet | Top-1 Error Rate | 21.7 | HardcoreNAS_A_KD |
| Neural Architecture Search | ImageNet | Accuracy | 78.1 | HardcoreNAS_F |
| Neural Architecture Search | ImageNet | Top-1 Error Rate | 21.9 | HardcoreNAS_F |
| Neural Architecture Search | ImageNet | Accuracy | 77.9 | HardcoreNAS_E |
| Neural Architecture Search | ImageNet | Top-1 Error Rate | 22.1 | HardcoreNAS_E |
| Neural Architecture Search | ImageNet | Accuracy | 77.4 | HardcoreNAS_D |
| Neural Architecture Search | ImageNet | Top-1 Error Rate | 22.6 | HardcoreNAS_D |
| Neural Architecture Search | ImageNet | Accuracy | 77.1 | HardcoreNAS_C |
| Neural Architecture Search | ImageNet | Top-1 Error Rate | 22.9 | HardcoreNAS_C |
| Neural Architecture Search | ImageNet | Accuracy | 76.5 | HardcoreNAS_B |
| Neural Architecture Search | ImageNet | Top-1 Error Rate | 23.5 | HardcoreNAS_B |
| Neural Architecture Search | ImageNet | Accuracy | 75.9 | HardcoreNAS_A |
| Neural Architecture Search | ImageNet | Top-1 Error Rate | 24.1 | HardcoreNAS_A |
| AutoML | ImageNet | Accuracy | 80.1 | HardcoreNAS_E_KD |
| AutoML | ImageNet | Top-1 Error Rate | 19.9 | HardcoreNAS_E_KD |
| AutoML | ImageNet | Accuracy | 79.5 | HardcoreNAS_D_KD |
| AutoML | ImageNet | Top-1 Error Rate | 20.5 | HardcoreNAS_D_KD |
| AutoML | ImageNet | Accuracy | 78.9 | HardcoreNAS_C_KD |
| AutoML | ImageNet | Top-1 Error Rate | 21.1 | HardcoreNAS_C_KD |
| AutoML | ImageNet | Accuracy | 78.8 | HardcoreNAS_B_KD |
| AutoML | ImageNet | Top-1 Error Rate | 21.2 | HardcoreNAS_B_KD |
| AutoML | ImageNet | Accuracy | 78.3 | HardcoreNAS_A_KD |
| AutoML | ImageNet | Top-1 Error Rate | 21.7 | HardcoreNAS_A_KD |
| AutoML | ImageNet | Accuracy | 78.1 | HardcoreNAS_F |
| AutoML | ImageNet | Top-1 Error Rate | 21.9 | HardcoreNAS_F |
| AutoML | ImageNet | Accuracy | 77.9 | HardcoreNAS_E |
| AutoML | ImageNet | Top-1 Error Rate | 22.1 | HardcoreNAS_E |
| AutoML | ImageNet | Accuracy | 77.4 | HardcoreNAS_D |
| AutoML | ImageNet | Top-1 Error Rate | 22.6 | HardcoreNAS_D |
| AutoML | ImageNet | Accuracy | 77.1 | HardcoreNAS_C |
| AutoML | ImageNet | Top-1 Error Rate | 22.9 | HardcoreNAS_C |
| AutoML | ImageNet | Accuracy | 76.5 | HardcoreNAS_B |
| AutoML | ImageNet | Top-1 Error Rate | 23.5 | HardcoreNAS_B |
| AutoML | ImageNet | Accuracy | 75.9 | HardcoreNAS_A |
| AutoML | ImageNet | Top-1 Error Rate | 24.1 | HardcoreNAS_A |