Our new X account is live! Follow @wizwand_team for updates
WorkDL logo mark

HardCoRe-NAS: Hard Constrained diffeRentiable Neural Architecture Search

About

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.

Niv Nayman, Yonathan Aflalo, Asaf Noy, Lihi Zelnik-Manor• 2021

Related benchmarks

TaskDatasetResultRank
Image ClassificationImageNet 1k (test)
Top-1 Accuracy78
798
Medical Image ClassificationMedMnist BloodMnist (test)
Accuracy96.3
65
Image ClassificationTissueMNIST (test)
Accuracy68.7
41
Medical Image ClassificationMedMNIST Derma (test)
Accuracy75.1
36
Medical Image ClassificationMedMNIST Pneumonia (test)
Accuracy92.5
36
Medical Image ClassificationMedMNIST Breast (test)
Accuracy85.3
36
Medical Image ClassificationMedMNIST OrganS (test)
Accuracy80.9
30
Showing 7 of 7 rows

Other info

Follow for update